Update app.py

app.py

@@ -1,21 +1,54 @@
+import os
 import gradio as gr
-import random
+from groq import Groq
 
-# 定義抽籤函數
-def draw_lottery(max_number):
-    if max_number <= 0:
-        return "請輸入一個正整數"
-    return random.randint(1, max_number)
+# 安裝 groq 套件
+os.system('pip install groq')
 
-# 創建 Gradio 介面
-with gr.Blocks() as demo:
-    with gr.Row():
-        max_number_input = gr.Number(label="班級座號的最大值", value=40, precision=0)
-        result_output = gr.Textbox(label="抽中的號碼")
+# 從環境變數獲取 API 密鑰
+groq_key = os.getenv('groq_key')
+
+# 定義 Chatbot 回應函數
+def chatbot_response(user_message):
+    client = Groq(api_key=groq_key)
+    completion = client.chat.completions.create(
+        model="llama-3.1-70b-versatile",
+        messages=[
+            {
+                "role": "system",
+                "content": "brat"
+            },
+            {
+                "role": "user",
+                "content": user_message
+            }
+        ],
+        temperature=1,
+        max_tokens=1024,
+        top_p=1,
+        stream=True,
+        stop=None,
+    )
     
-    draw_button = gr.Button("抽籤")
+    response = ""
+    for chunk in completion:
+        response += chunk.choices[0].delta.content or ""
     
-    draw_button.click(draw_lottery, inputs=max_number_input, outputs=result_output)
+    return response
+
+# 創建 Gradio 介面
+with gr.Blocks() as demo:
+    chatbot = gr.Chatbot()
+    msg = gr.Textbox()
+    clear = gr.Button("Clear")
+
+    def respond(message, chat_history):
+        bot_message = chatbot_response(message)
+        chat_history.append((message, bot_message))
+        return chat_history, ""
+
+    msg.submit(respond, [msg, chatbot], [chatbot, msg])
+    clear.click(lambda: None, None, chatbot, queue=False)
 
 # 啟動介面
 demo.launch()