Update app.py

app.py

@@ -21,33 +21,41 @@ 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
+    try:
+        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
+    except Exception as e:
+        print(f"Error in chatbot_response: {e}")
+        return "Error: Unable to get response from the API."
 
 # 使用 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
+    try:
+        history.append({"role": "user", "content": message})
+        bot_response = chatbot_response(history)
+        history.append({"role": "assistant", "content": bot_response})
+        return "", history
+    except Exception as e:
+        print(f"Error in respond function: {e}")
+        return "", history
 
 # 初始化 Gradio 应用
 with gr.Blocks() as demo: