v2

app.py

@@ -156,9 +156,9 @@ def run_crewai_process(user_query, model, temperature):
                 yield agent_thoughts, final_answer_chat, generated_code, execution_output, generated_plot_path
                 return  # Exit the generator
 
-        # Format for Gradio Chatbot (list of dictionaries for type='messages')
+        # Format for Gradio Chatbot (list of dictionaries with 'role' and 'content' keys only)
         final_answer_chat = [
-            {"role": "user", "content": user_query},
+            {"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
@@ -213,11 +213,13 @@ def run_crewai_process(user_query, model, temperature):
 
         # Update final answer chat to reflect execution attempt
         final_answer_chat = [
-            {"role": "user", "content": user_query},
+            {"role": "user", "content": str(user_query)},
             {"role": "assistant", "content": "Code execution finished. See 'Execution Output'."}
         ]
         if generated_plot_path:
             final_answer_chat.append({"role": "assistant", "content": "Plot generated successfully. See 'Generated Plot'."})
+        else:
+            final_answer_chat.append({"role": "assistant", "content": "No plot was generated. Check the execution output for details."})
 
         yield agent_thoughts, final_answer_chat, generated_code, execution_output, generated_plot_path
 
@@ -225,8 +227,12 @@ def run_crewai_process(user_query, model, temperature):
         # If an error occurs during CrewAI process, return the error message
         traceback_str = traceback.format_exc()
         agent_thoughts += f"\nAn error occurred during CrewAI process: {e}\n{traceback_str}"
-        final_answer_chat.append({"role": "assistant", "content": f"An error occurred during CrewAI process: {e}"})
-        yield agent_thoughts, final_answer_chat, generated_code, execution_output, generated_plot_path  # Return empty list for chat on error
+        error_message = f"An error occurred during CrewAI process: {e}"
+        final_answer_chat = [
+            {"role": "user", "content": str(user_query)},
+            {"role": "assistant", "content": error_message}
+        ]
+        yield agent_thoughts, final_answer_chat, generated_code, execution_output, generated_plot_path
 
     finally:
         # Restore original stdout