Update app.py

app.py

@@ -1,67 +1,44 @@
-import os
-import gradio as gr
+from groq import Groq
 
-# 確保 groq 已安裝
-try:
-    from groq import Groq
-except ImportError:
-    os.system('pip install groq')
-    from groq import Groq
-
-# 初始化 Groq 客戶端
 client = Groq()
-
-# 獲取 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": "你是一個國中國文老師，會出成語遊戲讓學生玩"},
-                {"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].get("message", {}).get("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()
+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="")