v3

app.py

@@ -128,17 +128,21 @@ def run_crewai_process(user_query, model, temperature):
     final_answer_chat = [{"role": "user", "content": user_query}]
 
     try:
-        # Kick off the crew process
-        # CrewAI's kickoff doesn't directly support streaming to a buffer
-        # We'll run it and then capture the full output at the end of the CrewAI process
-        # However, for demonstration, we can yield intermediate status updates.
-        yield "Starting CrewAI process...", final_answer_chat, agent_thoughts, generated_code, execution_output, generated_plot_path
+        # Initial status update with proper message format
+        initial_message = {"role": "assistant", "content": "Starting CrewAI process..."}
+        final_answer_chat = [{"role": "user", "content": str(user_query)}, initial_message]
+        yield final_answer_chat, agent_thoughts, generated_code, execution_output, None, None
 
+        # Run the crew process
         final_result = crew.kickoff(inputs={"query": user_query})
 
         # Get the captured CrewAI output (agent thoughts)
         agent_thoughts = output_buffer.getvalue()
-        yield agent_thoughts, final_answer_chat, generated_code, execution_output, generated_plot_path
+        
+        # Update with processing message
+        processing_message = {"role": "assistant", "content": "Processing complete. Generating code..."}
+        final_answer_chat = [{"role": "user", "content": str(user_query)}, processing_message]
+        yield final_answer_chat, agent_thoughts, generated_code, execution_output, None, None
 
         # The final result is the generated code from the code_output_agent
         generated_code_raw = str(final_result).strip()
@@ -157,11 +161,9 @@ def run_crewai_process(user_query, model, temperature):
                 return  # Exit the generator
 
         # Format for Gradio Chatbot (list of dictionaries with 'role' and 'content' keys only)
-        final_answer_chat = [
-            {"role": "user", "content": str(user_query)},
-            {"role": "assistant", "content": "Code generation complete. See the 'Generated Code' box. Attempting to execute code..."}
-        ]
-        yield agent_thoughts, final_answer_chat, generated_code, execution_output, generated_plot_path
+        code_gen_message = {"role": "assistant", "content": "Code generation complete. See the 'Generated Code' box. Attempting to execute code..."}
+        final_answer_chat = [{"role": "user", "content": str(user_query)}, code_gen_message]
+        yield final_answer_chat, agent_thoughts, generated_code, execution_output, None, None
 
         # --- Execute the generated code ---
         plot_file_path = 'plot.png'  # Expected plot file name
@@ -212,14 +214,21 @@ def run_crewai_process(user_query, model, temperature):
             execution_output = "No code was generated to execute."
 
         # Update final answer chat to reflect execution attempt
-        final_answer_chat = [
-            {"role": "user", "content": str(user_query)},
-            {"role": "assistant", "content": "Code execution finished. See 'Execution Output'."}
-        ]
+        execution_complete_msg = "Code execution finished. See 'Execution Output'."
         if generated_plot_path:
-            final_answer_chat.append({"role": "assistant", "content": "Plot generated successfully. See 'Generated Plot'."})
+            plot_msg = "Plot generated successfully. See 'Generated Plot'."
+            final_answer_chat = [
+                {"role": "user", "content": str(user_query)},
+                {"role": "assistant", "content": execution_complete_msg},
+                {"role": "assistant", "content": plot_msg}
+            ]
         else:
-            final_answer_chat.append({"role": "assistant", "content": "No plot was generated. Check the execution output for details."})
+            no_plot_msg = "No plot was generated. Check the execution output for details."
+            final_answer_chat = [
+                {"role": "user", "content": str(user_query)},
+                {"role": "assistant", "content": execution_complete_msg},
+                {"role": "assistant", "content": no_plot_msg}
+            ]
 
         yield agent_thoughts, final_answer_chat, generated_code, execution_output, generated_plot_path
 
@@ -232,7 +241,7 @@ def run_crewai_process(user_query, model, temperature):
             {"role": "user", "content": str(user_query)},
             {"role": "assistant", "content": error_message}
         ]
-        yield agent_thoughts, final_answer_chat, generated_code, execution_output, generated_plot_path
+        yield final_answer_chat, agent_thoughts, generated_code, execution_output, None, None
 
     finally:
         # Restore original stdout