Update ai_agent.py

ai_agent.py

@@ -30,7 +30,15 @@ def sql_agent(file, agent_input):
     return agent_output
 
 def pandas_agent(file, agent_input):
-    df = pd.read_csv(file)
+    # Check if the file extension is CSV
+    if file.endswith('.csv'):
+        df = pd.read_csv(file)
+    # Check if the file extension is XLS or XLSX
+    elif file.endswith('.xls') or file.endswith('.xlsx'):
+        df = pd.read_excel(file)
+    else:
+        return "Unsupported file format. Only CSV, XLS, or XLSX files are supported."
+    # Proceed with your agent code
     agent = create_pandas_dataframe_agent(llm, df, agent_type="openai-tools", verbose=True)
     agent_output = agent.invoke(agent_input)
     return agent_output