Update app.py

app.py

@@ -27,7 +27,8 @@ class FastGPTChat:
         data = {
             "chatId": chat_id,
             "stream": False,
-            "detail": True,  # 开启详细日志
+            "detail": True,
+            "model": "gpt-4",  # 指定使用 gpt-4 模型
             "messages": [
                 {
                     "role": "user",
@@ -47,7 +48,7 @@ class FastGPTChat:
                 f"{self.base_url}/v1/chat/completions",
                 headers=self.headers,
                 json=data,
-                timeout=30  # 添加超时设置
+                timeout=30
             )
             
             # 打印响应信息
@@ -67,8 +68,30 @@ class FastGPTChat:
                 response_data = response.json()
                 logger.debug(f"Parsed Response: {json.dumps(response_data, indent=2)}")
                 
+                # 检查是否有错误信息
+                if "responseData" in response_data:
+                    for item in response_data["responseData"]:
+                        if "error" in item:
+                            error_msg = item["error"].get("message", "Unknown error")
+                            logger.error(f"API Error: {error_msg}")
+                            chat_history.append((message, f"Error: {error_msg}"))
+                            return "", chat_history
+                
                 if "choices" in response_data and len(response_data["choices"]) > 0:
-                    ai_message = response_data["choices"][0]["message"]["content"]
+                    content = response_data["choices"][0]["message"]["content"]
+                    # 如果content是列表，提取文本内容
+                    if isinstance(content, list):
+                        text_contents = []
+                        for item in content:
+                            if isinstance(item, dict) and "text" in item:
+                                text_contents.append(item["text"].get("content", ""))
+                        ai_message = " ".join(text_contents)
+                    else:
+                        ai_message = content
+                    
+                    if not ai_message:
+                        ai_message = "No response generated from the API"
+                        logger.warning("Empty response content")
                 else:
                     ai_message = "No response generated from the API"
                     logger.warning("No choices in response data")
@@ -94,7 +117,7 @@ def create_chat_interface(api_key):
     
     # 创建Gradio界面
     with gr.Blocks(title="FastGPT Chat") as interface:
-        gr.Markdown("# FastGPT Chat Interface")
+        gr.Markdown("# FastGPT Chat Interface (Using GPT-4)")
         
         with gr.Row():
             api_key_input = gr.Textbox(
@@ -144,4 +167,4 @@ def create_chat_interface(api_key):
 if __name__ == "__main__":
     API_KEY = "fastgpt-aleL83PzPiul9lZDJQFD4NHYfVIxP7mDIWFCyin2FuMhYgwlRC3NMkV2K5lxc8gF"  # 替换为你的API密钥
     demo = create_chat_interface(API_KEY)
-    demo.launch(debug=True)  # 启用debug模式
+    demo.launch(debug=True)