Update app.py

app.py

@@ -76,34 +76,17 @@ def handle_query(query):
 st.title("(PDF) Information and Inference")
 st.markdown("Retrieval-Augmented Generation") 
 
-if 'messages' not in st.session_state:
-    preloaded_pdfs_list = ", ".join(PRELOADED_PDFS) if PRELOADED_PDFS else "No preloaded PDFs available"
-    st.session_state.messages = [{'role': 'assistant', "content": f'Hello! I have the following preloaded PDFs: {preloaded_pdfs_list}. You can also upload your own PDF. Once a PDF is selected or uploaded, ask me anything about its content.'}]
+# Automatically load all PDFs from the data directory
+if 'pdfs_loaded' not in st.session_state:
+    with st.spinner("Loading PDFs..."):
+        data_ingestion()
+    st.session_state.pdfs_loaded = True
 
-with st.sidebar:
-    st.title("Menu:")
-    pdf_source = st.radio("Choose PDF source", ["Upload", "Preloaded"])
-    
-    if pdf_source == "Upload":
-        uploaded_file = st.file_uploader("Upload your PDF Files and Click on the Submit & Process Button")
-        if st.button("Submit & Process"):
-            with st.spinner("Processing..."):
-                filepath = os.path.join(DATA_DIR, "uploaded_pdf.pdf")
-                with open(filepath, "wb") as f:
-                    f.write(uploaded_file.getbuffer())
-                displayPDF(filepath)  # Display the uploaded PDF
-                data_ingestion()  # Process PDF every time new file is uploaded
-                st.success("Done")
-    else:
-        selected_pdf = st.selectbox("Choose a preloaded PDF", PRELOADED_PDFS)
-        if st.button("Load Selected PDF"):
-            with st.spinner("Processing..."):
-                filepath = os.path.join(DATA_DIR, selected_pdf)
-                displayPDF(filepath)  # Display the selected PDF
-                data_ingestion()  # Process the selected PDF
-                st.success("Done")
+if 'messages' not in st.session_state:
+    preloaded_pdfs_list = ", ".join(PRELOADED_PDFS) if PRELOADED_PDFS else "No PDFs available"
+    st.session_state.messages = [{'role': 'assistant', "content": f'Hello! I have loaded the following PDFs: {preloaded_pdfs_list}. Ask me anything about their content.'}]
 
-user_prompt = st.chat_input("Ask me anything about the content of the PDF:")
+user_prompt = st.chat_input("Ask me anything about the content of the PDFs:")
 if user_prompt:
     st.session_state.messages.append({'role': 'user', "content": user_prompt})
     response = handle_query(user_prompt)