Update app.py

app.py

@@ -98,7 +98,7 @@ def handle(user_msg: str, history_messages: List[Dict[str, str]], files: list) -
         # --- NEW LOGIC: Activate data agent if files are uploaded ---
         if file_paths:
             try:
-                # For this example, we'll load the first CSV file.
+                # Load the first uploaded CSV into a pandas DataFrame.
                 df = pd.read_csv(file_paths[0])
 
                 # Initialize the Cohere Chat LLM for the agent
@@ -111,11 +111,10 @@ def handle(user_msg: str, history_messages: List[Dict[str, str]], files: list) -
                     agent_type=AgentType.ZERO_SHOT_REACT_DESCRIPTION,
                     verbose=True,
                     allow_dangerous_code=True,
-                    handle_parsing_errors=True  # Make the agent robust to LLM formatting errors
+                    handle_parsing_errors=True  # <-- THE FINAL FIX IS HERE
                 )
 
-                # Run the agent with the user's scenario text. The agent will
-                # write and execute code to answer the query based on the dataframe.
+                # Run the agent with the user's scenario text.
                 reply = agent.run(safe_in)
                 reply = _sanitize_text(reply)