File size: 2,630 Bytes
979518f |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 |
import streamlit as st
import requests
import utils
# Custom CSS
with open('styles.css') as f:
css = f.read()
st.markdown(f'<style>{css}</style>', unsafe_allow_html=True)
## LOGO and TITLE
## -------------------------------------------------------------------------------------------
# Show the logo and title side by side
col1, col2 = st.columns([1, 4])
with col1:
st.image("brainbot.png", width=100)
with col2:
st.title("File-Chat")
question = None
llm = st.session_state["llm"]
if "current_file" in st.session_state:
current_file = st.session_state['current_file']
if st.sidebar.button("Upload New File"):
st.switch_page("BrainBot.py")
st.subheader("Your file has been uploaded successfully. You can now chat with it.")
st.success(current_file)
question = st.chat_input("Type your question here...")
else:
st.warning("Upload a file to begin chat with it.")
if st.button("Upload File"):
st.switch_page("BrainBot.py")
## CHAT
# Clear the file chat history if user has uploaded a new file
if st.session_state['uploaded_file'] == True:
st.session_state['file_chat_history'] = []
# Display the file chat history
for message in st.session_state['file_chat_history']:
with st.chat_message("user"):
st.write(message["Human"])
with st.chat_message("ai"):
st.markdown(utils.format_response(message["AI"]))
## QUESTION - WITH CHAT HISTORY
## -------------------------------------------------------------------------------------------
# Retrieve the answer to the question asked by the user
if question is not None:
# Display the question entered by the user in chat
with st.chat_message("user"):
st.write(question)
resource = "file"
try:
# Send POST request to a FastAPI endpoint to retrieve an answer for the question
data = {"question": question, "resource": resource}
FASTAPI_URL = f"http://localhost:8000/answer_with_chat_history/{llm}"
with st.spinner("Generating response..."):
response = requests.post(FASTAPI_URL, json=data)
# Append the response to the chat history
st.session_state['file_chat_history'].append({"Human": question, "AI": response.text})
st.session_state['uploaded_file'] = False
# Display the AI's response to the question in chat
with st.chat_message("ai"):
# Format the response
formatted_response = utils.format_response(response.text)
st.markdown(formatted_response)
except Exception as e:
st.switch_page("error.py") |