Update app.py

app.py

@@ -3,7 +3,7 @@ import pandas as pd
 import io
 
 def extract_contacts(accounts_file, contacts_file, selected_status, selected_sites, contact_type):
-    # Read CSVs
+    # Read CSVs from uploaded files
     final_accounts = pd.read_csv(accounts_file.name)
     final_contacts = pd.read_csv(contacts_file.name)
 
@@ -13,7 +13,7 @@ def extract_contacts(accounts_file, contacts_file, selected_status, selected_sit
         (final_accounts['SITE'].isin(selected_sites))
     ]
 
-    # Filter contacts
+    # Filter contacts based on status and account match
     filtered_contacts = final_contacts[
         (final_contacts['STATUS'].str.upper() == contact_type.upper()) &
         (final_contacts['ACCOUNT_NAME'].isin(filtered_accounts['ACCOUNT_NAME']))
@@ -30,15 +30,16 @@ def extract_contacts(accounts_file, contacts_file, selected_status, selected_sit
         'SF_IDS': [sfids if sfids else f"No {contact_type} IDs"]
     })
 
-    # Create CSV for download
+    # Convert to 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
 
+# Build Gradio UI with Blocks
 with gr.Blocks() as demo:
-    gr.Markdown("# Contact Extractor (Upload & Filter)")
+    gr.Markdown("# 🗂️ Contact Extractor - Upload, Filter, Download")
 
     with gr.Row():
         accounts_file = gr.File(label="Upload Accounts CSV", file_types=[".csv"])
@@ -49,24 +50,28 @@ with gr.Blocks() as demo:
     contact_type = gr.Radio(["Active", "Inactive"], label="Contact Type")
 
     submit_btn = gr.Button("Extract Contacts")
-    
-    output_table = gr.Dataframe()
+
+    output_table = gr.Dataframe(label="Filtered Contact Data")
     download_csv = gr.File(label="Download CSV")
 
-    # Populate filters after file upload
+    # Populate filter options after accounts file upload
     def populate_filters(accounts_file):
         df = pd.read_csv(accounts_file.name)
         return list(df['OVERALL_STATUS'].unique()), list(df['SITE'].unique())
 
-    accounts_file.change(populate_filters, 
-                         inputs=accounts_file, 
-                         outputs=[status_dropdown, site_dropdown])
+    accounts_file.change(
+        populate_filters,
+        inputs=accounts_file,
+        outputs=[status_dropdown, site_dropdown]
+    )
 
-    # Run extraction
+    # On button click, run extract_contacts
     submit_btn.click(
         extract_contacts,
         inputs=[accounts_file, contacts_file, status_dropdown, site_dropdown, contact_type],
         outputs=[output_table, download_csv]
     )
 
-demo.launch()
+# Launch with public link if local run
+if __name__ == "__main__":
+    demo.launch(share=True)