Update app.py

app.py

@@ -2,6 +2,11 @@ import gradio as gr
 import requests
 import json
 import uuid
+import logging
+
+# 配置日志
+logging.basicConfig(level=logging.DEBUG)
+logger = logging.getLogger(__name__)
 
 class FastGPTChat:
     def __init__(self, api_key, base_url="https://api.fastgpt.in/api"):
@@ -11,16 +16,18 @@ class FastGPTChat:
             "Authorization": f"Bearer {api_key}",
             "Content-Type": "application/json"
         }
-        
+    
     def chat(self, message, chat_history):
-        # 生成唯一的聊天ID
+        if not message.strip():
+            return "", chat_history
+            
         chat_id = str(uuid.uuid4())
         
         # 准备请求数据
         data = {
             "chatId": chat_id,
             "stream": False,
-            "detail": False,
+            "detail": True,  # 开启详细日志
             "messages": [
                 {
                     "role": "user",
@@ -30,24 +37,56 @@ class FastGPTChat:
         }
         
         try:
+            # 打印请求信息
+            logger.debug(f"Request URL: {self.base_url}/v1/chat/completions")
+            logger.debug(f"Request Headers: {json.dumps(self.headers, indent=2)}")
+            logger.debug(f"Request Data: {json.dumps(data, indent=2)}")
+            
             # 发送请求
             response = requests.post(
                 f"{self.base_url}/v1/chat/completions",
                 headers=self.headers,
-                json=data
+                json=data,
+                timeout=30  # 添加超时设置
             )
-            response.raise_for_status()
             
-            # 解析响应
-            response_data = response.json()
-            ai_message = response_data["choices"][0]["message"]["content"]
+            # 打印响应信息
+            logger.debug(f"Response Status Code: {response.status_code}")
+            logger.debug(f"Response Headers: {dict(response.headers)}")
+            logger.debug(f"Response Content: {response.text}")
             
-            # 更新对话历史
-            chat_history.append((message, ai_message))
-            return "", chat_history
+            # 检查响应状态
+            if response.status_code != 200:
+                error_msg = f"API Error: Status {response.status_code}, Response: {response.text}"
+                logger.error(error_msg)
+                chat_history.append((message, f"Error: {error_msg}"))
+                return "", chat_history
             
-        except Exception as e:
-            return f"Error: {str(e)}", chat_history
+            # 解析响应
+            try:
+                response_data = response.json()
+                logger.debug(f"Parsed Response: {json.dumps(response_data, indent=2)}")
+                
+                if "choices" in response_data and len(response_data["choices"]) > 0:
+                    ai_message = response_data["choices"][0]["message"]["content"]
+                else:
+                    ai_message = "No response generated from the API"
+                    logger.warning("No choices in response data")
+                
+                chat_history.append((message, ai_message))
+                return "", chat_history
+                
+            except json.JSONDecodeError as e:
+                error_msg = f"Failed to parse API response: {str(e)}"
+                logger.error(error_msg)
+                chat_history.append((message, f"Error: {error_msg}"))
+                return "", chat_history
+                
+        except requests.exceptions.RequestException as e:
+            error_msg = f"Request failed: {str(e)}"
+            logger.error(error_msg)
+            chat_history.append((message, f"Error: {error_msg}"))
+            return "", chat_history
 
 def create_chat_interface(api_key):
     # 创建FastGPT聊天实例
@@ -57,11 +96,39 @@ def create_chat_interface(api_key):
     with gr.Blocks(title="FastGPT Chat") as interface:
         gr.Markdown("# FastGPT Chat Interface")
         
+        with gr.Row():
+            api_key_input = gr.Textbox(
+                label="API Key",
+                value=api_key,
+                type="password"
+            )
+        
         chatbot = gr.Chatbot(height=400)
-        message = gr.Textbox(label="Type your message here...", placeholder="Enter your message and press enter")
-        clear = gr.Button("Clear Chat")
+        with gr.Row():
+            message = gr.Textbox(
+                label="Type your message here...",
+                placeholder="Enter your message and press enter",
+                lines=2
+            )
+        with gr.Row():
+            submit = gr.Button("Submit")
+            clear = gr.Button("Clear Chat")
+        
+        # 状态显示
+        status = gr.Textbox(label="Status", interactive=False)
+        
+        def update_api_key(new_key):
+            nonlocal fastgpt_chat
+            fastgpt_chat = FastGPTChat(new_key)
+            return "API Key updated"
         
         # 设置事件处理
+        submit_event = submit.click(
+            fastgpt_chat.chat,
+            inputs=[message, chatbot],
+            outputs=[message, chatbot]
+        )
+        
         message.submit(
             fastgpt_chat.chat,
             inputs=[message, chatbot],
@@ -69,6 +136,7 @@ def create_chat_interface(api_key):
         )
         
         clear.click(lambda: None, None, chatbot, queue=False)
+        api_key_input.change(update_api_key, inputs=[api_key_input], outputs=[status])
         
     return interface
 
@@ -76,4 +144,4 @@ def create_chat_interface(api_key):
 if __name__ == "__main__":
     API_KEY = "fastgpt-aleL83PzPiul9lZDJQFD4NHYfVIxP7mDIWFCyin2FuMhYgwlRC3NMkV2K5lxc8gF"  # 替换为你的API密钥
     demo = create_chat_interface(API_KEY)
-    demo.launch()
+    demo.launch(debug=True)  # 启用debug模式