Update app.py

app.py

@@ -13,6 +13,9 @@ import json
 os.environ["OPENAI_API_KEY"] = os.getenv("OPENAI_API_KEY")
 client = OpenAI()
 
+original_outputs = None
+xml_outputs = None
+
 # These are the necessary components that make up the Trials 
 #class Variables(BaseModel):
 #    controlled: List[str] = Field(..., title="Controlled Variables", description="A list of controlled variables, which will be constant (controlled) across all trials")
@@ -300,6 +303,8 @@ def process_specifications(data, model_version, json_creation, additional_json_c
     #    resulting_schema = generate_json_pieces(data, model_version, additional_json_creation_options, field_data_input, planting_data_input, logs_data_input, soil_data_input, yield_data_input) 
     #return resulting_schema 
     output1, output2, output3 = generate_json(data, model_version)
+    original_outputs = [output1, output2, output3]
+    
     return output1, output2, output3, Toggle(visible=True)
 
 with gr.Blocks() as demo: 
@@ -340,13 +345,21 @@ with gr.Blocks() as demo:
             return [gr.Textbox(visible=bool(1)), gr.Textbox(visible=bool(0)), gr.Textbox(visible=bool(0)), gr.Textbox(visible=bool(0)), gr.Textbox(visible=bool(0)), gr.Textbox(visible=bool(0))]
 
     def update_toggle(toggle, farm_output_box, interactions_output_box, trials_output_box):
-        if toggle:
+        if toggle and not xml_outputs:
             farm_dict = json.loads(farm_output_box)
             interactions_dict = json.loads(interactions_output_box)
             trials_dict = json.loads(trials_output_box)
-            return dicttoxml(farm_dict), dicttoxml(interactions_dict), dicttoxml(trials_dict)
+
+            farm_xml = dicttoxml(farm_dict)
+            interactions_xml = dicttoxml(interactions_dict)
+            trials_xml = dicttoxml(trials_dict)
+            
+            xml_outputs = [farm_xml, dicttoxml(interactions_xml), trials_xml]
+            return farm_xml, interactions_xml, trials_xml
+        elif toggle and xml_outputs:
+            return xml_outputs[0], xml_outputs[1], xml_outputs[2]
         else:
-            pass
+            return original_outputs[0], original_outputs[1], original_outputs[2]
         
     json_creation_input.change(fn=update_visibility, inputs=[json_creation_input, additional_json_creation_options], outputs=[additional_json_creation_options, field_data_input, planting_data_input, logs_data_input, soil_data_input, yield_data_input])
     additional_json_creation_options.change(fn=update_visibility2, inputs=[additional_json_creation_options], outputs=[data_input, field_data_input, planting_data_input, logs_data_input, soil_data_input, yield_data_input])