Update app.py

app.py

@@ -1,9 +1,8 @@
-# app.py (Clean Version - Imports Visuals)
+# app.py (Fixed SSE Parsing for Large JSON Logs)
 import gradio as gr
 import httpx
 import os
 import json
-# Import the visualization logic from our new module
 from visuals import create_progress_chart, create_calibration_table
 
 # --- Config ---
@@ -46,20 +45,39 @@ def call_blaxel_backend(user_problem, google_key, anthropic_key, sambanova_key):
 
             final_json = None
             full_log = ""
-            
-            # Stream the text updates
+            buffer = ""
+
+            # Iterate over the stream
             for chunk in response.iter_text():
-                if chunk.startswith("FINAL:"):
-                    final_json = json.loads(chunk.replace("FINAL:", ""))
-                else:
-                    full_log += chunk + "\n"
-                    yield {status_output: full_log} 
+                buffer += chunk
+                
+                # Process complete SSE events (separated by double newline)
+                while "\n\n" in buffer:
+                    message, buffer = buffer.split("\n\n", 1)
+                    
+                    # Remove "data: " prefix if present
+                    if message.startswith("data: "):
+                        content = message.replace("data: ", "", 1)
+                    else:
+                        content = message
+
+                    # Check if this is the FINAL payload
+                    if content.startswith("FINAL:"):
+                        try:
+                            json_str = content.replace("FINAL:", "", 1)
+                            final_json = json.loads(json_str)
+                        except json.JSONDecodeError as e:
+                            full_log += f"\n[Error Parsing Final JSON]: {e}"
+                    else:
+                        # Standard log update
+                        full_log += content + "\n"
+                        yield {status_output: full_log}
 
-            # Once finished, process the JSON log to create visuals
+            # Once stream ends, check if we got the JSON and render visuals
             if final_json:
                 log_data = final_json.get("log")
                 
-                # Generate Visuals using the imported functions
+                # Generate Visuals
                 chart = create_progress_chart(log_data)
                 calib_table = create_calibration_table(log_data)