Update app.py

app.py

@@ -1,3 +1,4 @@
+%%writefile app.py
 import streamlit as st
 import tempfile
 from dspy_qa import DocQA
@@ -13,20 +14,31 @@ st.markdown("""
 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;
 }
 
-.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)
 
@@ -37,16 +49,16 @@ if "knowledge_base" not in st.session_state:
 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
+        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 = []
@@ -56,22 +68,29 @@ 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)
-
+        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
+    st.session_state.knowledge_base = None
+    st.session_state.messages = []
+    st.rerun()  # Rerun the script