v2

app.py

@@ -157,7 +157,10 @@ def run_crewai_process(user_query, model, temperature):
                 return  # Exit the generator
 
         # Format for Gradio Chatbot (list of dictionaries for type='messages')
-        final_answer_chat.append({"role": "assistant", "content": "Code generation complete. See the 'Generated Code' box. Attempting to execute code..."})
+        final_answer_chat = [
+            {"role": "user", "content": 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
 
         # --- Execute the generated code ---
@@ -209,9 +212,12 @@ 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.append({"role": "assistant", "content": f"Code execution finished. See 'Execution Output'."})
+        final_answer_chat = [
+            {"role": "user", "content": user_query},
+            {"role": "assistant", "content": "Code execution finished. See 'Execution Output'."}
+        ]
         if generated_plot_path:
-            final_answer_chat.append({"role": "assistant", "content": f"Plot generated successfully. See 'Generated Plot'."})
+            final_answer_chat.append({"role": "assistant", "content": "Plot generated successfully. See 'Generated Plot'."})
 
         yield agent_thoughts, final_answer_chat, generated_code, execution_output, generated_plot_path
 
@@ -263,7 +269,8 @@ def create_interface():
                         final_answer_chat = gr.Chatbot(
                             label="Analysis Results",
                             height=300,
-                            show_copy_button=True
+                            show_copy_button=True,
+                            type="messages"  # Explicitly set to use OpenAI-style message format
                         )
                     
                     with gr.TabItem("Agent Thoughts"):