Abhilashvj commited on
Commit
b5db58b
1 Parent(s): 0391a40

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +43 -0
app.py CHANGED
@@ -0,0 +1,43 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import streamlit as st
2
+ import requests
3
+
4
+ st.title("Document Uploader and Chatbot Trainer")
5
+
6
+ st.header("Upload Documents")
7
+
8
+ # Upload evergreen document
9
+ evergreen_file = st.file_uploader("Choose an Evergreen Document", type=['txt', 'pdf', 'doc', 'docx'])
10
+ if evergreen_file:
11
+ files = {'file': evergreen_file.getvalue()}
12
+ response = requests.post('http://your_fastapi_endpoint/upload/evergreen/', files=files)
13
+ if response.json().get("success"):
14
+ st.success("Evergreen document uploaded successfully!")
15
+ else:
16
+ st.error("Failed to upload evergreen document!")
17
+
18
+ # Upload dynamic document
19
+ dynamic_file = st.file_uploader("Choose a Dynamic Document", type=['txt', 'pdf', 'doc', 'docx'])
20
+ if dynamic_file:
21
+ files = {'file': dynamic_file.getvalue()}
22
+ response = requests.post('http://your_fastapi_endpoint/upload/dynamic/', files=files)
23
+ if response.json().get("success"):
24
+ st.success("Dynamic document uploaded successfully!")
25
+ else:
26
+ st.error("Failed to upload dynamic document!")
27
+
28
+ # Train bot button
29
+ if st.button("Train Bot"):
30
+ response = requests.post('http://your_fastapi_endpoint/train/')
31
+ bot_url = response.json().get("bot_url")
32
+ if bot_url:
33
+ st.success(f"Bot trained successfully! Access the bot at {bot_url}")
34
+ else:
35
+ st.error("Failed to train the bot!")
36
+
37
+ # Chat with bot
38
+ st.header("Chat with Bot")
39
+ user_input = st.text_input("Ask your question:")
40
+ # Assuming you have an endpoint to send user questions and get responses
41
+ # response = requests.post('http://your_fastapi_endpoint/ask/', json={"question": user_input})
42
+ # bot_response = response.json().get("answer")
43
+ # st.text_area("Bot's Response:", value=bot_response)