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:
@@ -211,7 +211,7 @@ class ChatDocumentQA:
 
     def create_agent_chain_for_csv_qa(self,memory,tools,prompt):
 
-        llm_chain = LLMChain(llm=OpenAI(temperature=0), prompt=prompt)
+        llm_chain = LLMChain(llm=AzureChatOpenAI(azure_deployment = "ChatGPT"), prompt=prompt)
         agent = ZeroShotAgent(llm_chain=llm_chain, tools=tools, verbose=True)
         agent_chain = AgentExecutor.from_agent_and_tools(
             agent=agent, tools=tools, verbose=True, memory=memory