Spaces:
Sleeping
Sleeping
File size: 2,708 Bytes
1ee06ff 235deb4 1ee06ff 0e21b1c 1ee06ff 0e21b1c 1ee06ff 0e21b1c 1ee06ff 0e21b1c 1ee06ff 0e21b1c 1ee06ff 3ba303e 1ee06ff f3f3243 0e21b1c 1ee06ff 0e21b1c 1ee06ff 0e21b1c 1ee06ff 0e21b1c be39a5f 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 98 |
import streamlit as st
import tempfile
from dspy_qa import DocQA
import os
os.system("pip install dspy-ai[faiss-cpu]")
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", "csv")
if pdf_file:
with tempfile.NamedTemporaryFile(delete=False, suffix=".csv") 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
|