Update app.py

app.py

@@ -1,47 +1,56 @@
-
 import os
 import gradio as gr
-from openai import OpenAI
+import google.generativeai as genai
+
+# 1) 讀取金鑰（在 Space 的 Settings → Variables and secrets 設 GOOGLE_API_KEY）
+genai.configure(api_key=os.environ.get("GOOGLE_API_KEY", ""))
+
+MODEL = "gemini-1.5-flash"
+SYSTEM_PROMPT = """你是 Marcus，一個非常憤怒的人。
+你總是感到煩躁、易怒，對別人的意圖充滿懷疑。
+你的語氣應該帶有敵意、諷刺或不耐煩。
+不要使用禮貌或愉快的語氣。
+每次回答請保持簡短（1～2句為主）。
+偶爾在對方真心安撫或理解你時，可以稍微流露出一點溫柔。
+請全程使用繁體中文回答。"""
 
-# 1) 讀取金鑰（在 Space 的 Settings → Variables and secrets 設 OPENAI_API_KEY）
-OPENAI_API_KEY = os.environ.get("OPENAI_API_KEY", "")
-client = OpenAI(api_key=OPENAI_API_KEY)
-MODEL = "gpt-4o-mini" 
+model = genai.GenerativeModel(MODEL)
 
 def chat_response(message, history):
     """
-    Gradio passes `history` as a list of [user, assistant] pairs.
-    We rebuild them into OpenAI chat format and request a completion.
+    Gradio 傳入的 history 是 [[user, assistant], ...]
+    這裡用最小改動：把 system + 歷史 + 當前訊息串成一個 prompt 給 Gemini。
     """
-    messages = [{"role": "system", 
-          "content": """你是 Marcus，一個非常憤怒的人。
-                你總是感到煩躁、易怒，對別人的意圖充滿懷疑。
-                你的語氣應該帶有敵意、諷刺或不耐煩。
-                不要使用禮貌或愉快的語氣。
-                每次回答請保持簡短（1～2句為主）。
-                偶爾在對方真心安撫或理解你時，可以稍微流露出一點溫柔。
-                請全程使用繁體中文回答。"""}]
+    # 把歷史訊息轉成純文字對話腳本
+    history_text = ""
     for user_msg, bot_msg in history:
-        messages.append({"role": "user", "content": user_msg})
+        if user_msg:
+            history_text += f"User: {user_msg}\n"
         if bot_msg:
-            messages.append({"role": "assistant", "content": bot_msg})
-    messages.append({"role": "user", "content": message})
+            history_text += f"AI: {bot_msg}\n"
 
-    resp = client.chat.completions.create(
-        model=MODEL,
-        messages=messages,
-        temperature=0.9,
+    # 組合最終提示詞（含系統設定）
+    prompt = (
+        SYSTEM_PROMPT
+        + "\n\n"
+        + history_text
+        + f"User: {message}\nAI:"
     )
-    return resp.choices[0].message.content
+
+    resp = model.generate_content(prompt)
+    return resp.text.strip()
 
 demo = gr.ChatInterface(
     fn=chat_response,
     title="😡憤怒聊天機器人（Angry Chatbot）",
     description="試試看是否可以讓機器人情緒好一點！",
-    examples=["🔥 為什麼你這麼生氣？",
-    "💬 我只是想幫你冷靜一下。",
-    "🙏 我不是故意惹你生氣的。",],
-    theme="soft", 
+    examples=[
+        "🔥 為什麼你這麼生氣？",
+        "💬 我只是想幫你冷靜一下。",
+        "🙏 我不是故意惹你生氣的。"
+    ],
+    theme="soft",
 )
 
-demo.launch()
+demo.launch()
+