Update UI.py

UI.py

@@ -76,9 +76,9 @@ def create_interface(process_and_plot):
             output_gallery = gr.Gallery(label="Results", columns=2, height='auto')
             analysis_output = gr.Textbox(label="Yi-Coder Analysis", lines=15)
 
-        biomass_eq_count = gr.State(1)
-        substrate_eq_count = gr.State(1)
-        product_eq_count = gr.State(1)
+        biomass_eq_count = gr.Number(value=1, visible=False)
+        substrate_eq_count = gr.Number(value=1, visible=False)
+        product_eq_count = gr.Number(value=1, visible=False)
 
         def add_equation(equations_list, eq_count):
             eq_count = min(eq_count + 1, MAX_EQUATIONS)
@@ -143,3 +143,11 @@ def create_interface(process_and_plot):
             substrate_eq_count,
             product_eq_count
         ]
+
+        simulate_btn.click(
+            fn=process_and_plot,
+            inputs=simulate_inputs,
+            outputs=[output_gallery, analysis_output]
+        )
+
+    return demo  # Make sure to return the demo object here