iamshaik commited on
Commit
cb19108
·
verified ·
1 Parent(s): 90d21eb
Files changed (1) hide show
  1. app.py +36 -0
app.py ADDED
@@ -0,0 +1,36 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import streamlit as st
2
+ import pandas as pd
3
+ import pandasai
4
+ from pandasai.llm import OpenAI
5
+ import os
6
+ from pandasai import SmartDataframe
7
+
8
+ import os
9
+ from dotenv import load_dotenv
10
+
11
+ # Load environment variables from .env
12
+ load_dotenv()
13
+
14
+ st.image("/Users/shaikmujeeburrahman/Downloads/Odinschool.jpg")
15
+
16
+ st.title("Odinschool's Data Analysis App")
17
+ st.subheader("Created by Mujeeb")
18
+ file=st.file_uploader("Upload a CSV for data analysis")
19
+ if file is not None:
20
+ df=pd.read_csv(file)
21
+ st.write(df.head(5))
22
+
23
+
24
+ query=st.text_area("What do you want to know from the csv file?")
25
+
26
+
27
+ llm = OpenAI(api_token=os.getenv("API_KEY"))
28
+ df=SmartDataframe(df,config={"llm":llm})
29
+ if st.button("Fetch"):
30
+ if query:
31
+ with st.spinner("Fecthing info"):
32
+ st.write(df.chat(query))
33
+
34
+ else:
35
+ st.warning("please enter your query")
36
+