Update app.py

app.py

@@ -1,10 +1,14 @@
 import gradio as gr
 import requests
 from urllib.parse import urljoin, urlencode, quote
+from gradio_toggle import Toggle
+import json
+import yaml
 
 from process_data import parse_survey_stack_parameters, parse_survey_stack_data, process_specifications
 
 global parsed_input_data
+global original_outputs, yaml_outputs
 
 def process_survey(button_type, submission_id): 
     if button_type == 'data':
@@ -32,7 +36,6 @@ def process_survey(button_type, submission_id):
         response = requests.get(survey_url)
         response.raise_for_status()
         data = response.json()
-        #actual_data = data[0]['data']
 
         print("DATA FROM SURVEY")
         print(data)
@@ -44,7 +47,7 @@ def process_survey(button_type, submission_id):
         else:
             parsed_param_data = parse_survey_stack_parameters(data)
             parsed_farm_json, parsed_interactions_json, parsed_trial_json = process_specifications(parsed_input_data, parsed_param_data)
-            return parsed_farm_json, parsed_interactions_json, parsed_trial_json, gr.TextBox(visible=True), gr.TextBox(visible=True), gr.TextBox(visible=True)
+            return parsed_farm_json, parsed_interactions_json, parsed_trial_json, gr.Textbox(visible=True), gr.Textbox(visible=True), gr.Textbox(visible=True), Toggle(visible=True)
         
     except Exception as e:
         return f"An error occurred while fetching the survey data: {e}"
@@ -76,7 +79,27 @@ def display_data_survey():
     
     iframe_html = f'<iframe src="{survey_url}" width="100%" height="600px"></iframe>'
     return iframe_html
-
+    
+def update_toggle(toggle, farm_output_box, interactions_output_box, trials_output_box):
+        global original_outputs, yaml_outputs
+    
+        if toggle and not yaml_outputs:
+            farm_dict = json.loads(farm_output_box)
+            interactions_dict = json.loads(interactions_output_box)
+            trials_dict = json.loads(trials_output_box)
+
+            farm_yaml = yaml.dump(farm_dict)
+            interactions_yaml = yaml.dump(interactions_dict)
+            trials_yaml = yaml.dump(trials_dict)
+            
+            yaml_outputs = [farm_yaml, interactions_yaml, trials_yaml]
+            
+            return farm_yaml, interactions_yaml, trials_yaml
+        elif toggle and yaml_outputs:
+            return yaml_outputs[0], yaml_outputs[1], yaml_outputs[2]
+        else:
+            return original_outputs[0], original_outputs[1], original_outputs[2]
+            
 with gr.Blocks() as app: 
     with gr.Row():
         data_survey_output = gr.HTML(value=display_data_survey())
@@ -102,13 +125,15 @@ with gr.Blocks() as app:
         json_output_trials = gr.TextBox(label="Trials Data JSON", visible=False)
 
 
-    #param_submit_button.click(fn=process_survey, inputs=[survey2_submission_ID, survey_number], outputs=[get_survey_back2])
+    toggle_output = Toggle(label="JSON <-> YAML", value=False, info="Toggle Output Data", interactive=True, visible=False)
+    toggle_output.change(fn=update_toggle, inputs=[toggle_output, farm_output_box, interactions_output_box, trials_output_box], outputs=[farm_output_box, interactions_output_box, trials_output_box])
+    
     param_submit_button.click(
         lambda button_type: process_survey("param", button_type), 
         inputs=survey2_submission_ID, 
-        outputs=[json_output_farm, json_output_interactions, json_output_trials, json_output_farm, json_output_interactions, json_output_trials]
+        outputs=[json_output_farm, json_output_interactions, json_output_trials, json_output_farm, json_output_interactions, json_output_trials, toggle_output]
     )
-
+    
     data_submit_button.click(
         lambda button_type: process_survey("data", button_type), 
         inputs=[survey_submission_ID],