Update app.py

app.py

@@ -32,21 +32,25 @@ def ai_response(message, chat_history):
     }
 
     # 发送请求到第三方 API
-    response = requests.post(API_BASE_URL, json=payload, headers=headers)
-    
-    if response.status_code == 200:
-        # 获取 API 响应内容
-        response_data = response.json()
-        assistant_message = response_data['choices'][0]['message']['content']
-        
-        # 返回新的聊天记录，转换为符合 gr.Chatbot 期望的元组格式
-        chat_history.append(("user", message))
-        chat_history.append(("assistant", assistant_message))
-        
-        return chat_history
-    else:
-        # 如果请求失败，返回错误消息
-        return chat_history + [("assistant", "Sorry, something went wrong with the API.")]
+    try:
+        response = requests.post(API_BASE_URL, json=payload, headers=headers)
+        response.raise_for_status()  # 如果响应状态码不是 2xx，会抛出异常
+        if response.status_code == 200:
+            # 获取 API 响应内容
+            response_data = response.json()
+            assistant_message = response_data['choices'][0]['message']['content']
+
+            # 返回新的聊天记录，转换为符合 gr.Chatbot 期望的元组格式
+            chat_history.append(("user", message))
+            chat_history.append(("assistant", assistant_message))
+
+            return chat_history
+        else:
+            # 如果请求失败，输出错误信息
+            return chat_history + [("assistant", f"API error: {response.status_code}, {response.text}")]
+    except requests.exceptions.RequestException as e:
+        # 捕获任何请求错误，并输出详细错误信息
+        return chat_history + [("assistant", f"Request failed: {str(e)}")]
 
 # 创建 Gradio 应用
 def create_interface():