Update app.py

app.py

@@ -135,17 +135,11 @@ 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)
-            agent_chain = create_csv_agent(
-            OpenAI(temperature=0),
-            file_path,
-            verbose=True,
-            agent_type=AgentType.ZERO_SHOT_REACT_DESCRIPTION,
-            )
+            agent = 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)
+
             return "file uploaded", {"knowledge_base": agent_chain}
             
         else:
@@ -168,6 +162,14 @@ class ChatDocumentQA:
 #************************
 #   csv qa
 #************************
+    def create_agent(self,file_path):
+        agent_chain = create_csv_agent(
+        OpenAI(temperature=0),
+        file_path,
+        verbose=True,
+        agent_type=AgentType.ZERO_SHOT_REACT_DESCRIPTION,
+        )
+        return agent_chain
     def get_agent_tools(self,agent):
       # search = agent
       tools = [