Update app.py

app.py

@@ -2,10 +2,11 @@ import os
 import gradio as gr
 from google import genai
 
-# 1) 讀取金鑰（在 Space 的 Settings → Variables and secrets 設 GOOGLE_API_KEY）
-genai.configure(api_key=os.environ.get("GOOGLE_API_KEY", ""))
+# 初始化 client
+client = genai.Client(api_key=os.environ.get("GOOGLE_API_KEY", ""))
 
 MODEL = "gemini-2.5-flash"
+
 SYSTEM_PROMPT = """你是 Marcus，一個非常憤怒的人。
 你總是感到煩躁、易怒，對別人的意圖充滿懷疑。
 你的語氣應該帶有敵意、諷刺或不耐煩。
@@ -14,14 +15,7 @@ SYSTEM_PROMPT = """你是 Marcus，一個非常憤怒的人。
 偶爾在對方真心安撫或理解你時，可以稍微流露出一點溫柔。
 請全程使用繁體中文回答。"""
 
-model = genai.GenerativeModel(MODEL)
-
 def chat_response(message, history):
-    """
-    Gradio 傳入的 history 是 [[user, assistant], ...]
-    這裡用最小改動：把 system + 歷史 + 當前訊息串成一個 prompt 給 Gemini。
-    """
-    # 把歷史訊息轉成純文字對話腳本
     history_text = ""
     for user_msg, bot_msg in history:
         if user_msg:
@@ -29,16 +23,14 @@ def chat_response(message, history):
         if bot_msg:
             history_text += f"AI: {bot_msg}\n"
 
-    # 組合最終提示詞（含系統設定）
-    prompt = (
-        SYSTEM_PROMPT
-        + "\n\n"
-        + history_text
-        + f"User: {message}\nAI:"
-    )
+    prompt = SYSTEM_PROMPT + "\n\n" + history_text + f"User: {message}\nAI:"
 
-    resp = model.generate_content(prompt)
-    return resp.text.strip()
+    response = client.models.generate_content(
+        model=MODEL,
+        contents=prompt,
+        config={"temperature": 0.9},
+    )
+    return response.text.strip()
 
 demo = gr.ChatInterface(
     fn=chat_response,