Update app.py

app.py

@@ -1,22 +1,45 @@
 import gradio as gr
-import random
+import os
+from groq import Groq
+
+# Set up the Groq client with the secret key
+groq_key = os.getenv('groq_key')
+client = Groq(api_key=groq_key)
 
 class SimpleChatBot:
     def __init__(self):
-        self.responses = {
-            "你好": "你好！有什麼我可以幫助你的嗎？",
-            "再見": "再見！希望很快能再見到你。",
-            "你叫什麼名字": "我是你的Gradio聊天機器人。",
-            "今天天氣如何": "我不太清楚，但你可以查一下當地的天氣預報。"
-        }
-    
-    def get_response(self, message):
-        return self.responses.get(message, "抱歉，我不太明白你的意思。")
+        self.initial_prompt = [
+            {
+                "role": "system",
+                "content": "你是一個老師"
+            }
+        ]
+
+    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="llama3-8b-8192",
+            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
 
 chatbot = SimpleChatBot()
 
 def respond(message, chat_history):
-    response = chatbot.get_response(message)
+    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, ""