Upload 6 files
Browse files- .env +1 -0
- .env.example +2 -0
- app.py +20 -0
- employees.csv +51 -0
- requirements.txt +0 -0
- utils.py +20 -0
.env
ADDED
@@ -0,0 +1 @@
|
|
|
|
|
1 |
+
OPENAI_API_KEY="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"
|
.env.example
ADDED
@@ -0,0 +1,2 @@
|
|
|
|
|
|
|
1 |
+
OPENAI_API_KEY=""
|
2 |
+
HUGGINGFACEHUB_API_TOKEN=""
|
app.py
ADDED
@@ -0,0 +1,20 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
import streamlit as st
|
2 |
+
from dotenv import load_dotenv
|
3 |
+
from utils import query_agent
|
4 |
+
|
5 |
+
load_dotenv()
|
6 |
+
|
7 |
+
|
8 |
+
st.title("Let's do some analysis on your CSV")
|
9 |
+
st.header("Please upload your CSV file here:")
|
10 |
+
|
11 |
+
# Capture the CSV file
|
12 |
+
data = st.file_uploader("Upload CSV file",type="csv")
|
13 |
+
|
14 |
+
query = st.text_area("Enter your query")
|
15 |
+
button = st.button("Generate Response")
|
16 |
+
|
17 |
+
if button:
|
18 |
+
# Get Response
|
19 |
+
answer = query_agent(data,query)
|
20 |
+
st.write(answer)
|
employees.csv
ADDED
@@ -0,0 +1,51 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
EMPLOYEE_ID,FIRST_NAME,LAST_NAME,EMAIL,PHONE_NUMBER,HIRE_DATE,JOB_ID,SALARY,COMMISSION_PCT,MANAGER_ID,DEPARTMENT_ID
|
2 |
+
198,Donald,OConnell,DOCONNEL,650.507.9833,21-Jun-07,SH_CLERK,2600, - ,124,50
|
3 |
+
199,Douglas,Grant,DGRANT,650.507.9844,13-Jan-08,SH_CLERK,2600, - ,124,50
|
4 |
+
200,Jennifer,Whalen,JWHALEN,515.123.4444,17-Sep-03,AD_ASST,4400, - ,101,10
|
5 |
+
201,Michael,Hartstein,MHARTSTE,515.123.5555,17-Feb-04,MK_MAN,13000, - ,100,20
|
6 |
+
202,Pat,Fay,PFAY,603.123.6666,17-Aug-05,MK_REP,6000, - ,201,20
|
7 |
+
203,Susan,Mavris,SMAVRIS,515.123.7777,07-Jun-02,HR_REP,6500, - ,101,40
|
8 |
+
204,Hermann,Baer,HBAER,515.123.8888,07-Jun-02,PR_REP,10000, - ,101,70
|
9 |
+
205,Shelley,Higgins,SHIGGINS,515.123.8080,07-Jun-02,AC_MGR,12008, - ,101,110
|
10 |
+
206,William,Gietz,WGIETZ,515.123.8181,07-Jun-02,AC_ACCOUNT,8300, - ,205,110
|
11 |
+
100,Steven,King,SKING,515.123.4567,17-Jun-03,AD_PRES,24000, - , - ,90
|
12 |
+
101,Neena,Kochhar,NKOCHHAR,515.123.4568,21-Sep-05,AD_VP,17000, - ,100,90
|
13 |
+
102,Lex,De Haan,LDEHAAN,515.123.4569,13-Jan-01,AD_VP,17000, - ,100,90
|
14 |
+
103,Alexander,Hunold,AHUNOLD,590.423.4567,03-Jan-06,IT_PROG,9000, - ,102,60
|
15 |
+
104,Bruce,Ernst,BERNST,590.423.4568,21-May-07,IT_PROG,6000, - ,103,60
|
16 |
+
105,David,Austin,DAUSTIN,590.423.4569,25-Jun-05,IT_PROG,4800, - ,103,60
|
17 |
+
106,Valli,Pataballa,VPATABAL,590.423.4560,05-Feb-06,IT_PROG,4800, - ,103,60
|
18 |
+
107,Diana,Lorentz,DLORENTZ,590.423.5567,07-Feb-07,IT_PROG,4200, - ,103,60
|
19 |
+
108,Nancy,Greenberg,NGREENBE,515.124.4569,17-Aug-02,FI_MGR,12008, - ,101,100
|
20 |
+
109,Daniel,Faviet,DFAVIET,515.124.4169,16-Aug-02,FI_ACCOUNT,9000, - ,108,100
|
21 |
+
110,John,Chen,JCHEN,515.124.4269,28-Sep-05,FI_ACCOUNT,8200, - ,108,100
|
22 |
+
111,Ismael,Sciarra,ISCIARRA,515.124.4369,30-Sep-05,FI_ACCOUNT,7700, - ,108,100
|
23 |
+
112,Jose Manuel,Urman,JMURMAN,515.124.4469,07-Mar-06,FI_ACCOUNT,7800, - ,108,100
|
24 |
+
113,Luis,Popp,LPOPP,515.124.4567,07-Dec-07,FI_ACCOUNT,6900, - ,108,100
|
25 |
+
114,Den,Raphaely,DRAPHEAL,515.127.4561,07-Dec-02,PU_MAN,11000, - ,100,30
|
26 |
+
115,Alexander,Khoo,AKHOO,515.127.4562,18-May-03,PU_CLERK,3100, - ,114,30
|
27 |
+
116,Shelli,Baida,SBAIDA,515.127.4563,24-Dec-05,PU_CLERK,2900, - ,114,30
|
28 |
+
117,Sigal,Tobias,STOBIAS,515.127.4564,24-Jul-05,PU_CLERK,2800, - ,114,30
|
29 |
+
118,Guy,Himuro,GHIMURO,515.127.4565,15-Nov-06,PU_CLERK,2600, - ,114,30
|
30 |
+
119,Karen,Colmenares,KCOLMENA,515.127.4566,10-Aug-07,PU_CLERK,2500, - ,114,30
|
31 |
+
120,Matthew,Weiss,MWEISS,650.123.1234,18-Jul-04,ST_MAN,8000, - ,100,50
|
32 |
+
121,Adam,Fripp,AFRIPP,650.123.2234,10-Apr-05,ST_MAN,8200, - ,100,50
|
33 |
+
122,Payam,Kaufling,PKAUFLIN,650.123.3234,01-May-03,ST_MAN,7900, - ,100,50
|
34 |
+
123,Shanta,Vollman,SVOLLMAN,650.123.4234,10-Oct-05,ST_MAN,6500, - ,100,50
|
35 |
+
124,Kevin,Mourgos,KMOURGOS,650.123.5234,16-Nov-07,ST_MAN,5800, - ,100,50
|
36 |
+
125,Julia,Nayer,JNAYER,650.124.1214,16-Jul-05,ST_CLERK,3200, - ,120,50
|
37 |
+
126,Irene,Mikkilineni,IMIKKILI,650.124.1224,28-Sep-06,ST_CLERK,2700, - ,120,50
|
38 |
+
127,James,Landry,JLANDRY,650.124.1334,14-Jan-07,ST_CLERK,2400, - ,120,50
|
39 |
+
128,Steven,Markle,SMARKLE,650.124.1434,08-Mar-08,ST_CLERK,2200, - ,120,50
|
40 |
+
129,Laura,Bissot,LBISSOT,650.124.5234,20-Aug-05,ST_CLERK,3300, - ,121,50
|
41 |
+
130,Mozhe,Atkinson,MATKINSO,650.124.6234,30-Oct-05,ST_CLERK,2800, - ,121,50
|
42 |
+
131,James,Marlow,JAMRLOW,650.124.7234,16-Feb-05,ST_CLERK,2500, - ,121,50
|
43 |
+
132,TJ,Olson,TJOLSON,650.124.8234,10-Apr-07,ST_CLERK,2100, - ,121,50
|
44 |
+
133,Jason,Mallin,JMALLIN,650.127.1934,14-Jun-04,ST_CLERK,3300, - ,122,50
|
45 |
+
134,Michael,Rogers,MROGERS,650.127.1834,26-Aug-06,ST_CLERK,2900, - ,122,50
|
46 |
+
135,Ki,Gee,KGEE,650.127.1734,12-Dec-07,ST_CLERK,2400, - ,122,50
|
47 |
+
136,Hazel,Philtanker,HPHILTAN,650.127.1634,06-Feb-08,ST_CLERK,2200, - ,122,50
|
48 |
+
137,Renske,Ladwig,RLADWIG,650.121.1234,14-Jul-03,ST_CLERK,3600, - ,123,50
|
49 |
+
138,Stephen,Stiles,SSTILES,650.121.2034,26-Oct-05,ST_CLERK,3200, - ,123,50
|
50 |
+
139,John,Seo,JSEO,650.121.2019,12-Feb-06,ST_CLERK,2700, - ,123,50
|
51 |
+
140,Joshua,Patel,JPATEL,650.121.1834,06-Apr-06,ST_CLERK,2500, - ,123,50
|
requirements.txt
ADDED
Binary file (332 Bytes). View file
|
|
utils.py
ADDED
@@ -0,0 +1,20 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
##from langchain.agents import create_pandas_dataframe_agent #This import has been recently replaced by the below
|
2 |
+
from langchain_experimental.agents import create_pandas_dataframe_agent
|
3 |
+
import pandas as pd
|
4 |
+
#from langchain.llms import OpenAI
|
5 |
+
#New import from langchain, which replaces the above
|
6 |
+
from langchain_openai import OpenAI
|
7 |
+
|
8 |
+
def query_agent(data, query):
|
9 |
+
|
10 |
+
# Parse the CSV file and create a Pandas DataFrame from its contents.
|
11 |
+
df = pd.read_csv(data)
|
12 |
+
|
13 |
+
llm = OpenAI()
|
14 |
+
|
15 |
+
# Create a Pandas DataFrame agent.
|
16 |
+
agent = create_pandas_dataframe_agent(llm, df, verbose=True)
|
17 |
+
|
18 |
+
#Python REPL: A Python shell used to evaluating and executing Python commands.
|
19 |
+
#It takes python code as input and outputs the result. The input python code can be generated from another tool in the LangChain
|
20 |
+
return agent.invoke(query)
|