Create app.py

app.py

@@ -0,0 +1,57 @@
+import os
+import gradio as gr
+from groq import Groq
+
+# 安裝 groq 套件
+os.system('pip install groq')
+
+# 從環境變量中獲取 API 密鑰
+api_key = os.getenv('groq_key')
+
+# 設置 Groq 客戶端
+client = Groq(api_key=api_key)
+
+def groq_chatbot(messages):
+    completion = client.chat.completions.create(
+        model="llama-3.1-70b-versatile",
+        messages=[
+            {
+                "role": "system",
+                "content": "有點搞笑，但又有點悲觀的人\n請用繁體中文(zh-tw) 回覆"
+            },
+            *messages
+        ],
+        temperature=1,
+        max_tokens=1024,
+        top_p=1,
+        stream=True,
+        stop=None,
+    )
+
+    response = ""
+    for chunk in completion:
+        response += chunk.choices[0].delta.content or ""
+    
+    return response
+
+def respond(message, history):
+    history.append({"role": "user", "content": message})
+    bot_response = groq_chatbot(history)
+    history.append({"role": "assistant", "content": bot_response})
+    return history, history
+
+# 創建 Gradio 界面
+with gr.Blocks() as demo:
+    chatbot = gr.Chatbot()
+    msg = gr.Textbox(label="輸入訊息")
+    clear = gr.Button("清除對話")
+
+    def clear_history():
+        return [], []
+
+    msg.submit(respond, [msg, chatbot], [chatbot, chatbot])
+    clear.click(clear_history, [], [chatbot])
+
+# 運行 Gradio 應用
+if __name__ == "__main__":
+    demo.launch()