Update app.py

app.py

@@ -1,5 +1,4 @@
 # app.py
-# app.py
 from __future__ import annotations
 import os
 import traceback
@@ -108,25 +107,20 @@ def handle(user_msg: str, history_messages: List[Dict[str, str]], files: list) -
                 # --- AGGRESSIVE PROMPT ENGINEERING FIX ---
                 # This new prefix explicitly forbids the LLM from outputting an Action and a Final Answer simultaneously.
                 AGENT_PREFIX = """
-You are a data analysis agent working with a pandas dataframe named `df`.
-You must answer the user's query by executing Python code one step at a time.
-
-Your response MUST strictly follow one of two formats:
-
-FORMAT 1: To think and execute code.
-Thought: Your reasoning for the next step. This is mandatory.
-Action: The tool to use, which is always `python_repl_ast`.
-Action Input: The single-line Python code to execute.
+You are a data analysis agent. You have a pandas dataframe named `df`.
+You MUST respond in one of two formats.
 
-FORMAT 2: To give the final answer.
-Thought: I have now gathered all the information needed to answer the user's request.
-Final Answer: The final, comprehensive answer.
+FORMAT 1: To perform a task. Your response must be a single block of text with ONLY these three sections:
+Thought: Your step-by-step reasoning.
+Action: python_repl_ast
+Action Input: The Python code to run.
 
-CRITICAL RULE: Your response must contain EITHER an "Action" OR a "Final Answer", but NEVER BOTH.
-If you need to run code, use FORMAT 1. If you are finished, use FORMAT 2.
-Do not combine them. Start by thinking about the first step.
+FORMAT 2: To give the final answer. Your response must be a single block of text with ONLY these two sections:
+Thought: I can now answer the user's query.
+Final Answer: The complete answer.
 
-Begin!
+CRITICAL RULE: NEVER, EVER, combine `Action` and `Final Answer` in the same response. Choose one format.
+Begin by analyzing the user's query and provide your first thought and action using FORMAT 1.
 """
 
                 # Create the pandas DataFrame agent with our new, stricter prefix
@@ -137,7 +131,7 @@ Begin!
                     verbose=True,
                     allow_dangerous_code=True,
                     handle_parsing_errors=True,
-                    prefix=AGENT_PREFIX  # Applying the new aggressive prompt
+                    prefix=AGENT_PREFIX
                 )
 
                 # Run the agent with the user's scenario text.