Update app.py

app.py

@@ -53,11 +53,16 @@ def process_single_pdf(file_path):
 # Function to load PDFs with progress display
 def load_pdfs_with_progress(folder_path):
     all_text = ""
+    # Check if the folder exists before proceeding
+    if not os.path.exists(folder_path):
+        st.error(f"The folder {folder_path} does not exist.")
+        return
+
     pdf_files = [os.path.join(folder_path, filename) for filename in os.listdir(folder_path) if filename.endswith('.pdf')]
     num_files = len(pdf_files)
 
     if num_files == 0:
-        st.error("No PDF files found in the specified folder.")
+        st.error(f"No PDF files found in the folder: {folder_path}")
         st.session_state['vector_store'] = None
         st.session_state['loading'] = False
         return
@@ -113,7 +118,9 @@ def main():
     # Start loading documents if not already loaded
     if 'loading' not in st.session_state or st.session_state['loading']:
         st.session_state['loading'] = True
-        load_pdfs_with_progress('documents1')
+        # Update folder path for your environment, e.g., 'documents1' should be checked
+        folder_path = 'documents1'  
+        load_pdfs_with_progress(folder_path)
 
     user_question = st.text_input("Ask a Question:", placeholder="Type your question here...")
 
@@ -130,4 +137,3 @@ def main():
 
 if __name__ == "__main__":
     main()
-