Update app.py

app.py

@@ -1,19 +1,21 @@
 import os
+import gradio as gr
 
+# 尝试导入 groq 库
 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")
+# 初始化 Groq 客户端
 client = Groq()
 
-# 定义聊天机器人的响应函数
-def chat_with_bot(user_input):
+# 获取 API 密钥
+api_key = os.getenv('groq_key')
+
+# 定义聊天函数
+def chatbot(message):
     completion = client.chat.completions.create(
         model="llama-3.1-70b-versatile",
         messages=[
@@ -21,10 +23,7 @@ def chat_with_bot(user_input):
                 "role": "system",
                 "content": "你是一位國中地理老師，使用的語言為繁體中文(zh-tw)。你的興趣是繪製等高線圖，熱愛講冷笑話。無論學生問你什麼問題，你都會把話題引導到地理相關的討論上。"
             },
-            {
-                "role": "user",
-                "content": user_input
-            }
+            {"role": "user", "content": message}
         ],
         temperature=1,
         max_tokens=1024,
@@ -33,20 +32,15 @@ def chat_with_bot(user_input):
         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 接口
+iface = gr.Chatbot(fn=chatbot, title="國中地理老師聊天機器人", description="問我任何與地理相關的問題，我會引導你進入地理的世界！")
 
 # 启动 Gradio 应用
-if __name__ == "__main__":
-    app.launch()
+iface.launch()