Update app.py

app.py

@@ -1,4 +1,5 @@
 import streamlit as st
+from streamlit.state.session_state import SessionState
 from PyPDF2 import PdfReader
 from langchain.text_splitter import RecursiveCharacterTextSplitter
 from langchain_groq import ChatGroq
@@ -71,38 +72,41 @@ def main():
     st.markdown("<h1 style='font-size:20px;'>ChatBot by Muhammad Huzaifa</h1>", unsafe_allow_html=True)
     api_key = st.secrets["inference_api_key"]
 
+    session_state = SessionState.get(pdf_docs=None, raw_text=None, processing_complete=False)
+
     # Sidebar column for file upload
     with st.sidebar:
         st.header("Chat with PDF")
-        pdf_docs = st.file_uploader("Upload your PDF Files", accept_multiple_files=True, type=["pdf"])
+        session_state.pdf_docs = st.file_uploader("Upload your PDF Files", accept_multiple_files=True, type=["pdf"])
 
     # Main column for displaying extracted text and user interaction
     col1, col2 = st.columns([1, 2])
-    raw_text = None
-    if pdf_docs:
+
+    if session_state.pdf_docs:
         with col1:
             if st.button("Submit"):
                 with st.spinner("Processing..."):
-                    raw_text = get_pdf_text(pdf_docs)
-                    text_chunks = get_text_chunks(raw_text)
+                    session_state.raw_text = get_pdf_text(session_state.pdf_docs)
+                    text_chunks = get_text_chunks(session_state.raw_text)
                     get_vector_store(text_chunks, api_key)
                     st.success("Processing Complete")
-    
+                    session_state.processing_complete = True
+ 
     # Check if PDF documents are uploaded and processing is complete
-    if pdf_docs and raw_text:
+    if session_state.pdf_docs and session_state.raw_text and session_state.processing_complete:
         with col1:
             user_question = st.text_input("Ask a question from the Docs")
             if user_question:
                 user_input(user_question, api_key)
 
     # Display extracted text if available
-    if raw_text is not None:
+    if session_state.raw_text is not None:
         with col2:
             st.subheader("Extracted Text from PDF:")
-            st.text(raw_text)
+            st.text(session_state.raw_text)
 
     # Show message if no PDF documents are uploaded
-    if not pdf_docs:
+    if not session_state.pdf_docs:
         with col1:
             st.write("Please upload a document first to proceed.")