Update app.py

app.py

@@ -135,11 +135,16 @@ class ChatDocumentQA:
             vectorstore = self._create_vector_store_from_text_chunks(text_chunks)
             return "file uploaded", {"knowledge_base": vectorstore}
         elif file_extension == '.csv':  
-            agent = self.create_agent(file_path)
-            tools = self.get_agent_tools(agent)
-            memory,tools,prompt = self.create_memory_for_csv_qa(tools)
-            agent_chain = self.create_agent_chain_for_csv_qa(memory,tools,prompt)
-
+            # agent = self.create_agent(file_path)
+            # tools = self.get_agent_tools(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: