Update app.py

app.py

@@ -9,6 +9,8 @@ import yaml
 #from process_data import parse_survey_stack_parameters, parse_survey_stack_data, process_specifications
 from process_data import parse_survey_stack, process_specifications
 
+from script_for_automation import drive_process
+
 global original_outputs, yaml_outputs
 
 #def process_survey(button_type, survey_submission_ID): 
@@ -228,7 +230,14 @@ def update_toggle(toggle, farm_output_box, interactions_output_box, trials_outpu
         return yaml_outputs[0], yaml_outputs[1], yaml_outputs[2]
     else:
         return original_outputs[0], original_outputs[1], original_outputs[2]
-            
+
+def generate_output_and_download():
+    zip_file_path = drive_process()
+    if os.path.exists(zip_file_path):
+        return zip_file_path
+    else: 
+        return "Error: Output Folder not generated".
+        
 with gr.Blocks() as app: 
     # This is the main driver of the program that sets up the gradio front end pieces, as well as specific interactivity with buttons
     with gr.Row():
@@ -261,6 +270,15 @@ with gr.Blocks() as app:
         inputs=survey_submission_ID,
         outputs=[get_survey_back, json_output_farm, json_output_interactions, json_output_trials, json_output_farm, json_output_interactions, json_output_trials, toggle_output]
     )
+
+    generate_output_button = gr.Button("Generate Output Folder and Download")
+
+    generate_output_button.click(
+        fn=generate_output_and_download,
+        inputs=[],
+        outputs=[gr.File(label="Download Output Folder")]
+    )
+
     
     #param_submit_button.click(
     #    lambda button_type: process_survey("param", button_type),