Update app.py

app.py

@@ -197,18 +197,23 @@ class App:
 
     def build_app(self):
         with gr.Blocks() as demo:
+            self.components = []  # Store component objects
             for component in self.app_state["components"]:
                 component_type = component["type"]
                 properties = component["properties"]
                 
                 if component_type == "Button":
-                    gr.Button(value=properties["label"], variant="primary")
+                    button = gr.Button(value=properties["label"], variant="primary")
+                    self.components.append(button)  # Store the button object
                 elif component_type == "Text Input":
-                    gr.Textbox(label=properties["placeholder"])
+                    textbox = gr.Textbox(label=properties["placeholder"])
+                    self.components.append(textbox)
                 elif component_type == "Image":
-                    gr.Image(label=properties["alt"])
+                    image = gr.Image(label=properties["alt"])
+                    self.components.append(image)
                 elif component_type == "Dropdown":
-                    gr.Dropdown(choices=properties["choices"], label="Dropdown")
+                    dropdown = gr.Dropdown(choices=properties["choices"], label="Dropdown")
+                    self.components.append(dropdown)
 
             with gr.Tab("Terminal"):
                 gr.Markdown("## Terminal")