Update app.py

app.py

@@ -59,7 +59,7 @@ def chat_with_fastgpt(message, history):
         "detail": False,
         "messages": messages,
     }
-
+    ####
     try:
         response = requests.post(
             f"{BASE_URL}/chat/completions", headers=headers, json=data, stream=True
@@ -72,20 +72,12 @@ def chat_with_fastgpt(message, history):
             if chunk:
                 full_response += chunk.decode('utf-8')
 
+        # Print the raw response for debugging
+        print(f"------ Raw API Response ------\n{full_response}\n-----------------------------")
+
         # 2. Parse the response as a whole
         try:
-            # 尝试解析整个响应为一个 JSON 对象
-            json_data = json.loads(full_response)
-            
-            # 3. Yield in chunks (assuming a simple structure for now)
-            if json_data.get("choices"):
-                for choice in json_data["choices"]:
-                    if choice.get("message") and choice["message"].get("content"):
-                        content = choice["message"]["content"]
-                        for i in range(0, len(content), 10):  # Yield chunks of 10 characters
-                            yield content[i:i+10]
-            else:
-                yield "Invalid response format: 'choices' key not found."
+            # ... (rest of the parsing and yielding logic)
 
         except json.JSONDecodeError:
             print(f"Failed to parse JSON: {full_response}")