v114

app.py

@@ -399,24 +399,36 @@ def create_interface():
             agent_thoughts,
             generated_code,
             execution_output,
-            None,  # This is for plot_output (we'll handle it separately)
-            None   # This is for image_output (we'll handle it separately)
+            plot_output,  # Pass the actual component, not None
+            image_output  # Pass the actual component, not None
         ]
         
         def process_results(chat, thoughts, code, output, plot_path):
             # This function will be called after run_crewai_process
             # Show the image in the image_output component
-            return chat, thoughts, code, output, gr.update(visible=plot_path is not None), gr.update(value=plot_path, visible=plot_path is not None)
+            return [
+                chat, 
+                thoughts, 
+                code, 
+                output, 
+                gr.update(visible=plot_path is not None and os.path.exists(plot_path)), 
+                gr.update(value=plot_path if (plot_path and os.path.exists(plot_path)) else None, 
+                        visible=plot_path is not None and os.path.exists(plot_path))
+            ]
         
-        submit_btn.click(
+        # First, run the crewAI process
+        click_event = submit_btn.click(
             fn=run_crewai_process,
             inputs=inputs,
             outputs=outputs,
             api_name="analyze"
-        ).then(
+        )
+        
+        # Then update the UI with the results
+        click_event.then(
             fn=process_results,
             inputs=[final_answer_chat, agent_thoughts, generated_code, execution_output, image_output],
-            outputs=[final_answer_chat, agent_thoughts, generated_code, execution_output, plot_output, image_output]
+            outputs=outputs
         )
     
     return interface