Add dynamic tab rendering for Extra Space, SAM2.1 Masking Tool, and Custom IFrame Loader with open/close functionality

app.py

@@ -139,55 +139,128 @@ print(client.predict("Alex", 5, False, api_name="/greet"))
         image_output = gr.Image(label="Sepia Image")
         sepia_btn.click(sepia, image_input, image_output)
 
+    # Add state variables for each dynamic tab
+    extra_space_open = gr.State(True)
+    sam_tab_open = gr.State(True)
+    sam_iframe_tab_open = gr.State(True)
+    sam_api_tab_open = gr.State(True)
+    custom_iframe_tab_open = gr.State(True)
+
     with gr.Tab("Extra Space", visible=False) as extra_space_tab:
-        gr.Markdown("## External Gradio Space")
-        gr.HTML('<iframe src="https://huggingface.co/spaces/gradio/calculator" width="100%" height="600" style="border:none;"></iframe>')
+        @gr.render(inputs=extra_space_open)
+        def render_extra_space(is_open):
+            if is_open:
+                gr.Markdown("## External Gradio Space")
+                gr.HTML('<iframe src="https://huggingface.co/spaces/gradio/calculator" width="100%" height="600" style="border:none;"></iframe>')
+                close_btn = gr.Button("Close Tab")
+                def close_tab():
+                    return False
+                close_btn.click(fn=close_tab, outputs=extra_space_open)
+            else:
+                gr.Markdown("Tab closed. Click below to reopen.")
+                reopen_btn = gr.Button("Reopen Tab")
+                def reopen_tab():
+                    return True
+                reopen_btn.click(fn=reopen_tab, outputs=extra_space_open)
 
     with gr.Tab("SAM2.1 Masking Tool (gr.load)", visible=False) as sam_tab:
-        gr.Markdown("## LPX55/SAM2_1-Image-Predictor-Masking-Tool-CPU (gr.load)")
-        sam_interface = gr.load("LPX55/SAM2_1-Image-Predictor-Masking-Tool-CPU", src="spaces")
+        @gr.render(inputs=sam_tab_open)
+        def render_sam_tab(is_open):
+            if is_open:
+                gr.Markdown("## LPX55/SAM2_1-Image-Predictor-Masking-Tool-CPU (gr.load)")
+                sam_interface = gr.load("LPX55/SAM2_1-Image-Predictor-Masking-Tool-CPU", src="spaces")
+                close_btn = gr.Button("Close Tab")
+                def close_tab():
+                    return False
+                close_btn.click(fn=close_tab, outputs=sam_tab_open)
+            else:
+                gr.Markdown("Tab closed. Click below to reopen.")
+                reopen_btn = gr.Button("Reopen Tab")
+                def reopen_tab():
+                    return True
+                reopen_btn.click(fn=reopen_tab, outputs=sam_tab_open)
 
     with gr.Tab("SAM2.1 Masking Tool (iframe)", visible=False) as sam_iframe_tab:
-        gr.Markdown("## LPX55/SAM2_1-Image-Predictor-Masking-Tool-CPU (iframe)")
-        sam_iframe = gr.HTML(visible=True)
+        @gr.render(inputs=sam_iframe_tab_open)
+        def render_sam_iframe_tab(is_open):
+            if is_open:
+                gr.Markdown("## LPX55/SAM2_1-Image-Predictor-Masking-Tool-CPU (iframe)")
+                gr.HTML('<iframe src="https://lpx55-sam2-1-image-predictor-masking-tool-cpu.hf.space" width="100%" height="800" style="border:none;"></iframe>')
+                close_btn = gr.Button("Close Tab")
+                def close_tab():
+                    return False
+                close_btn.click(fn=close_tab, outputs=sam_iframe_tab_open)
+            else:
+                gr.Markdown("Tab closed. Click below to reopen.")
+                reopen_btn = gr.Button("Reopen Tab")
+                def reopen_tab():
+                    return True
+                reopen_btn.click(fn=reopen_tab, outputs=sam_iframe_tab_open)
 
     with gr.Tab("SAM2.1 Masking Tool (API Proxy)", visible=False) as sam_api_tab:
