Update app.py

app.py

@@ -35,24 +35,36 @@ async def send_chat_message(LLM_URL, LLM_API, category, file_id):
             }
         ]
     }
-    async with aiohttp.ClientSession() as session:
-        async with session.post(
-            f"{LLM_URL}/chat-messages",
-            headers={"Authorization": f"Bearer {LLM_API}"},
-            json=payload
-        ) as response:
-            if response.status == 404:
-                return "Error: Endpoint not found (404)"
-            last_thought = None
-            async for line in response.content:
-                if line:
-                    try:
-                        data = json.loads(line.split(b"data: ")[1].decode("utf-8"))
-                        if data.get("event") == "agent_thought":
-                            last_thought = data.get("thought")
-                    except (IndexError, json.JSONDecodeError):
-                        continue
-            return last_thought.strip() if last_thought else "Error: No thought found in the response"
+
+    try:
+        async with aiohttp.ClientSession() as session:
+            async with session.post(
+                f"{LLM_URL}/chat-messages",
+                headers={"Authorization": f"Bearer {LLM_API}"},
+                json=payload
+            ) as response:
+
+                if response.status != 200:
+                    err = await response.text()
+                    return f"Error: Server returned status {response.status}: {err}"
+
+                last_thought = None
+                async for line in response.content:
+                    if line:
+                        try:
+                            if b"data: " in line:
+                                json_part = line.split(b"data: ")[1].strip()
+                                data = json.loads(json_part.decode("utf-8"))
+                                if data.get("event") == "agent_thought":
+                                    last_thought = data.get("thought")
+                        except Exception as e:
+                            print(f"[stream error] {e} line={line}")
+                            continue
+                return last_thought.strip() if last_thought else "Error: No thought found in response"
+    except aiohttp.ClientPayloadError as e:
+        return f"Error: Incomplete response stream ({str(e)})"
+    except Exception as e:
+        return f"Error: Unexpected exception - {str(e)}"
 
 async def upload_file(LLM_URL, LLM_API, file_path, user_id):
     if not os.path.exists(file_path):