++

main.py

@@ -153,6 +153,24 @@ def manual_prompt(prompt):
     print(f"手动提示词保存到: {filepath}")
 
 
+def call_ai(prompt):
+    balance = get_user_info()
+
+    folder = create_result_folder()
+
+    for model in MODELS:
+        print(f"正在调用AI...{model}")
+        result = call_siliconflow(prompt, model)
+
+        if result:
+            print(f"AI回复: {result[:200]}...")
+            save_result(result, folder)
+            used_balance = balance - get_user_info()
+            print(f'本次调用 api 使用额度 {used_balance}')
+        else:
+            print("AI调用失败")
+
+
 def prepare_prompt():
     prompt = ''
 
@@ -208,21 +226,8 @@ def main():
     # # 如果需要手动在页面段模型, 使用提示词, 打开这个, 将生成的提示词存到本地
     manual_prompt(prompt)
 
-    balance = get_user_info()
-
-    folder = create_result_folder()
-
-    for model in MODELS:
-        print(f"正在调用AI...{model}")
-        result = call_siliconflow(prompt, model)
-
-        if result:
-            print(f"AI回复: {result[:200]}...")
-            save_result(result, folder)
-            used_balance = balance - get_user_info()
-            print(f'本次调用 api 使用额度 {used_balance}')
-        else:
-            print("AI调用失败")
+    # # 如果需要使用模型, 打开这个
+    # call_ai(prompt)
 
 
 if __name__ == "__main__":