Update app.py

app.py

@@ -130,19 +130,26 @@ async def send_chat_message(user_input):
                     return f"Error: {response.status}"
 
                 full_response = []
-                async for line in response.content:
-                    line = line.decode('utf-8').strip()
-                    if not line:
-                        continue
-                    if "data: " not in line:
-                        continue
-                    try:
-                        data = json.loads(line.split("data: ")[1])
-                        if "answer" in data:
-                            full_response.append(data["answer"])
-                    except (IndexError, json.JSONDecodeError) as e:
-                        print(f"Error parsing line: {line}, error: {e}")
+                async for chunk in response.content.iter_any():
+                    if len(chunk) > 10 * 1024:
+                        print("Chunk too big, skipping")
                         continue
+                    
+                    decoded = chunk.decode('utf-8', errors='ignore')
+                    lines = decoded.splitlines()
+
+                    for line in lines:
+                        line = line.strip()
+                        if not line.startswith("data: "):
+                            continue
+                        try:
+                            data_str = line.split("data: ")[1]
+                            data = json.loads(data_str)
+                            if "answer" in data:
+                                full_response.append(data["answer"])
+                        except (IndexError, json.JSONDecodeError) as e:
+                            print(f"Error parsing line: {line}, error: {e}")
+                            continue
 
                 if full_response:
                     return ''.join(full_response).strip()