Update app.py

app.py

@@ -33,7 +33,6 @@ API_KEY = "fastgpt-aleL83PzPiul9lZDJQFD4NHYfVIxP7mDIWFCyin2FuMhYgwlRC3NMkV2K5lxc
 import ijson  # Install with: pip install ijson
 
 
-
 def chat_with_fastgpt(message, history):
     """
     与 FastGPT 进行对话的函数
@@ -70,28 +69,37 @@ def chat_with_fastgpt(message, history):
         )
         response.raise_for_status()
 
-        # Use ijson to parse the stream
-        parser = ijson.parse(response.raw)  # Use response.raw for the raw byte stream
-        current_content = ""
-        for prefix, event, value in parser:
-            if prefix == "choices.item.message.content" and event == "string":
-                current_content += value
-            elif current_content:
-                 # Yield the accumulated content 
-                for i in range(0, len(current_content), 10):
-                    yield current_content[i:i + 10]
-                current_content = ""
-        
-        # Yield any remaining content
-        if current_content:
-            for i in range(0, len(current_content), 10):
-                yield current_content[i:i + 10]
+        buffer = ""
+        for chunk in response.iter_content(chunk_size=8192):
+            if chunk:
+                buffer += chunk.decode('utf-8')
+                
+                # 尝试解析 JSON 对象
+                while True:  # Continue processing the buffer until no more complete objects are found
+                    try:
+                        first_brace_index = buffer.find('{')
+                        if first_brace_index == -1:
+                            break
+                        
+                        obj, end_index = json.JSONDecoder().raw_decode(buffer[first_brace_index:])
+                        
+                        # 解析成功，处理 JSON 对象
+                        if obj.get("choices"):
+                            delta = obj["choices"][0].get("delta")
+                            if delta and delta.get("content"):
+                                yield delta["content"]
+                        
+                        buffer = buffer[first_brace_index + end_index:]
+                    
+                    except json.JSONDecodeError as e:
+                        break
 
     except requests.exceptions.RequestException as e:
         print(f"请求错误: {e}")
         yield "请求 FastGPT API 出错，请检查网络或 API Key。"
     except Exception as e:
         print(f"未知错误: {e}")
+        print(f"错误信息: {e}")
         yield "发生未知错误。"
 
 # Gradio 界面