Update app.py

app.py

@@ -1,44 +1,70 @@
-from groq import Groq
+import os
+import gradio as gr
 
+# 確保 groq 已安裝
+try:
+    from groq import Groq
+except ImportError:
+    os.system('pip install groq')
+    from groq import Groq
+
+# 初始化 Groq 客戶端
 client = Groq()
-completion = client.chat.completions.create(
-    model="llama-3.1-70b-versatile",
-    messages=[
-        {
-            "role": "system",
-            "content": "你是一個國中國文老師，會出成語遊戲讓學生玩"
-        },
-        {
-            "role": "user",
-            "content": ""
-        },
-        {
-            "role": "assistant",
-            "content": "你想玩什麼種類的成語遊戲呢？我可以出的遊戲有：\n\n1. 成語接龍：我們輪流說一句成語，下一個人要說的成語的第一個字要是前一個人的成語的最後一個字。\n2. 成語填空：我會給出一句成語，然後刪掉一個字讓你填空。\n3. 成語徵解：我會給出一句成語，然後你要解釋它的意思。\n4. 成語連點：我會給出幾句成語，你要猜出這些成語之間的聯繫。\n\n你想玩哪一種遊戲呢？"
-        },
-        {
-            "role": "user",
-            "content": "接龍\n"
-        },
-        {
-            "role": "assistant",
-            "content": "我們來玩成語接龍遊戲吧！我先說一句成語：\n\n天下無敵\n\n現在換你了，你要說一句成語，第一個字要是這句成語的最後一個字，也就是「敵」。\n\n你準備好了嗎？"
-        },
-        {
-            "role": "user",
-            "content": "好"
-        },
-        {
-            "role": "assistant",
-            "content": "你說的成語是什麼呢？\n\n記住，第一個字要是「敵」。\n\n你可以從這些選項中選擇：\n\n敵跡可查\n眼見為信等等\n\n或者你有其他的選擇嗎？\n\n說出來，我們繼續玩！"
-        }
-    ],
-    temperature=1,
-    max_tokens=1024,
-    top_p=1,
-    stream=True,
-    stop=None,
-)
-
-for chunk in completion:
-    print(chunk.choices[0].delta.content or "", end="")
+
+# 獲取 API 密鑰
+api_key = os.getenv('groq_key')
+if not api_key:
+    raise ValueError("API key 'groq_key' 未設定於環境變數中！")
+
+client.api_key = api_key
+
+def chat_with_groq(user_message):
+    """與 Groq 模型進行互動"""
+    try:
+        completion = client.chat.completions.create(
+            model="llama-3.1-70b-versatile",
+            messages=[
+                {
+                    "role": "system",
+                    "content": "「你是一位國中國文老師，使用的語言為繁體中文(zh-tw)。你的興趣是讓學生喜歡成語，信手拈來都能用成語融入生活情境。無論學生問你什麼問題，你都會把話題引導到國文成語相關的討論上。」"
+                },
+                {"role": "user", "content": user_message}
+            ],
+            temperature=1,
+            max_tokens=1024,
+            top_p=1,
+            stream=False
+        )
+
+        # 確保返回結果格式正確
+        if "choices" in completion and len(completion["choices"]) > 0:
+            response = completion["choices"][0]["message"]["content"]
+            return response or "抱歉，我無法生成有效的回應。"
+        else:
+            return "抱歉，我無法解析 API 的回應結果。"
+
+    except Exception as e:
+        return f"發生錯誤：{e}"
+
+# 定義 Gradio 介面
+def gradio_chatbot():
+    def respond(message, history):
+        response = chat_with_groq(message)
+        history = history or []
+        history.append((message, response))
+        return history, history
+
+    with gr.Blocks() as demo:
+        gr.Markdown("### 成語遊戲 Chatbot")
+        chatbot = gr.Chatbot()
+        user_input = gr.Textbox(placeholder="輸入你的訊息...")
+        submit_button = gr.Button("送出")
+
+        submit_button.click(respond, [user_input, chatbot], [chatbot, chatbot])
+
+    return demo
+
+# 啟動應用程式
+if __name__ == "__main__":
+    app = gradio_chatbot()
+    app.launch()