Update app.py

app.py

@@ -177,10 +177,10 @@ 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.")
     
-    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)")
+    regulations_checkboxes = gr.CheckboxGroup(
+        choices=["GDPR (EU)", "FERPA (US)", "COPPA (US <13)"],
+        label="Select Regulations"
+    )
     
     gr.Markdown("**Optional: upload additional PDFs if needed (national regulation, school policy)**")
     additional_pdfs = gr.File(
@@ -202,14 +202,12 @@ with gr.Blocks() as iface:
     gemini_output = gr.Textbox(label="Long Context (Gemini 1.5 Pro) Response")
     final_output = gr.HTML(label="Final (GPT-4o) Response")
     
+    def prepare_regulations(selected):
+        return [reg.split()[0] for reg in selected]
+    
     load_button.click(
-        load_pdfs,
-        inputs=[
-            gdpr_checkbox,
-            ferpa_checkbox,
-            coppa_checkbox,
-            additional_pdfs
-        ],
+        lambda selected, additional: load_pdfs(prepare_regulations(selected), additional),
+        inputs=[regulations_checkboxes, additional_pdfs],
         outputs=load_output
     )