Update app.py

app.py

@@ -1,67 +1,65 @@
 import gradio as gr
 import os
+from groq import Groq
+from gradio import ChatMessage
 
-# Install the groq package if it is not installed
-try:
-    from groq import Groq
-except ImportError:
-    os.system('pip install groq')
-    from groq import Groq
+# 從環境變數中獲取 API 密鑰
+api_key = os.getenv("groq_key")
 
-# Set up the Groq client with the secret key
-groq_key = os.getenv('groq_key')
-if not groq_key:
-    raise ValueError("groq_key environment variable is not set")
+# 初始化 groq 客戶端
+client = Groq()
+client.set_api_key(api_key)
 
-client = Groq(api_key=groq_key)
+# 定義回應函數
+def respond(message, chat_history):
+    # 構建聊天記錄
+    messages = [{"role": "system", "content": "你是一個友善的 AI 助手，請幫助使用者解答問題。"}]
+    for entry in chat_history:
+        if entry['role'] == 'user':
+            messages.append({"role": "user", "content": entry['content']})
+        else:
+            messages.append({"role": "assistant", "content": entry['content']})
+    
+    # 添加當前用戶消息
+    messages.append({"role": "user", "content": message})
 
-class SimpleChatBot:
-    def __init__(self):
-        self.initial_prompt = [
-            {
-                "role": "system",
-                "content": "你是一個英文老師，會教我背單字，並用蘇格拉底問答引導我，引導時請用繁體中文 zhTW"
-            }
-        ]
+    # 獲取回應
+    completion = client.chat.completions.create(
+        model="llama-3.1-70b-versatile",
+        messages=messages,
+        temperature=1,
+        max_tokens=1024,
+        top_p=1,
+        stream=True,
+        stop=None,
+    )
 
-    def get_response(self, message, chat_history):
-        messages = self.initial_prompt + chat_history
-        messages.append({"role": "user", "content": message})
-        
-        completion = client.chat.completions.create(
-            model="llama-3.1-70b-versatile",
-            messages=messages,
-            temperature=1,
-            max_tokens=1024,
-            top_p=1,
-            stream=True,
-            stop=None,
-        )
-        
-        response_content = ""
-        for chunk in completion:
-            response_content += chunk.choices[0].delta.content or ""
-        
-        return response_content
+    reply = ""
+    for chunk in completion:
+        reply += chunk.choices[0].delta.content or ""
+    
+    # 添加當前回應到聊天記錄
+    chat_history.append(ChatMessage(role="user", content=message))
+    chat_history.append(ChatMessage(role="assistant", content=reply))
+    return chat_history
 
-chatbot = SimpleChatBot()
+# 創建 Gradio Chatbot 界面
+with gr.Blocks() as demo:
+    chatbot = gr.Chatbot(type='messages')
+    msg = gr.Textbox(label="輸入訊息")
+    clear = gr.Button("清除聊天")
 
-def respond(message, chat_history):
-    chat_history = [{"role": entry["role"], "content": entry["content"]} for entry in chat_history]
-    response = chatbot.get_response(message, chat_history)
-    chat_history.append({"role": "user", "content": message})
-    chat_history.append({"role": "assistant", "content": response})
-    return chat_history, ""
+    def user(user_message, history):
+        return "", history + [ChatMessage(role="user", content=user_message)]
 
-with gr.Blocks(title="簡單的Gradio聊天機器人") as demo:
-    gr.Markdown("# 簡單的Gradio聊天機器人")
-    
-    chatbot_interface = gr.Chatbot(type="messages")
-    
-    with gr.Row():
-        user_input = gr.Textbox(placeholder="輸入訊息...", label="你的訊息")
-        send_button = gr.Button("發送")
-    
-    send_button.click(respond, inputs=[user_input, chatbot_interface], outputs=[chatbot_interface, user_input])
-    
-    demo.launch(share=True)
+    def bot(history):
+        user_message = history[-1]['content']
+        history = respond(user_message, history[:-1])
+        return history
+
+    msg.submit(user, [msg, chatbot], [msg, chatbot], queue=False).then(
+        bot, chatbot, chatbot
+    )
+    clear.click(lambda: [], None, chatbot, queue=False)
+
+demo.launch()