Update app.py

app.py

@@ -281,7 +281,7 @@ def visualize_logprobs(json_input, chunk=0, chunk_size=100):
         logger.error("Visualization failed: %s (Input: %s)", str(e), json_input[:100] + "..." if len(json_input) > 100 else json_input)
         return (create_empty_figure("Log Probabilities of Generated Tokens"), None, "No finite log probabilities to display.", create_empty_figure("Top Token Log Probabilities"), create_empty_figure("Significant Probability Drops"), 1, 0)
 
-# Analysis functions for detecting correct vs. incorrect traces
+# Analysis functions for detecting correct vs. incorrect traces (unchanged from previous)
 def analyze_confidence_signature(logprobs, tokens):
     if not logprobs or not tokens:
         return "No data for confidence signature analysis.", None
@@ -573,7 +573,10 @@ with gr.Blocks(title="Log Probability Visualizer") as app:
 
             # Trigger precomputation when chunk changes (via button clicks or initial load)
             def trigger_precomputation(json_input, current_chunk):
-                asyncio.create_task(precompute_next_chunk(json_input, current_chunk, None))
+                try:
+                    asyncio.create_task(precompute_next_chunk(json_input, current_chunk, None))
+                except Exception as e:
+                    logger.error("Precomputation trigger failed: %s", str(e))
                 return gr.update(value=current_chunk)
 
             # Use a dummy event to trigger precomputation on chunk change (simplified for Gradio)