Update app.py

app.py

@@ -134,11 +134,17 @@ class ChatDocumentQA:
             vectorstore = self._create_vector_store_from_text_chunks(text_chunks)
             return "file uploaded", {"knowledge_base": vectorstore}
         elif file_extension == '.csv':  
-            df = pd.read_csv(file_path)
-            pd_agent = create_pandas_dataframe_agent(OpenAI(temperature=0), df, verbose=True)
-            tools = self.get_agent_tools(pd_agent)
-            memory,tools,prompt = self.create_memory_for_csv_qa(tools)
-            agent_chain = self.create_agent_chain_for_csv_qa(memory,tools,prompt)
+            # df = pd.read_csv(file_path)
+            # pd_agent = create_pandas_dataframe_agent(OpenAI(temperature=0), df, verbose=True)
+            # tools = self.get_agent_tools(pd_agent)
+            # memory,tools,prompt = self.create_memory_for_csv_qa(tools)
+            # agent_chain = self.create_agent_chain_for_csv_qa(memory,tools,prompt)
+            agent_chain = create_csv_agent(
+                          OpenAI(temperature=0),
+                          file_path,
+                          verbose=True,
+                          agent_type=AgentType.ZERO_SHOT_REACT_DESCRIPTION,
+            )            
             return "file uploaded", {"knowledge_base": agent_chain}
             
         else: