Update app.py

app.py

@@ -92,13 +92,21 @@ def generate_final_response(response1, response2):
 def markdown_to_html(content):
     return markdown2.markdown(content)
 
-def load_pdfs(selected_regulations, additional_pdfs):
+def load_pdfs(gdpr, ferpa, coppa, additional_pdfs):
     global full_pdf_content, vector_store, rag_chain
     
     documents = []
     full_pdf_content = ""
     
     # Load selected regulation PDFs
+    selected_regulations = []
+    if gdpr:
+        selected_regulations.append("GDPR")
+    if ferpa:
+        selected_regulations.append("FERPA")
+    if coppa:
+        selected_regulations.append("COPPA")
+    
     for regulation in selected_regulations:
         if regulation in regulation_pdfs:
             pdf_content = extract_pdf(regulation_pdfs[regulation])
@@ -121,6 +129,11 @@ def load_pdfs(selected_regulations, additional_pdfs):
     rag_chain = create_rag_chain(vector_store)
     
     return "PDFs loaded and RAG system updated successfully!"
+    
+    vector_store = generate_embeddings(documents)
+    rag_chain = create_rag_chain(vector_store)
+    
+    return "PDFs loaded and RAG system updated successfully!"
 
 def process_query(user_query):
     global rag_chain, full_pdf_content
@@ -151,19 +164,14 @@ rag_chain = None
 # Gradio interface
 with gr.Blocks() as iface:
     gr.Markdown("# Data Protection Team")
-    gr.Markdown("**Get responses combining advanced RAG, Long Context, and SOTA models to data protection related questions**")
+    gr.Markdown("Get responses combining advanced RAG, Long Context, and SOTA models to data protection related questions.")
     
     with gr.Row():
         gdpr_checkbox = gr.Checkbox(label="GDPR (EU)")
         ferpa_checkbox = gr.Checkbox(label="FERPA (US)")
         coppa_checkbox = gr.Checkbox(label="COPPA (US <13)")
-    css = """
-#file_upload {
-    max-height: 100px;
-    overflow-y: auto;
-}
-"""
-    gr.Markdown("**Optional: upload additional PDFs if needed (national regulation, school policy)**")
+    
+    gr.Markdown("Optional: upload additional PDFs if needed (national regulation, school policy)")
     additional_pdfs = gr.File(
         file_count="multiple",
         label="Upload additional PDFs",
@@ -174,11 +182,11 @@ with gr.Blocks() as iface:
     load_button = gr.Button("Load PDFs")
     load_output = gr.Textbox(label="Load Status")
     
-    gr.Markdown("**Ask your data protection related question**")
+    gr.Markdown("Ask your data protection related question")
     query_input = gr.Textbox(label="Your Question", placeholder="Ask your question here...")
     query_button = gr.Button("Submit Query")
     
-    gr.Markdown("**Results**")
+    gr.Markdown("Results")
     rag_output = gr.Textbox(label="RAG Pipeline (Llama3.1) Response")
     gemini_output = gr.Textbox(label="Long Context (Gemini 1.5 Pro) Response")
     final_output = gr.HTML(label="Final (GPT-4o) Response")
@@ -186,7 +194,9 @@ with gr.Blocks() as iface:
     load_button.click(
         load_pdfs,
         inputs=[
-            gr.Checkboxgroup([gdpr_checkbox, ferpa_checkbox, coppa_checkbox]),
+            gdpr_checkbox,
+            ferpa_checkbox,
+            coppa_checkbox,
             additional_pdfs
         ],
         outputs=load_output