Added exception handling

pages/Load_Documents.py

@@ -1,9 +1,15 @@
+import os
 from utils._admin_util import create_embeddings, create_vector_store, read_pdf_data, split_data
 import streamlit as st
 from dotenv import load_dotenv
 
 def main():
     load_dotenv()
+    
+    # Add API key verification at the start
+    if not os.getenv("OPENAI_API_KEY"):
+        st.error("OpenAI API key not found! Please ensure it's set in the environment variables.")
+        st.stop()
 
     st.set_page_config(page_title="Dump PDFs to QDrant - Vector Store")
     st.title("Please upload your files...📁 ")
@@ -12,34 +18,37 @@ def main():
     uploaded_files = st.file_uploader("Upload PDF files", type=["pdf"], accept_multiple_files=True)
 
     if uploaded_files:
-        with st.spinner('Processing PDF files...'):
-            all_chunks = []
-            
-            # Process each PDF file
-            for pdf in uploaded_files:
-                st.write(f"Processing: {pdf.name}")
+        try:
+            with st.spinner('Processing PDF files...'):
+                all_chunks = []
                 
-                # Extract text from PDF
-                text = read_pdf_data(pdf)
-                st.write(f"👉 Reading {pdf.name} done")
-
-                # Create chunks for this PDF
-                chunks = split_data(text)
-                all_chunks.extend(chunks)
-                st.write(f"👉 Splitting {pdf.name} into chunks done")
-
-            # Create embeddings once for all chunks
-            st.write("Creating embeddings...")
-            embeddings = create_embeddings()
-            st.write("👉 Creating embeddings instance done")
-
-            # Create vector store with all chunks
-            vector_store = create_vector_store(embeddings, all_chunks)
-            st.session_state.vector_store = vector_store
-            
-            st.success(f"Successfully processed {len(uploaded_files)} files and pushed embeddings to Qdrant")
-            st.write(f"Total chunks created: {len(all_chunks)}")
-
+                # Process each PDF file
+                for pdf in uploaded_files:
+                    st.write(f"Processing: {pdf.name}")
+                    
+                    # Extract text from PDF
+                    text = read_pdf_data(pdf)
+                    st.write(f"👉 Reading {pdf.name} done")
+
+                    # Create chunks for this PDF
+                    chunks = split_data(text)
+                    all_chunks.extend(chunks)
+                    st.write(f"👉 Splitting {pdf.name} into chunks done")
+
+                # Create embeddings once for all chunks
+                st.write("Creating embeddings...")
+                embeddings = create_embeddings()
+                st.write("👉 Creating embeddings instance done")
+
+                # Create vector store with all chunks
+                vector_store = create_vector_store(embeddings, all_chunks)
+                st.session_state.vector_store = vector_store
+                
+                st.success(f"Successfully processed {len(uploaded_files)} files and pushed embeddings to Qdrant")
+                st.write(f"Total chunks created: {len(all_chunks)}")
+        except Exception as e:
+            st.error(f"An error occurred: {str(e)}")
+            st.error("Please check your API key and permissions.")
 
 if __name__ == '__main__':
     main()