Update app.py

app.py

@@ -135,16 +135,16 @@ class ChatDocumentQA:
         file_extension = os.path.splitext(file_path)[1]
 
         if 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_chain = create_csv_agent(
-            #               AzureOpenAI(azure_deployment = "ChatGPT"),
-            #               file_path,
-            #               verbose=True,
-            #               agent_type=AgentType.ZERO_SHOT_REACT_DESCRIPTION,
-            # )
+            # 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(
+                          AzureOpenAI(azure_deployment = "ChatGPT"),
+                          file_path,
+                          verbose=True,
+                          agent_type=AgentType.ZERO_SHOT_REACT_DESCRIPTION,
+            )
             return "file uploaded", {"knowledge_base": agent_chain}
             
         else: