Spaces:
Sleeping
Sleeping
File size: 2,650 Bytes
f2ab19a 1ee06ff 0e21b1c 1ee06ff 0e21b1c 1ee06ff 0e21b1c 1ee06ff 0e21b1c 1ee06ff 0e21b1c 1ee06ff 321638b 1ee06ff 321638b 0e21b1c 1ee06ff 0e21b1c 1ee06ff 0e21b1c 1ee06ff 0e21b1c 321638b 0e21b1c 321638b 0e21b1c 1ee06ff 9f7bee0 0e21b1c |
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 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 |
import streamlit as st
import tempfile
from dspy_qa import DocQA
st.set_page_config(page_title="DoC QA", layout="wide")
# st.title("📄 Chat over PDF using DSPy 🚀")
st.markdown("""
<div class="st-emotion-cache-12fmjuu">
<h1>Document QA DSPY</h1>
</div>
""", unsafe_allow_html=True)
st.markdown("""
<style>
.st-emotion-cache-12fmjuu {
text-align: center;
}
.margin-0 {
margin: 0;
font-size: 24px;
justify-content: center;
}
.st-af.st-b6.st-b7.st-ar.st-as.st-b8.st-b9.st-ba.st-bb.st-bc.st-bd.st-be.st-b1{
color: black;
}
.reset-container {
display: flex;
justify-content: space-between;
align-items: center;
margin-bottom: 10px;
}
.stButton>button {
margin-left: auto;
order: 1;
}
</style>
""", unsafe_allow_html=True)
# Initialize chat history
if "knowledge_base" not in st.session_state:
st.session_state.knowledge_base = None
if st.session_state.knowledge_base is None:
pdf_file = st.file_uploader("Upload a PDF file", "pdf")
if pdf_file:
with tempfile.NamedTemporaryFile(delete=False, suffix=".pdf") as temp_file:
temp_file.write(pdf_file.getbuffer())
temp_file_path = temp_file.name
docqa = DocQA(temp_file_path)
if docqa:
st.success("PDF file uploaded successfully!")
st.session_state.knowledge_base = docqa
docqa = st.session_state.knowledge_base
# Initialize chat history
if "messages" not in st.session_state:
st.session_state.messages = []
# Display chat messages from history on app rerun
for message in st.session_state.messages:
with st.chat_message(message["role"]):
st.markdown(message["content"])
# Container for input and reset button
# Accept user input
if question := st.chat_input("What is up?"):
# Add user message to chat history
st.session_state.messages.append({"role": "user", "content": question})
# Display user message in chat message container
with st.chat_message("user"):
st.markdown(question)
if docqa is None:
with st.chat_message("assistant"):
st.markdown("Please upload a PDF file first.")
st.session_state.messages.append({"role": "assistant", "content": "Please upload a PDF file first."})
st.stop()
# Display assistant response in chat message container
with st.chat_message("assistant"):
response = docqa.run(question)
st.markdown(response)
st.session_state.messages.append({"role": "assistant", "content": response})
if st.button("Reset"):
st.session_state.knowledge_base = None
st.session_state.messages = []
st.rerun() # Rerun the script
|