Update app.py
Browse files
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 |
-
#
|
27 |
-
|
|
|
|
|
|
|
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 |
|