Update app.py

app.py

@@ -3,9 +3,9 @@ import pandas as pd
 import io
 
 def extract_contacts(accounts_file, contacts_file, selected_status, selected_sites, contact_type):
-    # Read uploaded CSVs
-    final_accounts = pd.read_csv(accounts_file)
-    final_contacts = pd.read_csv(contacts_file)
+    # Read CSVs
+    final_accounts = pd.read_csv(accounts_file.name)
+    final_contacts = pd.read_csv(contacts_file.name)
 
     # Filter accounts
     filtered_accounts = final_accounts[
@@ -19,7 +19,7 @@ def extract_contacts(accounts_file, contacts_file, selected_status, selected_sit
         (final_contacts['ACCOUNT_NAME'].isin(filtered_accounts['ACCOUNT_NAME']))
     ]
 
-    # Prepare outputs
+    # Prepare output
     contacts = ", ".join(filtered_contacts['CONTACT'].dropna().astype(str))
     emails = ", ".join(filtered_contacts['EMAIL'].dropna().astype(str))
     sfids = ", ".join(filtered_contacts['SALESFORCE_EXTERNAL_ID'].dropna().astype(str))
@@ -30,49 +30,43 @@ def extract_contacts(accounts_file, contacts_file, selected_status, selected_sit
         'SF_IDS': [sfids if sfids else f"No {contact_type} IDs"]
     })
 
-    # Convert to CSV for download
+    # Create CSV for download
     csv_buffer = io.StringIO()
     df_out.to_csv(csv_buffer, index=False)
     csv_bytes = csv_buffer.getvalue().encode()
 
     return df_out, csv_bytes
 
-def ui(accounts_file, contacts_file):
-    # Read files to populate filters
-    final_accounts = pd.read_csv(accounts_file)
-    overall_status_options = list(final_accounts['OVERALL_STATUS'].unique())
-    site_options = list(final_accounts['SITE'].unique())
+with gr.Blocks() as demo:
+    gr.Markdown("# Contact Extractor (Upload & Filter)")
 
-    # Return interactive function with preloaded options
-    def run(selected_status, selected_sites, contact_type):
-        return extract_contacts(accounts_file, contacts_file, selected_status, selected_sites, contact_type)
+    with gr.Row():
+        accounts_file = gr.File(label="Upload Accounts CSV", file_types=[".csv"])
+        contacts_file = gr.File(label="Upload Contacts CSV", file_types=[".csv"])
 
-    return gr.Interface(
-        fn=run,
-        inputs=[
-            gr.CheckboxGroup(choices=overall_status_options, label="Select OVERALL_STATUS"),
-            gr.CheckboxGroup(choices=site_options, label="Select SITE"),
-            gr.Radio(["Active", "Inactive"], label="Contact Type")
-        ],
-        outputs=[
-            gr.Dataframe(label="Extracted Contacts"),
-            gr.File(label="Download CSV")
-        ],
-        allow_flagging="never",
-        live=True,
-        title="Account - Contact Extractor"
-    )
+    status_dropdown = gr.CheckboxGroup(choices=[], label="Select OVERALL_STATUS")
+    site_dropdown = gr.CheckboxGroup(choices=[], label="Select SITE")
+    contact_type = gr.Radio(["Active", "Inactive"], label="Contact Type")
+
+    submit_btn = gr.Button("Extract Contacts")
+    
+    output_table = gr.Dataframe()
+    download_csv = gr.File(label="Download CSV")
+
+    # Populate filters after file upload
+    def populate_filters(accounts_file):
+        df = pd.read_csv(accounts_file.name)
+        return list(df['OVERALL_STATUS'].unique()), list(df['SITE'].unique())
 
-# Main interface to upload files
-def launcher(accounts_file, contacts_file):
-    return ui(accounts_file, contacts_file)
+    accounts_file.change(populate_filters, 
+                         inputs=accounts_file, 
+                         outputs=[status_dropdown, site_dropdown])
+
+    # Run extraction
+    submit_btn.click(
+        extract_contacts,
+        inputs=[accounts_file, contacts_file, status_dropdown, site_dropdown, contact_type],
+        outputs=[output_table, download_csv]
+    )
 
-gr.Interface(
-    fn=launcher,
-    inputs=[
-        gr.File(label="Upload Accounts CSV (accounts.csv)", file_types=[".csv"]),
-        gr.File(label="Upload Contacts CSV (contacts.csv)", file_types=[".csv"])
-    ],
-    outputs="component",
-    title="Upload Your Files to Start"
-).launch()
+demo.launch()