Add option to download generated data as file

app.py

@@ -20,7 +20,7 @@ def generate_data(file, num_samples):
     # Generate synthetic data
     samples = rtf_model.sample(n_samples=num_samples)
 
-    return samples
+    return samples, samples.to_csv('samples.csv')
 
 def generate_relational_data(parent_file, child_file, join_on):
     parent_df = pd.read_csv(parent_file.name)
@@ -68,7 +68,7 @@ def generate_relational_data(parent_file, child_file, join_on):
         input_df=parent_samples.drop(join_on, axis=1),
         gen_batch=5)
 
-    return parent_samples, child_samples, gr.update(visible = True)
+    return parent_samples, child_samples, gr.update(visible = True), parent_samples.to_csv('parent_samples.csv'), child_samples.to_csv('child_samples.csv')
     
 
 with gr.Blocks() as demo:
@@ -101,13 +101,15 @@ with gr.Blocks() as demo:
         with gr.Row():
             #data_sample = gr.Dataframe(label = "Original Data")
             data_output = gr.Dataframe(label = "Synthetic Data")
+            data_output_file = gr.File(label = "Synthetic Data File")
         with gr.Row(visible = False) as child_sample:
             data_output_child = gr.Dataframe(label = "Synthetic Data for Child Dataset")
+            data_output_file_child = gr.File(label = "Synthetic Data File for Child Dataset")
     
     
-    generate_data_btn.click(generate_data, inputs = [data_input_u,num_samples], outputs = [data_output])
-    generate_data_btn_relational.click(generate_relational_data, inputs = [data_input_parent,data_input_child,join_on], outputs = [data_output, data_output_child])
-    examples = gr.Examples(examples=[['diabetes.arff',5], ["titanic.csv", 15]],inputs = [data_input_u,num_samples], outputs = [data_output], cache_examples = True, fn = generate_data)
+    generate_data_btn.click(generate_data, inputs = [data_input_u,num_samples], outputs = [data_output, data_output_file])
+    generate_data_btn_relational.click(generate_relational_data, inputs = [data_input_parent,data_input_child,join_on], outputs = [data_output, data_output_child, child_sample, data_output_file, data_output_file_child])
+    examples = gr.Examples(examples=[['diabetes.arff',5], ["titanic.csv", 15]],inputs = [data_input_u,num_samples], outputs = [data_output,data_output_file], cache_examples = True, fn = generate_data)
 
     
 demo.launch()