Update app.py

app.py

@@ -1,8 +1,7 @@
+%%writefile app.py
 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 🚀")
@@ -48,9 +47,9 @@ 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")
+    pdf_file = st.file_uploader("Upload a PDF file", "pdf")
     if pdf_file:
-        with tempfile.NamedTemporaryFile(delete=False, suffix=".csv") as temp_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)
@@ -79,11 +78,11 @@ if question := st.chat_input("What is up?"):
     # 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.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"):