casestudyqa / streamlit.py
DylanASHillier's picture
current changes
b64d788
raw
history blame
2.47 kB
import streamlit as st
import asyncio
import gradio_client
st.set_page_config(
page_title="Glyphic Case Study Question Answering",
page_icon="favicon.ico",
layout="centered",
)
def check_password():
"""Returns `True` if the user had the correct password."""
def password_entered():
"""Checks whether a password entered by the user is correct."""
if st.session_state["password"] == st.secrets["password"]:
st.session_state["password_correct"] = True
del st.session_state["password"] # don't store password
else:
st.session_state["password_correct"] = False
if "password_correct" not in st.session_state:
# First run, show input for password.
st.text_input(
"Password", type="password", on_change=password_entered, key="password"
)
return False
elif not st.session_state.get("password_correct"):
# Password not correct, show input + error.
st.text_input(
"Password", type="password", on_change=password_entered, key="password"
)
st.error("πŸ˜• Password incorrect")
return False
else:
return True
async def run_query(question: str):
client = gradio_client.Client("https://glyphicai-casestudyqa.hf.space/")
answer = client.submit(question,
api_name="/predict")
answer = answer.result()
print(answer)
return answer["answer"], answer["source"]
# @st.cache_data
def get_answer(question: str):
response = asyncio.run(run_query(question))
return response[0], f"{response[1]}"
DESCRIPTION = """This tool is a demo for allowing you to ask questions over your case studies.
The case studies are from [Workable](https://resources.workable.com/tag/customer-stories/), a recruiting software company.
When you ask a question, the tool will search for the most relevant case study to the question and then use that to answer you."""
if check_password():
# st.title("Glyphic Case Study Question Answering")
col_1, col_2 = st.columns([1, 4])
col_1.image("favicon.ico")
col_2.title("Glyphic Case Study Question Answering")
st.markdown(DESCRIPTION, unsafe_allow_html=True)
question = st.text_input("Enter your question")
if st.button("Get Answer"):
answer, source = get_answer(question)
st.subheader("Answer:")
st.write(answer)
st.subheader("Source:")
st.write(source)