MK-316 commited on
Commit
c746d07
1 Parent(s): be25c5f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -3
app.py CHANGED
@@ -23,13 +23,16 @@ def group_names(file, members_per_group):
23
 
24
  def main_interface(file, members_per_group):
25
  grouped_df = group_names(file, members_per_group)
26
- # Here, we return the DataFrame directly instead of saving and returning a file path
27
- return grouped_df
 
 
 
28
 
29
  iface = gr.Interface(
30
  fn=main_interface,
31
  inputs=[gr.File(label="Upload CSV File"), gr.Number(label="Members per Group", value=5)],
32
- outputs=gr.Dataframe(label="Grouped Names"),
33
  allow_flagging="never"
34
  )
35
 
 
23
 
24
  def main_interface(file, members_per_group):
25
  grouped_df = group_names(file, members_per_group)
26
+ # Save the DataFrame to a CSV file with UTF-8 encoding
27
+ output_filename = '/tmp/grouped_names.csv'
28
+ grouped_df.to_csv(output_filename, index=False, encoding='utf-8-sig')
29
+ # Return both the DataFrame for display and the file path for downloading
30
+ return grouped_df, output_filename
31
 
32
  iface = gr.Interface(
33
  fn=main_interface,
34
  inputs=[gr.File(label="Upload CSV File"), gr.Number(label="Members per Group", value=5)],
35
+ outputs=[gr.Dataframe(label="Grouped Names"), gr.File(label="Download Grouped Names CSV")],
36
  allow_flagging="never"
37
  )
38