Update app.py

app.py

@@ -0,0 +1,61 @@
+import os
+import gradio as gr
+
+# 手动设置 API 金钥
+os.environ["groq_key"] = "your_groq_api_key"
+
+# 确保 groq 套件已安装
+try:
+    from groq import Groq
+except ImportError:
+    os.system('pip install groq')
+    from groq import Groq
+
+# 从环境变量中获取 API 金钥
+api_key = os.getenv("groq_key")
+if api_key is None:
+    raise ValueError("API key is not set. Please set it in your environment variables with the name 'groq_key'.")
+
+# 初始化 Groq 客户端
+client = Groq(api_key=api_key)
+
+# 定义聊天机器人的响应函数
+def chatbot_response(messages):
+    completion = client.chat.completions.create(
+        model="llama3-8b-8192",
+        messages=[
+            {
+                "role": "system",
+                "content": "You are a corporate secretary who is skilled at drafting business emails. The prompt will feed you addressee, main message, and final greetings."
+            }
+        ] + 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
+
+# 使用 gradio 创建聊天机器人界面
+def respond(message, history):
+    history.append({"role": "user", "content": message})
+    bot_response = chatbot_response(history)
+    history.append({"role": "assistant", "content": bot_response})
+    return "", history
+
+# 初始化 Gradio 应用
+with gr.Blocks() as demo:
+    chatbot = gr.Chatbot()
+    msg = gr.Textbox()
+    
+    msg.submit(respond, [msg, chatbot], [msg, chatbot])
+
+# 运行应用程序
+if __name__ == "__main__":
+    demo.launch(server_name="0.0.0.0", server_port=7860)