Update app.py

app.py

@@ -1,69 +1,67 @@
 import gradio as gr
 import os
 
-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")
 
-# 初始化 groq 客戶端
-client = Groq()
-client.set_api_key(api_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")
 
-# 定義回應函數
-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})
-
-    # 獲取回應
-    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,
-    )
-
-    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
+client = Groq(api_key=groq_key)
 
-# 創建 Gradio Chatbot 界面
-with gr.Blocks() as demo:
-    chatbot = gr.Chatbot(type='messages')
-    msg = gr.Textbox(label="輸入訊息")
-    clear = gr.Button("清除聊天")
+class SimpleChatBot:
+    def __init__(self):
+        self.initial_prompt = [
+            {
+                "role": "system",
+                "content": "你是一個英文老師，會教我背單字，並用蘇格拉底問答引導我，引導時請用繁體中文 zhTW"
+            }
+        ]
 
-    def user(user_message, history):
-        return "", history + [ChatMessage(role="user", content=user_message)]
+    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
 
-    def bot(history):
-        user_message = history[-1]['content']
-        history = respond(user_message, history[:-1])
-        return history
+chatbot = SimpleChatBot()
 
-    msg.submit(user, [msg, chatbot], [msg, chatbot], queue=False).then(
-        bot, chatbot, chatbot
-    )
-    clear.click(lambda: [], None, chatbot, queue=False)
+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, ""
 
-demo.launch()
+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)