Update app.py

app.py

@@ -108,9 +108,10 @@ def handle(user_msg: str, history_messages: List[Dict[str, str]], files: list) -
                 agent = create_pandas_dataframe_agent(
                     llm,
                     df,
-                    agent_type=AgentType.ZERO_SHOT_REACT_DESCRIPTION, # Use a model-agnostic agent type
+                    agent_type=AgentType.ZERO_SHOT_REACT_DESCRIPTION,
                     verbose=True,
-                    allow_dangerous_code=True
+                    allow_dangerous_code=True,
+                    handle_parsing_errors=True  # Make the agent robust to LLM formatting errors
                 )
 
                 # Run the agent with the user's scenario text. The agent will
@@ -210,4 +211,4 @@ if __name__ == "__main__":
         "open_fallbacks": USE_OPEN_FALLBACKS,
         "timeout_s": COHERE_TIMEOUT_S
     })
-    demo.launch(server_name="0.0.0.0", server_port=int(os.getenv("PORT", "7860")))
+    demo.launch(server_name="0.0.0.0", server_port=int(os.getenv("PORT", "7860")))```