Upload app.py

app.py

@@ -114,6 +114,41 @@ def generate_prompt(achievement: str, emotion: str, information: str,
 ############################################
 # 呼叫模型 API (OpenAI / Claude)
 ############################################
+def call_openai_api_4o(prompt: str) -> str:
+    """專門呼叫 ChatGPT 4o 模型"""
+    if not client.api_key:
+        return "[OpenAI API Key 未設定，無法呼叫 API]"
+
+    try:
+        # 以 ChatCompletion 方式呼叫
+        chat_completion = client.chat.completions.create(
+            messages=[{"role": "user", "content": prompt}],
+            model="gpt-4o",
+            temperature=0.7,
+            max_completion_tokens=2048,
+        )
+        return chat_completion.choices[0].message.content
+    except Exception as e:
+        return f"[OpenAI API 呼叫失敗]: {str(e)}"
+
+
+def call_openai_api_o1(prompt: str) -> str:
+    """專門呼叫 ChatGPT o1 模型"""
+    if not client.api_key:
+        return "[OpenAI API Key 未設定，無法呼叫 API]"
+
+    try:
+        # 以 ChatCompletion 方式呼叫
+        chat_completion = client.chat.completions.create(
+            messages=[{"role": "user", "content": prompt}],
+            model="o1",
+            max_completion_tokens=2048,
+        )
+        return chat_completion.choices[0].message.content
+    except Exception as e:
+        return f"[OpenAI API 呼叫失敗]: {str(e)}"
+
+  
 def call_openai_api(prompt: str, model_name: str) -> str:
     """參考新版 usage: from openai import OpenAI -> client.chat.completions.create()"""
     if not client.api_key:
@@ -170,7 +205,6 @@ def call_claude_api(prompt: str) -> str:
     except Exception as e:
         return f"[Claude API 呼叫失敗]: {str(e)}"
 
-
 def get_model_answer(model_choice: str, final_prompt: str, question: str) -> str:
     """
     根據 model_choice 呼叫對應的 API。
@@ -179,9 +213,9 @@ def get_model_answer(model_choice: str, final_prompt: str, question: str) -> str
     content_to_send = final_prompt + "\n\n" + question
 
     if model_choice == "ChatGPT 4o":
-        return call_openai_api(content_to_send, model_name="gpt-4o")
+        return call_openai_api_4o(content_to_send)
     elif model_choice == "ChatGPT o1":
-        return call_openai_api(content_to_send, model_name="o1")
+        return call_openai_api_o1(content_to_send)
     elif model_choice == "Claude 3.5 Sonnet":
         return call_claude_api(content_to_send)
     else: