Spaces:
Sleeping
Sleeping
Commit
·
7888633
1
Parent(s):
9e4c9f3
updated pom file
Browse files- EcommerceDataset.csv +0 -0
- app.py +29 -8
EcommerceDataset.csv
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
app.py
CHANGED
|
@@ -29,9 +29,18 @@ genre = st.radio(
|
|
| 29 |
"Choose dataset to finetune", radioButtonList
|
| 30 |
)
|
| 31 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 32 |
pdfCSVURLText = ""
|
| 33 |
if genre==radioButtonList[0]:
|
| 34 |
pdfCSVURLText = "CSV"
|
|
|
|
| 35 |
# st.write('You selected comedy.')
|
| 36 |
# else:
|
| 37 |
# st.write(f'''Password streamlit app: {st.secrets["PSWD"]}''')
|
|
@@ -56,20 +65,32 @@ if uploaded_file is not None:
|
|
| 56 |
uploadedFilename = f.name
|
| 57 |
|
| 58 |
enableChatBox = False
|
| 59 |
-
if genre==radioButtonList[0]:
|
| 60 |
enableChatBox = True
|
| 61 |
-
elif genre==radioButtonList[1]:
|
| 62 |
enableChatBox = uploadedFilename[-4:]==".csv"
|
| 63 |
-
elif genre==radioButtonList[2]:
|
| 64 |
enableChatBox = uploadedFilename[-4:]==".pdf"
|
| 65 |
-
elif genre==radioButtonList[3]:
|
| 66 |
enableChatBox = True
|
| 67 |
-
elif genre==radioButtonList[4]:
|
| 68 |
enableChatBox = True
|
| 69 |
|
| 70 |
chatTextStr = st.text_input(f'Ask me anything about this {pdfCSVURLText}', '', placeholder="Type your question here (e.g. what was the most sold item?)", disabled=not enableChatBox)
|
| 71 |
chatWithPDFButton = "CLICK HERE TO START CHATTING"
|
| 72 |
if st.button(chatWithPDFButton, disabled=not enableChatBox and not chatTextStr): # Button Cliked
|
| 73 |
-
|
| 74 |
-
|
| 75 |
-
#
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 29 |
"Choose dataset to finetune", radioButtonList
|
| 30 |
)
|
| 31 |
|
| 32 |
+
# Initialize language model
|
| 33 |
+
load_dotenv(find_dotenv()) # read local .env file
|
| 34 |
+
api_key = st.secrets["PALM_API_KEY"] # put your API key here
|
| 35 |
+
os.environ["GOOGLE_API_KEY"] = st.secrets["PALM_API_KEY"]
|
| 36 |
+
palm.configure(api_key=api_key)
|
| 37 |
+
llm = GooglePalm()
|
| 38 |
+
llm.temperature = 0.1
|
| 39 |
+
|
| 40 |
pdfCSVURLText = ""
|
| 41 |
if genre==radioButtonList[0]:
|
| 42 |
pdfCSVURLText = "CSV"
|
| 43 |
+
dataDF = pd.read_csv('EcommerceDataset.csv', encoding= 'unicode_escape')
|
| 44 |
# st.write('You selected comedy.')
|
| 45 |
# else:
|
| 46 |
# st.write(f'''Password streamlit app: {st.secrets["PSWD"]}''')
|
|
|
|
| 65 |
uploadedFilename = f.name
|
| 66 |
|
| 67 |
enableChatBox = False
|
| 68 |
+
if genre==radioButtonList[0]: # E-commerce CSV
|
| 69 |
enableChatBox = True
|
| 70 |
+
elif genre==radioButtonList[1]: # Custom CSV Upload
|
| 71 |
enableChatBox = uploadedFilename[-4:]==".csv"
|
| 72 |
+
elif genre==radioButtonList[2]: # Custom PDF Upload
|
| 73 |
enableChatBox = uploadedFilename[-4:]==".pdf"
|
| 74 |
+
elif genre==radioButtonList[3]: # Google Alphabet URL Earnings Report
|
| 75 |
enableChatBox = True
|
| 76 |
+
elif genre==radioButtonList[4]: # Custom URL
|
| 77 |
enableChatBox = True
|
| 78 |
|
| 79 |
chatTextStr = st.text_input(f'Ask me anything about this {pdfCSVURLText}', '', placeholder="Type your question here (e.g. what was the most sold item?)", disabled=not enableChatBox)
|
| 80 |
chatWithPDFButton = "CLICK HERE TO START CHATTING"
|
| 81 |
if st.button(chatWithPDFButton, disabled=not enableChatBox and not chatTextStr): # Button Cliked
|
| 82 |
+
|
| 83 |
+
|
| 84 |
+
if genre==radioButtonList[0]: # E-commerce CSV
|
| 85 |
+
# Initializing the agent
|
| 86 |
+
agent = create_pandas_dataframe_agent(llm, dataDF, verbose=False)
|
| 87 |
+
answer = agent.run(chatTextStr)
|
| 88 |
+
st.write(answer)
|
| 89 |
+
|
| 90 |
+
elif genre==radioButtonList[1]: # Custom CSV Upload
|
| 91 |
+
|
| 92 |
+
elif genre==radioButtonList[2]: # Custom PDF Upload
|
| 93 |
+
|
| 94 |
+
elif genre==radioButtonList[3]: # Google Alphabet URL Earnings Report
|
| 95 |
+
|
| 96 |
+
elif genre==radioButtonList[4]: # Custom URL
|