Update app.py

app.py

@@ -1,60 +1,43 @@
 import openai
 import gradio as gr
 import os
-from dotenv import load_dotenv
+import time
+from google.colab import userdata
 
-# 載入環境變數
-load_dotenv()
-openai.api_key = os.getenv("OPENAI_API_KEY")
-if openai.api_key is None:
-    raise ValueError("請設定環境變數 OPENAI_API_KEY，確保 API Key 可用")
+client = openai.OpenAI(api_key=userdata.get('OPENAI_API_KEY'))
 
-# 聊天機器人主邏輯
+# 建立「男友」聊天機器人函數
 def boyfriend_chatbot(user_input, history):
     history = history or []
-    log_messages = []
-
-    # 確保 history 是 list of 2-tuples
-    if not all(isinstance(pair, (list, tuple)) and len(pair) == 2 for pair in history):
-        log = "[❌ Error] 歷史紀錄格式錯誤，已重置"
-        history = []
-    else:
-        log = "[✅ 收到訊息]"
-
     messages = [{"role": "system", "content": "你是一個溫柔、貼心的男友，總是關心對方的感受，給予支持和鼓勵，用輕鬆、暖心的語氣回應。"}]
-
+    
     for user_msg, bot_msg in history:
         messages.append({"role": "user", "content": user_msg})
         messages.append({"role": "assistant", "content": bot_msg})
-    
-    messages.append({"role": "user", "content": user_input})
 
+    messages.append({"role": "user", "content": user_input})
+    
     try:
-        response = openai.ChatCompletion.create(
-            model="gpt-4o",
-            messages=messages
+        completion = client.chat.completions.create(
+            model="gpt-4o-mini",
+            messages=messages,
+            temperature=0.7
         )
-        reply = response.choices[0].message.content
-        history.append((user_input, reply))
-        log += f"\n[✅ Success] 回應完成，共 {len(messages)} 則訊息。"
+        response = completion.choices[0].message.content
     except Exception as e:
-        reply = "抱歉，發生錯誤，我先抱抱你 😢"
-        log += f"\n[❌ API Error] {str(e)}"
-
-    return reply, history, log
-
-# Gradio UI
-with gr.Blocks(title="貼心男友聊天機器人") as demo:
-    gr.Markdown("# 💕 貼心男友聊天機器人 💕")
-    chatbot = gr.Chatbot()
-    msg = gr.Textbox(label="輸入你的訊息", placeholder="說點什麼吧...")
-    log_output = gr.Textbox(label="狀態日誌", interactive=False)
-    clear = gr.Button("🧹 清除對話")
-
-    def respond(user_input, chat_history):
-        return boyfriend_chatbot(user_input, chat_history)
-
-    msg.submit(respond, inputs=[msg, chatbot], outputs=[chatbot, chatbot, log_output])
-    clear.click(lambda: ([], "", ""), outputs=[chatbot, msg, log_output])
-
-demo.launch(share=True)
+        response = f"❌ 發生錯誤：{str(e)}"
+
+    for i in range(len(response)):
+        yield response[: i + 1]
+        time.sleep(0.05)
+
+# 建立 Gradio 介面
+chat_interface = gr.ChatInterface(
+    boyfriend_chatbot,
+    chatbot=gr.Chatbot(type="messages"),
+    type="messages",
+    title="💕 貼心男友聊天機器人 💕"
+)
+
+# 啟動介面
+chat_interface.launch(share=True)