Add SAM2.1 Masking Tool integration with new API Proxy tab and custom IFrame loader

app.py

@@ -54,16 +54,22 @@ theme = gr.Theme.from_hub("LPX55/modal_ai")
 with gr.Blocks(theme=theme, js=js_func, css=css_func) as demo:
     show_space_tab = gr.State(False)
     with gr.Sidebar(width="25vw"):
-        gr.Image("logo.png", width=80, show_label=False)
         gr.Markdown("""
-        # 🤖 Gradio API Demo
-        _Build, test, and share AI tools instantly!_
+        # 🤖 API + MCP Demo
+        _Dynamically load tabs 🤫_
 
+        ---
+        **Ways to load the remote SAM2.1 Masking Tool:**
+        - **gr.load()**: Loads the remote interface at startup, just hidden until shown.
+        - **iframe**: Loads the remote UI only when you click the button (no API integration).
+        - **API Proxy**: Upload an image, and the backend will call the remote Space's API and show the result.
+        - **Open in New Tab**: Opens the remote Space in a new browser tab.
         ---
         **Navigation**
         - [Greeting](#greeting)
         - [Calculator](#calculator)
         - [Sepia Image](#sepia-image)
+        - [Custom IFrame Loader](#custom-iframe-loader)
         ---
         ### [GitHub Repo](https://github.com/yourrepo) | [Docs](https://yourdocs)
         ---
@@ -84,6 +90,8 @@ print(client.predict("Alex", 5, False, api_name="/greet"))
         load_space_btn = gr.Button("Load Extra Space", elem_id="load-space-btn")
         load_sam_btn = gr.Button("Load SAM2.1 Masking Tool (gr.load)", elem_id="load-sam-btn")
         load_sam_iframe_btn = gr.Button("Load SAM2.1 Masking Tool (iframe)", elem_id="load-sam-iframe-btn")
+        load_sam_api_btn = gr.Button("Load SAM2.1 Masking Tool (API Proxy)", elem_id="load-sam-api-btn")
+        open_sam_tab_btn = gr.Button("Open SAM2.1 Masking Tool in New Tab", elem_id="open-sam-tab-btn")
 
     with gr.Tab("Greeting"):
         gr.Markdown("## Greeting Generator")
@@ -143,6 +151,44 @@ print(client.predict("Alex", 5, False, api_name="/greet"))
         gr.Markdown("## LPX55/SAM2_1-Image-Predictor-Masking-Tool-CPU (iframe)")
         sam_iframe = gr.HTML(visible=True)
 
+    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)")
+
+    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)
+            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)
+
     def show_tab():
         return gr.Tab(visible=True)
 
@@ -159,6 +205,25 @@ print(client.predict("Alex", 5, False, api_name="/greet"))
 
     load_sam_iframe_btn.click(fn=show_sam_iframe_tab, outputs=[sam_iframe_tab, sam_iframe])
 
+    def show_sam_api_tab():
+        return gr.Tab(visible=True)
+
+    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
+
+    open_sam_tab_btn.click(fn=open_in_new_tab, inputs=None, outputs=None, js="window.open('https://lpx55-sam2-1-image-predictor-masking-tool-cpu.hf.space', '_blank')")
+
     gr.Markdown("""
     ---
     ## 📡 API Usage