Refactor tab close functionality to return visibility state directly for Extra Space, SAM2.1 Masking Tool, and Custom IFrame Loader

app.py

@@ -154,14 +154,8 @@ print(client.predict("Alex", 5, False, api_name="/greet"))
                 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)
+                    return gr.Tab(visible=False), False
+                close_btn.click(fn=close_tab, outputs=[extra_space_tab, extra_space_open])
 
     with gr.Tab("SAM2.1 Masking Tool (gr.load)", visible=False) as sam_tab:
         @gr.render(inputs=sam_tab_open)
@@ -171,14 +165,8 @@ print(client.predict("Alex", 5, False, api_name="/greet"))
                 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)
+                    return gr.Tab(visible=False), False
+                close_btn.click(fn=close_tab, outputs=[sam_tab, sam_tab_open])
 
     with gr.Tab("SAM2.1 Masking Tool (iframe)", visible=False) as sam_iframe_tab:
         @gr.render(inputs=sam_iframe_tab_open)
@@ -188,14 +176,8 @@ print(client.predict("Alex", 5, False, api_name="/greet"))
                 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)
+                    return gr.Tab(visible=False), False
+                close_btn.click(fn=close_tab, outputs=[sam_iframe_tab, sam_iframe_tab_open])
 
     with gr.Tab("SAM2.1 Masking Tool (API Proxy)", visible=False) as sam_api_tab:
         @gr.render(inputs=sam_api_tab_open)
@@ -229,14 +211,8 @@ print(client.predict("Alex", 5, False, api_name="/greet"))
                 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)
+                    return gr.Tab(visible=False), False
+                close_btn.click(fn=close_tab, outputs=[sam_api_tab, sam_api_tab_open])
 
     with gr.Tab("Custom IFrame Loader") as custom_iframe_tab:
         @gr.render(inputs=custom_iframe_tab_open)
@@ -253,14 +229,8 @@ print(client.predict("Alex", 5, False, api_name="/greet"))
                 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:
-                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)
+                    return gr.Tab(visible=False), False
+                close_btn.click(fn=close_tab, outputs=[custom_iframe_tab, custom_iframe_tab_open])
 
     def show_tab():
         return gr.Tab(visible=True)