Create app.py

app.py

@@ -0,0 +1,78 @@
+import gradio as gr
+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)
+
+    # Filter accounts
+    filtered_accounts = final_accounts[
+        (final_accounts['OVERALL_STATUS'].isin(selected_status)) &
+        (final_accounts['SITE'].isin(selected_sites))
+    ]
+
+    # Filter contacts
+    filtered_contacts = final_contacts[
+        (final_contacts['STATUS'].str.upper() == contact_type.upper()) &
+        (final_contacts['ACCOUNT_NAME'].isin(filtered_accounts['ACCOUNT_NAME']))
+    ]
+
+    # Prepare outputs
+    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))
+
+    df_out = pd.DataFrame({
+        'CONTACTS': [contacts if contacts else f"No {contact_type} Contacts"],
+        'EMAILS': [emails if emails else f"No {contact_type} Emails"],
+        'SF_IDS': [sfids if sfids else f"No {contact_type} IDs"]
+    })
+
+    # 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
+
+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())
+
+    # 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)
+
+    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"
+    )
+
+# Main interface to upload files
+def launcher(accounts_file, contacts_file):
+    return ui(accounts_file, contacts_file)
+
+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()