-        gr.Markdown("## LPX55/SAM2_1-Image-Predictor-Masking-Tool-CPU (API Proxy)")
-        api_image = gr.Image(label="Input Image")
-        api_btn = gr.Button("Run Remote Masking")
-        api_output = gr.Textbox(label="API Response (raw)")
+        @gr.render(inputs=sam_api_tab_open)
+        def render_sam_api_tab(is_open):
+            if is_open:
+                gr.Markdown("## LPX55/SAM2_1-Image-Predictor-Masking-Tool-CPU (API Proxy)")
+                api_image = gr.Image(label="Input Image")
+                api_btn = gr.Button("Run Remote Masking")
+                api_output = gr.Textbox(label="API Response (raw)")
+                def call_sam_api(image):
+                    import requests
+                    import base64
+                    import json
+                    if image is None:
+                        return "No image uploaded."
+                    with open(image, "rb") as f:
+                        img_b64 = base64.b64encode(f.read()).decode()
+                    payload = {"data": [img_b64]}
+                    try:
+                        resp = requests.post(
+                            "https://lpx55-sam2-1-image-predictor-masking-tool-cpu.hf.space/run/predict",
+                            json=payload,
+                            timeout=60
+                        )
+                        if resp.status_code == 200:
+                            return json.dumps(resp.json(), indent=2)
+                        else:
+                            return f"Error: {resp.status_code} {resp.text}"
+                    except Exception as e:
+                        return f"Exception: {str(e)}"
+                api_btn.click(fn=call_sam_api, inputs=api_image, outputs=api_output)
+                close_btn = gr.Button("Close Tab")
+                def close_tab():
+                    return False
+                close_btn.click(fn=close_tab, outputs=sam_api_tab_open)
+            else:
+                gr.Markdown("Tab closed. Click below to reopen.")
+                reopen_btn = gr.Button("Reopen Tab")
+                def reopen_tab():
+                    return True
+                reopen_btn.click(fn=reopen_tab, outputs=sam_api_tab_open)
 
     with gr.Tab("Custom IFrame Loader") as custom_iframe_tab:
-        gr.Markdown("## Load Any IFrame URL")
-        custom_url = gr.Textbox(label="IFrame URL", placeholder="https://example.com")
-        load_custom_iframe_btn = gr.Button("Load IFrame")
-        custom_iframe = gr.HTML(visible=True)
-
-    def call_sam_api(image):
-        import requests
-        import base64
-        import json
-        if image is None:
-            return "No image uploaded."
-        # Convert image to base64
-        with open(image, "rb") as f:
-            img_b64 = base64.b64encode(f.read()).decode()
-        # Prepare payload (this is a placeholder, actual API may differ)
-        payload = {"data": [img_b64]}
-        try:
-            resp = requests.post(
-                "https://lpx55-sam2-1-image-predictor-masking-tool-cpu.hf.space/run/predict",
-                json=payload,
-                timeout=60
-            )
-            if resp.status_code == 200:
-                return json.dumps(resp.json(), indent=2)
+        @gr.render(inputs=custom_iframe_tab_open)
+        def render_custom_iframe_tab(is_open):
+            if is_open:
+                gr.Markdown("## Load Any IFrame URL")
+                custom_url = gr.Textbox(label="IFrame URL", placeholder="https://example.com")
+                load_custom_iframe_btn = gr.Button("Load IFrame")
+                custom_iframe = gr.HTML(visible=True)
+                def load_custom_iframe(url):
+                    if not url:
+                        return "<div style='color:red'>Please enter a URL.</div>"
+                    return f'<iframe src="{url}" width="100%" height="800" style="border:none;"></iframe>'
+                load_custom_iframe_btn.click(fn=load_custom_iframe, inputs=custom_url, outputs=custom_iframe)
+                close_btn = gr.Button("Close Tab")
+                def close_tab():
+                    return False
+                close_btn.click(fn=close_tab, outputs=custom_iframe_tab_open)
             else:
-                return f"Error: {resp.status_code} {resp.text}"
-        except Exception as e:
-            return f"Exception: {str(e)}"
-
-    api_btn.click(fn=call_sam_api, inputs=api_image, outputs=api_output)
+                gr.Markdown("Tab closed. Click below to reopen.")
+                reopen_btn = gr.Button("Reopen Tab")
+                def reopen_tab():
+                    return True
+                reopen_btn.click(fn=reopen_tab, outputs=custom_iframe_tab_open)
 
     def show_tab():
         return gr.Tab(visible=True)
@@ -210,14 +283,6 @@ print(client.predict("Alex", 5, False, api_name="/greet"))
 
     load_sam_api_btn.click(fn=show_sam_api_tab, outputs=[sam_api_tab])
 
-    def load_custom_iframe(url):
-        if not url:
-            return "<div style='color:red'>Please enter a URL.</div>"
-        return f'<iframe src="{url}" width="100%" height="800" style="border:none;"></iframe>'
-
-    load_custom_iframe_btn.click(fn=load_custom_iframe, inputs=custom_url, outputs=custom_iframe)
-
-
     def open_in_new_tab():
         # This function does nothing server-side, but the button will have a link
         pass