Update app.py

app.py

@@ -2,11 +2,19 @@ import gradio as gr
 import pandas as pd
 import tempfile
 
-# Load the data from repo (pre-uploaded CSVs)
-final_accounts = pd.read_csv("accounts.csv")
-final_contacts = pd.read_csv("contacts.csv")
+# Preloaded fallback data
+default_accounts = pd.read_csv("accounts.csv")
+default_contacts = pd.read_csv("contacts.csv")
+
+def extract_contacts(accounts_file, contacts_file, selected_status, selected_sites, contact_type):
+    # Decide source: uploaded or default
+    if accounts_file is not None and contacts_file is not None:
+        final_accounts = pd.read_csv(accounts_file.name)
+        final_contacts = pd.read_csv(contacts_file.name)
+    else:
+        final_accounts = default_accounts
+        final_contacts = default_contacts
 
-def extract_contacts(selected_status, selected_sites, contact_type):
     # Filter accounts
     filtered_accounts = final_accounts[
         (final_accounts['OVERALL_STATUS'].isin(selected_status)) &
@@ -39,14 +47,18 @@ def extract_contacts(selected_status, selected_sites, contact_type):
 
 # Gradio UI
 with gr.Blocks() as demo:
-    gr.Markdown("# 🗂️ Contact Extractor - Filter & Download (Pre-loaded Data)")
+    gr.Markdown("## 🗂️ Contact Extractor - Optional Upload (Stealth Mode)")
+
+    with gr.Row():
+        accounts_file = gr.File(label="Upload Accounts CSV (optional)", file_types=[".csv"])
+        contacts_file = gr.File(label="Upload Contacts CSV (optional)", file_types=[".csv"])
 
     status_dropdown = gr.CheckboxGroup(
-        choices=list(final_accounts['OVERALL_STATUS'].unique()), 
+        choices=list(default_accounts['OVERALL_STATUS'].unique()), 
         label="Select OVERALL_STATUS"
     )
     site_dropdown = gr.CheckboxGroup(
-        choices=list(final_accounts['SITE'].unique()), 
+        choices=list(default_accounts['SITE'].unique()), 
         label="Select SITE"
     )
     contact_type = gr.Radio(["Active", "Inactive"], label="Contact Type")
@@ -58,7 +70,7 @@ with gr.Blocks() as demo:
 
     submit_btn.click(
         extract_contacts,
-        inputs=[status_dropdown, site_dropdown, contact_type],
+        inputs=[accounts_file, contacts_file, status_dropdown, site_dropdown, contact_type],
         outputs=[output_table, download_csv]
     )