Xelvis commited on
Commit
5e8f1ea
1 Parent(s): cc3ae3e

Upload csv_analyst.py

Browse files
Files changed (1) hide show
  1. csv_analyst.py +24 -0
csv_analyst.py ADDED
@@ -0,0 +1,24 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # CSV Analyst
2
+ from pandas import read_csv
3
+ from dotenv import load_dotenv
4
+ import streamlit as st
5
+ from langchain_openai import OpenAI
6
+ from langchain_experimental.agents import create_csv_agent
7
+
8
+ load_dotenv()
9
+
10
+ st.set_page_config(page_title='CSV Analyst')
11
+ st.header('Query your CSV file')
12
+
13
+ file = st.file_uploader('Upload your CSV File', type='csv')
14
+
15
+ if file is not None:
16
+ quest = st.text_input('Enter your Query', placeholder='Query')
17
+
18
+ llm = OpenAI(temperature=0, max_tokens=100)
19
+ agent = create_csv_agent(llm, file, verbose=True)
20
+
21
+ if quest:
22
+ with st.spinner('Processing'):
23
+ response = agent.run(quest)
24
+ st.write(response)