Update app.py

app.py

@@ -158,11 +158,16 @@ def messages():
 
             def on_query_progress(data):
                 nonlocal response_text
-                if 'text' in data:
-                    text = json.loads(data['text'])
-                    chunk = text['chunks'][-1] if text['chunks'] else None
-                    if chunk:
-                        response_text.append(chunk)
+                try:
+                    if 'text' in data:
+                        text = json.loads(data['text'])  # 尝试解析为JSON
+                        chunk = text['chunks'][-1] if text['chunks'] else None
+                        if chunk:
+                            response_text.append(chunk)
+                except json.JSONDecodeError:
+                    # 如果解析失败，说明不是有效的JSON，忽略或记录错误
+                    logging.error("Failed to parse response as JSON")
+                    response_text.append("Invalid response format from API.")
 
                 # 检查是否是最终响应
                 if data.get('final', False):
@@ -253,11 +258,16 @@ def handle_non_stream(previous_messages, msg_id, model):
 
         def on_query_progress(data):
             nonlocal response_text
-            if 'text' in data:
-                text = json.loads(data['text'])
-                chunk = text['chunks'][-1] if text['chunks'] else None
-                if chunk:
-                    response_text.append(chunk)
+            try:
+                if 'text' in data:
+                    text = json.loads(data['text'])  # 尝试解析为JSON
+                    chunk = text['chunks'][-1] if text['chunks'] else None
+                    if chunk:
+                        response_text.append(chunk)
+            except json.JSONDecodeError:
+                # 如果解析失败，说明不是有效的JSON，忽略或记录错误
+                logging.error("Failed to parse response as JSON")
+                response_text.append("Invalid response format from API.")
 
             # 检查是否是最终响应
             if data.get('final', False):