akshay33 commited on
Commit
bdaf833
·
verified ·
1 Parent(s): a774f23

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -1,7 +1,6 @@
1
  import gradio as gr
2
  import pandas as pd
3
  import tempfile
4
- import io
5
 
6
  # Load the data from repo (pre-uploaded CSVs)
7
  final_accounts = pd.read_csv("accounts.csv")
@@ -31,7 +30,7 @@ def extract_contacts(selected_status, selected_sites, contact_type):
31
  'SF_IDS': [sfids if sfids else f"No {contact_type} IDs"]
32
  })
33
 
34
- # Save CSV to a temp file
35
  tmp_file = tempfile.NamedTemporaryFile(delete=False, suffix=".csv", mode='w', encoding='utf-8')
36
  df_out.to_csv(tmp_file, index=False)
37
  tmp_file.close()
@@ -40,7 +39,7 @@ def extract_contacts(selected_status, selected_sites, contact_type):
40
 
41
  # Gradio UI
42
  with gr.Blocks() as demo:
43
- gr.Markdown("# 🗂️ Contact Extractor - Filter & Download (No Upload Needed)")
44
 
45
  status_dropdown = gr.CheckboxGroup(
46
  choices=list(final_accounts['OVERALL_STATUS'].unique()),
@@ -63,4 +62,5 @@ with gr.Blocks() as demo:
63
  outputs=[output_table, download_csv]
64
  )
65
 
66
- demo.launch()
 
 
1
  import gradio as gr
2
  import pandas as pd
3
  import tempfile
 
4
 
5
  # Load the data from repo (pre-uploaded CSVs)
6
  final_accounts = pd.read_csv("accounts.csv")
 
30
  'SF_IDS': [sfids if sfids else f"No {contact_type} IDs"]
31
  })
32
 
33
+ # Save to temporary file for download
34
  tmp_file = tempfile.NamedTemporaryFile(delete=False, suffix=".csv", mode='w', encoding='utf-8')
35
  df_out.to_csv(tmp_file, index=False)
36
  tmp_file.close()
 
39
 
40
  # Gradio UI
41
  with gr.Blocks() as demo:
42
+ gr.Markdown("# 🗂️ Contact Extractor - Filter & Download (Pre-loaded Data)")
43
 
44
  status_dropdown = gr.CheckboxGroup(
45
  choices=list(final_accounts['OVERALL_STATUS'].unique()),
 
62
  outputs=[output_table, download_csv]
63
  )
64
 
65
+ if __name__ == "__main__":
66
+ demo.launch(share=True)