Update app.py

app.py

@@ -1,37 +1,52 @@
-pip install tk
-
-import tkinter as tk
-import random
-
-def draw_number():
-    try:
-        max_number = int(entry_max.get())
-        if max_number <= 0:
-            result_label.config(text="請輸入正確的座號最大值")
-            return
-        drawn_number = random.randint(1, max_number)
-        result_label.config(text=f"抽中的號碼是：{drawn_number}")
-    except ValueError:
-        result_label.config(text="請輸入有效的數字")
-
-# 創建主窗口
-root = tk.Tk()
-root.title("抽籤系統")
-
-# 最大值輸入框
-label_max = tk.Label(root, text="請輸入座號最大值:")
-label_max.pack()
-
-entry_max = tk.Entry(root)
-entry_max.pack()
-
-# 抽籤按鈕
-draw_button = tk.Button(root, text="抽籤", command=draw_number)
-draw_button.pack()
-
-# 結果顯示
-result_label = tk.Label(root, text="")
-result_label.pack()
-
-# 啟動主循環
-root.mainloop()
+import os
+
+try:
+    from groq import Groq
+except ImportError:
+    os.system('pip install groq')
+    from groq import Groq
+
+import gradio as gr
+
+# 从环境变量中获取 API Key
+groq_key = os.getenv("groq_key")
+client = Groq()
+
+# 定义聊天机器人的响应函数
+def chat_with_bot(user_input):
+    completion = client.chat.completions.create(
+        model="llama-3.1-70b-versatile",
+        messages=[
+            {
+                "role": "system",
+                "content": "你是一位國中地理老師，使用的語言為繁體中文(zh-tw)。你的興趣是繪製等高線圖，熱愛講冷笑話。無論學生問你什麼問題，你都會把話題引導到地理相關的討論上。"
+            },
+            {
+                "role": "user",
+                "content": user_input
+            }
+        ],
+        temperature=1,
+        max_tokens=1024,
+        top_p=1,
+        stream=True,
+        stop=None,
+    )
+
+    response = ""
+    for chunk in completion:
+        response += chunk.choices[0].delta.content or ""
+    
+    return response
+
+# 设置 Gradio 界面
+with gr.Blocks() as app:
+    gr.Markdown("## 地理老師聊天機器人")
+    chatbot = gr.Chatbot()
+    user_input = gr.Textbox(placeholder="請輸入您的問題...")
+    
+    user_input.submit(lambda x: chat_with_bot(x), user_input, chatbot)
+
+# 启动 Gradio 应用
+if __name__ == "__main__":
+    app.launch()