Update app.py

app.py

@@ -289,7 +289,6 @@ class MaestroEngine:
             
         return f"Recalled {len(fetched_data)} entries from memory:\n\n{json.dumps(fetched_data, indent=2)}"
 
-
 # --- GRADIO APPLICATION ---
 class GradioApp:
     def __init__(self, engine: MaestroEngine):
@@ -330,42 +329,69 @@ class GradioApp:
                     recall_button = gr.Button("Recall", variant="primary")
                     memory_output = gr.Textbox(label="Recalled Information", lines=20, interactive=False)
 
-            process_button.click(self._synthesis_workflow, [task_instructions, text_input, file_upload, url_input, pdf_url_input, pdf_batch_input, summarize_check, report_check, memory_check], [final_report_output, json_summary_output])
+            # --- CORRECTION PART 1: The event listener now expects 4 outputs ---
+            # The output components match the error: [state, textbox, textbox, button]
+            # In our code, these are: session_id, final_report_output, json_summary_output, process_button
+            process_button.click(
+                self._synthesis_workflow, 
+                [session_id, task_instructions, text_input, file_upload, url_input, pdf_url_input, pdf_batch_input, summarize_check, report_check, memory_check], 
+                [session_id, final_report_output, json_summary_output, process_button]
+            )
             recall_button.click(self.engine.recall_from_memory, [memory_query], [memory_output])
         return app
 
-    def _synthesis_workflow(self, task, text, files, url, pdf_url, pdf_batch, do_sum, do_rep, do_mem):
-        log("Starting synthesis workflow...")
-        # 1. Ingest Data
+    # --- CORRECTION PART 2: The handler function is now a generator that yields updates ---
+    def _synthesis_workflow(self, session, task, text, files, url, pdf_url, pdf_batch, do_sum, do_rep, do_mem):
+        log(f"Starting synthesis workflow for session: {session}")
+        
+        # 1. First yield: Immediately update the UI to show a "processing" state.
+        #    This provides a value for all 4 output components.
+        yield {
+            session_id: session, # The state component doesn't need to be changed
+            final_report_output: "⚙️ Processing... Please wait.",
+            json_summary_output: None,
+            process_button: gr.update(value="Processing...", interactive=False)
+        }
+
+        # 2. Perform the actual work
         ingested_text, errors = self.engine.process_data_sources(text, files, url, pdf_url, pdf_batch)
         if errors:
             log(f"Ingestion errors: {errors}")
-            # For simplicity, we show errors in the log. A real app might have a dedicated error box.
-        
+
         if not ingested_text:
-            return "No data was successfully ingested. Please check your inputs and logs.", None
+            # Final yield (or return) in case of error
+            yield {
+                session_id: session,
+                final_report_output: "## Error\nNo data was successfully ingested. Please check your inputs.",
+                json_summary_output: {"errors": errors},
+                process_button: gr.update(value="🚀 Process & Synthesize", interactive=True)
+            }
+            return # Stop execution here
 
-        # 2. Save to Memory (if requested)
         if do_mem:
             self.engine.save_to_memory(ingested_text, task)
-            # We don't wait for this to finish for the UI, it's a background-like task
 
-        # 3. Summarize and Report
+        report_result, summaries_result = "Processing was not requested.", None
         if do_sum or do_rep:
-            report, summaries = self.engine.synthesis_workflow(ingested_text, task, do_sum, do_rep)
-            return report, summaries
+            report_result, summaries_result = self.engine.synthesis_workflow(ingested_text, task, do_sum, do_rep)
         
-        return "Processing complete. No synthesis option was selected.", None
+        # 3. Final yield: Return the final results and re-enable the button.
+        #    This also provides a value for all 4 output components.
+        yield {
+            session_id: session,
+            final_report_output: report_result,
+            json_summary_output: summaries_result,
+            process_button: gr.update(value="🚀 Process & Synthesize", interactive=True)
+        }
 
     def launch(self): self.app.launch(debug=Config.VERBOSE, share=False)
 
-if __name__ == "__main__":
+if __name__ == "__main__ ":
     if not Config.HF_TOKEN:
+
         print("FATAL: HF_TOKEN environment variable not set.")
     else:
-        log("Instantiating Maestro Engine...")
-        engine = MaestroEngine()
-        app = GradioApp(engine)
+        log("Instantiating Maestro Engine...") 
+        engine = MaestroEngine() app = GradioApp(engine)
         log("Launching Gradio App...")
-        app.launch()
-
+        app.launch()