robertselvam commited on
Commit
6df28e0
·
1 Parent(s): 3649991

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +11 -5
app.py CHANGED
@@ -133,11 +133,17 @@ class ChatDocumentQA:
133
  vectorstore = self._create_vector_store_from_text_chunks(text_chunks)
134
  return "file uploaded", {"knowledge_base": vectorstore}
135
  elif file_extension == '.csv':
136
- df = pd.read_csv(file_path)
137
- pd_agent = create_pandas_dataframe_agent(OpenAI(temperature=0), df, verbose=True)
138
- tools = self.get_agent_tools(pd_agent)
139
- memory,tools,prompt = self.create_memory_for_csv_qa(tools)
140
- agent_chain = self.create_agent_chain_for_csv_qa(memory,tools,prompt)
 
 
 
 
 
 
141
  return "file uploaded", {"knowledge_base": agent_chain}
142
 
143
  else:
 
133
  vectorstore = self._create_vector_store_from_text_chunks(text_chunks)
134
  return "file uploaded", {"knowledge_base": vectorstore}
135
  elif file_extension == '.csv':
136
+ # df = pd.read_csv(file_path)
137
+ # pd_agent = create_pandas_dataframe_agent(OpenAI(temperature=0), df, verbose=True)
138
+ # tools = self.get_agent_tools(pd_agent)
139
+ # memory,tools,prompt = self.create_memory_for_csv_qa(tools)
140
+ # agent_chain = self.create_agent_chain_for_csv_qa(memory,tools,prompt)
141
+ agent_chain = create_csv_agent(
142
+ OpenAI(temperature=0),
143
+ file_path,
144
+ verbose=True,
145
+ agent_type=AgentType.ZERO_SHOT_REACT_DESCRIPTION,
146
+ )
147
  return "file uploaded", {"knowledge_base": agent_chain}
148
 
149
  else: