ratulsur commited on
Commit
3d43739
·
verified ·
1 Parent(s): 40e7864

Upload 6 files

Browse files
Files changed (6) hide show
  1. .env +1 -0
  2. .env.example +2 -0
  3. app.py +20 -0
  4. employees.csv +51 -0
  5. requirements.txt +0 -0
  6. 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)