this better work

app.py

@@ -303,6 +303,108 @@ def validate_passcode(input_passcode):
             gr.update(visible=False),
             "❌ Invalid passcode. Please try again."
         )
+# batched streaming instead
+def process_agent_response_batched(prompt, uploaded_file, chatbot_history, batch_size=5):
+    """Process agent response with BATCHED updates (every N steps)."""
+    global agent
+    
+    if agent is None:
+        chatbot_history.append({
+            "role": "assistant",
+            "content": "⚠️ Please enter the passcode first to initialize the agent."
+        })
+        yield chatbot_history, None, None, None, None, "⚠️ Agent not initialized"
+        return
+    
+    if not prompt.strip() and uploaded_file is None:
+        chatbot_history.append({
+            "role": "assistant",
+            "content": "⚠️ Please provide a prompt or upload a file."
+        })
+        yield chatbot_history, None, None, None, None, "⚠️ No input provided"
+        return
+    
+    # Handle file upload
+    file_path = None
+    file_info = ""
+    if uploaded_file is not None:
+        try:
+            data_dir = Path("./data")
+            data_dir.mkdir(exist_ok=True)
+            
+            file_name = Path(uploaded_file.name).name
+            file_path = data_dir / file_name
+            shutil.copy(uploaded_file.name, file_path)
+            
+            file_info = f"\n\n📎 **Uploaded file:** `{file_path}`\n"
+            
+            if prompt.strip():
+                prompt = f"{prompt}\n\nUploaded file path: {file_path}"
+            else:
+                prompt = f"I have uploaded a file at: {file_path}. Please analyze it."
+                
+        except Exception as e:
+            error_msg = f"❌ Error handling file upload: {str(e)}"
+            chatbot_history.append({
+                "role": "assistant",
+                "content": error_msg
+            })
+            yield chatbot_history, None, None, None, None, error_msg
+            return
+    
+    # Add user message to chat
+    user_message = prompt if not file_info else f"{prompt}{file_info}"
+    chatbot_history.append({"role": "user", "content": user_message})
+    yield chatbot_history, None, None, None, None, "🔄 Processing..."
+    
+    try:
+        # Stream with batching
+        step_count = 0
+        batch_count = 0
+        
+        for step in agent.go_stream(prompt):
+            step_count += 1
+            output = step.get("output", "")
+            
+            if output:
+                parsed = parse_agent_output(output)
+                formatted_message = format_message_for_display(parsed)
+                
+                # Update chatbot history
+                if chatbot_history and chatbot_history[-1]["role"] == "assistant":
+                    chatbot_history[-1]["content"] = formatted_message
+                else:
+                    chatbot_history.append({
+                        "role": "assistant",
+                        "content": formatted_message
+                    })
+                
+                # Only yield every batch_size steps
+                if step_count % batch_size == 0:
+                    batch_count += 1
+                    yield chatbot_history, None, None, None, None, f"🔄 Step {step_count}..."
+        
+        # Final yield with files
+        images, data = check_for_output_files()
+        
+        status_msg = f"✅ Complete ({step_count} steps)"
+        if images:
+            status_msg += f" | {len(images)} image(s)"
+        if data:
+            status_msg += f" | {len(data)} data file(s)"
+        
+        yield chatbot_history, images, data, None, None, status_msg
+        
+    except Exception as e:
+        error_trace = traceback.format_exc()
+        error_msg = f"❌ **Error:** {str(e)}\n\n<details>\n<summary>Stack Trace</summary>\n\n```\n{error_trace}\n```\n</details>"
+        
+        chatbot_history.append({
+            "role": "assistant",
+            "content": error_msg
+        })
+        
+        yield chatbot_history, None, None, None, None, f"❌ Error: {str(e)}"
 
 
 # Custom theme
@@ -500,7 +602,7 @@ with gr.Blocks(title="HistoPath Agent", theme=custom_theme, css="""
     )
     
     submit_btn.click(
-        fn=process_agent_response,
+        fn=process_agent_response_batched,
         inputs=[prompt_input, file_upload, chatbot],
         outputs=[chatbot, output_gallery, data_files, input_image_preview, input_file_preview, status_text]
     )
@@ -530,4 +632,4 @@ if __name__ == "__main__":
     print("=" * 60)
     
     # Launch the app
-    demo.launch()
+    demo.launch(show_api=False)