Update app.py

app.py

@@ -30,6 +30,12 @@ regulation_pdfs = {
     "COPPA": "COPPA.pdf"
 }
 
+# Global variables
+full_pdf_content = ""
+vector_store = None
+rag_chain = None
+pdfs_loaded = False  
+
 # Function to extract text from PDF
 def extract_pdf(pdf_path):
     try:
@@ -97,7 +103,7 @@ def markdown_to_html(content):
     return markdown2.markdown(content)
 
 def load_pdfs(gdpr, ferpa, coppa, additional_pdfs):
-    global full_pdf_content, vector_store, rag_chain
+    global full_pdf_content, vector_store, rag_chain, pdfs_loaded
     
     documents = []
     full_pdf_content = ""
@@ -137,6 +143,7 @@ def load_pdfs(gdpr, ferpa, coppa, additional_pdfs):
                 print(f"Failed to extract content from uploaded PDF: {pdf_file.name}")
     
     if not documents:
+        pdfs_loaded = False
         return "No PDFs were successfully loaded. Please check your selections and uploads."
     
     print(f"Total documents loaded: {len(documents)}")
@@ -145,12 +152,13 @@ def load_pdfs(gdpr, ferpa, coppa, additional_pdfs):
     vector_store = generate_embeddings(documents)
     rag_chain = create_rag_chain(vector_store)
     
+    pdfs_loaded = True
     return f"PDFs loaded and RAG system updated successfully! Loaded {len(documents)} document chunks."
 
 def process_query(user_query):
-    global rag_chain, full_pdf_content
+    global rag_chain, full_pdf_content, pdfs_loaded
     
-    if rag_chain is None or not full_pdf_content:
+    if not pdfs_loaded:
         return ("Please load PDFs before asking questions.", 
                 "Please load PDFs before asking questions.", 
                 "Please load PDFs and initialize the system before asking questions.")
@@ -168,11 +176,6 @@ def process_query(user_query):
     
     return rag_response, gemini_resp, html_content
 
-# Initialize
-full_pdf_content = ""
-vector_store = None
-rag_chain = None
-
 # Gradio interface
 with gr.Blocks() as iface:
     gr.Markdown("# Data Protection Team")