Update app.py

app.py

@@ -1,7 +1,7 @@
 import os
 import gradio as gr
 
-# Install the groq package if it is not installed
+# 安裝 groq 套件如果未安裝
 try:
     from groq import Groq
 except ImportError:
@@ -16,42 +16,53 @@ if not groq_key:
 # 初始化 Groq 客戶端
 client = Groq(api_key=groq_key)
 
-def chatbot_response(history):
-    user_message = history[-1]["content"]
-    messages = [
-        {
-            "role": "system",
-            "content": "請你當一個玻璃心但是還願意溝通，脆弱中帶點身心韌性的可憐男性，聽到別人罵髒話會很害怕。（Zh-TW）"
-        },
-        {
-            "role": "user",
-            "content": user_message
-        }
-    ]
-
-    completion = client.chat.completions.create(
-        model="llama-3.1-70b-versatile",
-        messages=messages,
-        temperature=1,
-        max_tokens=1024,
-        top_p=1,
-        stream=False,
-        stop=None
-    )
-
-    response_message = completion.choices[0].message['content']
-    history.append({"role": "assistant", "content": response_message})
-    return history
-
-with gr.Blocks() as demo:
-    chatbot = gr.Chatbot()
-    user_input = gr.Textbox(placeholder="輸入訊息...")
-
-    def user_message_handler(user_message, history):
-        history.append({"role": "user", "content": user_message})
-        history = chatbot_response(history)
-        return history, ""
-
-    user_input.submit(user_message_handler, [user_input, chatbot], [chatbot, user_input])
-
-demo.launch()
+class SimpleChatBot:
+    def __init__(self):
+        self.initial_prompt = [
+            {
+                "role": "system",
+                "content": "請你當一個玻璃心但是還願意溝通，脆弱中帶點身心韌性的可憐男性，聽到別人罵髒話會很害怕。（Zh-TW）"
+            }
+        ]
+
+    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
+
+chatbot = SimpleChatBot()
+
+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, ""
+
+with gr.Blocks(title="玻璃心聊天機器人") as demo:
+    gr.Markdown("# 玻璃心聊天機器人")
+    
+    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)