Update app.py

app.py

@@ -99,7 +99,6 @@ def handle(user_msg: str, history_messages: List[Dict[str, str]], files: list) -
         if file_paths:
             try:
                 # For this example, we'll load the first CSV file.
-                # This can be extended to handle multiple DataFrames.
                 df = pd.read_csv(file_paths[0])
 
                 # Initialize the Cohere Chat LLM for the agent
@@ -109,9 +108,9 @@ def handle(user_msg: str, history_messages: List[Dict[str, str]], files: list) -
                 agent = create_pandas_dataframe_agent(
                     llm,
                     df,
-                    agent_type=AgentType.OPENAI_FUNCTIONS,
+                    agent_type=AgentType.ZERO_SHOT_REACT_DESCRIPTION, # Use a model-agnostic agent type
                     verbose=True,
-                    allow_dangerous_code=True  # <-- THE FIX IS HERE
+                    allow_dangerous_code=True
                 )
 
                 # Run the agent with the user's scenario text. The agent will
@@ -193,7 +192,7 @@ with gr.Blocks(analytics_enabled=False) as demo:
     ping_out = gr.Markdown()
 
     def _on_send(m, h, f):
-        h2, _ = handle(m, h or [], f or [])
+        h2, _ = handle(m, h, f or [])
         return h2, ""
 
     send.click(_on_send, inputs=[msg, chat, files], outputs=[chat, msg])