Return public URLs from MCP media tools

app.py

@@ -1800,6 +1800,38 @@ def build_gradio_media_url(media_path: Optional[str]) -> str:
     return f"{API_PREFIX or ''}/file={quote(source, safe='/:')}"
 
 
+def get_public_app_base_url() -> str:
+    """Return the public base URL for absolute MCP media links when available."""
+    configured = os.getenv("LANCE_PUBLIC_BASE_URL", "").strip()
+    if configured:
+        return configured.rstrip("/")
+
+    for env_name in ("SPACE_HOST", "HF_SPACE_HOST"):
+        host = os.getenv(env_name, "").strip()
+        if host:
+            if host.startswith(("http://", "https://")):
+                return host.rstrip("/")
+            return f"https://{host.rstrip('/')}"
+
+    space_id = os.getenv("SPACE_ID", "").strip()
+    if "/" in space_id:
+        return f"https://{space_id.replace('/', '-').lower()}.hf.space"
+
+    return ""
+
+
+def build_public_gradio_media_url(media_path: Optional[str]) -> str:
+    """Build a client-readable Gradio file URL for MCP outputs."""
+    media_url = build_gradio_media_url(media_path)
+    if not media_url or _is_http_url(media_url):
+        return media_url
+
+    base_url = get_public_app_base_url()
+    if not base_url:
+        return media_url
+    return f"{base_url}{media_url if media_url.startswith('/') else '/' + media_url}"
+
+
 def _normalize_media_kind(media_kind: str) -> str:
     normalized = str(media_kind or "auto").strip().lower()
     if normalized in {"img", "image"}:
@@ -1943,9 +1975,9 @@ def prepare_media_input(media_value: Any, media_kind: str, required: bool = Fals
 
 
 def upload_file_to_gradio(file_url: str, media_type: str = "image") -> tuple[str, str]:
-    """Upload a remote image/video URL into this Gradio app and return a reusable local path plus a Gradio file URL."""
+    """Upload a remote image/video URL into this Gradio app and return a reusable local path plus a public Gradio file URL."""
     local_path = prepare_media_input(file_url, _normalize_media_kind(media_type), required=True)
-    return local_path, build_gradio_media_url(local_path)
+    return local_path, build_public_gradio_media_url(local_path)
 
 
 def build_example_media_html(media_path: Optional[str], media_type: str, fallback_media_path: Optional[str] = None) -> str:
@@ -3890,7 +3922,8 @@ def _format_mcp_media_result(result: tuple[Any, Any, str, str], output_kind: str
     output_video, output_image, _output_text, status = result
     output_path = output_video if output_kind == "video" else output_image
     if output_path:
-        return str(output_path), status or "Success."
+        output_url = build_public_gradio_media_url(str(output_path))
+        return output_url, status or f"Success. Local output path: {output_path}"
     return "", status or f"No output {output_kind} was produced."
 
 
@@ -3911,7 +3944,7 @@ def mcp_generate_video(
     validation_timestep_shift: float = DEFAULT_TIMESTEP_SHIFT,
     cfg_text_scale: float = DEFAULT_CFG_TEXT_SCALE,
 ) -> tuple[str, str]:
-    """Generate a video from a text prompt. Returns the generated MP4 path and a status message."""
+    """Generate a video from a text prompt. Returns a public Gradio URL for the generated MP4 and a status message."""
     result = _run_mcp_task(
         task=TASK_T2V,
         prompt=prompt,
@@ -3940,7 +3973,7 @@ def mcp_edit_video(
     validation_timestep_shift: float = DEFAULT_TIMESTEP_SHIFT,
     cfg_text_scale: float = DEFAULT_CFG_TEXT_SCALE,
 ) -> tuple[str, str]:
-    """Edit an input video from a local path, Gradio file URL, or HTTP(S) URL. Returns the edited MP4 path and status."""
+    """Edit an input video from a local path, Gradio file URL, or HTTP(S) URL. Returns a public Gradio URL for the edited MP4 and status."""
     result = _run_mcp_task(
         task=TASK_VIDEO_EDIT,
         prompt=prompt,
@@ -3994,7 +4027,7 @@ def mcp_generate_image(
     validation_timestep_shift: float = DEFAULT_TIMESTEP_SHIFT,
     cfg_text_scale: float = DEFAULT_CFG_TEXT_SCALE,
 ) -> tuple[str, str]:
-    """Generate an image from a text prompt. Returns the generated image path and a status message."""
+    """Generate an image from a text prompt. Returns a public Gradio URL for the generated image and a status message."""
     result = _run_mcp_task(
         task=TASK_T2I,
         prompt=prompt,
@@ -4021,7 +4054,7 @@ def mcp_edit_image(
     validation_timestep_shift: float = DEFAULT_TIMESTEP_SHIFT,
     cfg_text_scale: float = DEFAULT_CFG_TEXT_SCALE,
 ) -> tuple[str, str]:
-    """Edit an input image from a local path, Gradio file URL, or HTTP(S) URL. Returns the edited image path and status."""
+    """Edit an input image from a local path, Gradio file URL, or HTTP(S) URL. Returns a public Gradio URL for the edited image and status."""
     result = _run_mcp_task(
         task=TASK_IMAGE_EDIT,
         prompt=prompt,
@@ -4422,13 +4455,13 @@ def build_demo() -> gr.Blocks:
             video_steps = mcp_number("validation_num_timesteps", DEFAULT_TIMESTEPS)
             video_shift = mcp_number("validation_timestep_shift", DEFAULT_TIMESTEP_SHIFT)
             video_cfg = mcp_number("cfg_text_scale", DEFAULT_CFG_TEXT_SCALE)
-            video_output_path = mcp_text("output_video_path")
+            video_output_url = mcp_text("output_video_url")
             video_status = mcp_text("status")
             video_button = gr.Button("MCP generate_video", visible=False)
             video_button.click(
                 fn=mcp_generate_video,
                 inputs=[video_prompt, video_resolution, video_aspect_ratio, video_duration, video_seed, video_steps, video_shift, video_cfg],
-                outputs=[video_output_path, video_status],
+                outputs=[video_output_url, video_status],
                 api_name="generate_video",
                 show_progress="minimal",
             )
@@ -4442,7 +4475,7 @@ def build_demo() -> gr.Blocks:
             video_edit_steps = mcp_number("validation_num_timesteps", DEFAULT_TIMESTEPS)
             video_edit_shift = mcp_number("validation_timestep_shift", DEFAULT_TIMESTEP_SHIFT)
             video_edit_cfg = mcp_number("cfg_text_scale", DEFAULT_CFG_TEXT_SCALE)
-            video_edit_output_path = mcp_text("output_video_path")
+            video_edit_output_url = mcp_text("output_video_url")
             video_edit_status = mcp_text("status")
             video_edit_button = gr.Button("MCP edit_video", visible=False)
             video_edit_button.click(
@@ -4458,7 +4491,7 @@ def build_demo() -> gr.Blocks:
                     video_edit_shift,
                     video_edit_cfg,
                 ],
-                outputs=[video_edit_output_path, video_edit_status],
+                outputs=[video_edit_output_url, video_edit_status],
                 api_name="edit_video",
                 show_progress="minimal",
             )
@@ -4497,13 +4530,13 @@ def build_demo() -> gr.Blocks:
             image_steps = mcp_number("validation_num_timesteps", DEFAULT_TIMESTEPS)
             image_shift = mcp_number("validation_timestep_shift", DEFAULT_TIMESTEP_SHIFT)
             image_cfg = mcp_number("cfg_text_scale", DEFAULT_CFG_TEXT_SCALE)
-            image_output_path = mcp_text("output_image_path")
+            image_output_url = mcp_text("output_image_url")
             image_status = mcp_text("status")
             image_button = gr.Button("MCP generate_image", visible=False)
             image_button.click(
                 fn=mcp_generate_image,
                 inputs=[image_prompt, image_aspect_ratio, image_seed, image_steps, image_shift, image_cfg],
-                outputs=[image_output_path, image_status],
+                outputs=[image_output_url, image_status],
                 api_name="generate_image",
                 show_progress="minimal",
             )
@@ -4515,7 +4548,7 @@ def build_demo() -> gr.Blocks:
             image_edit_steps = mcp_number("validation_num_timesteps", DEFAULT_TIMESTEPS)
             image_edit_shift = mcp_number("validation_timestep_shift", DEFAULT_TIMESTEP_SHIFT)
             image_edit_cfg = mcp_number("cfg_text_scale", DEFAULT_CFG_TEXT_SCALE)
-            image_edit_output_path = mcp_text("output_image_path")
+            image_edit_output_url = mcp_text("output_image_url")
             image_edit_status = mcp_text("status")
             image_edit_button = gr.Button("MCP edit_image", visible=False)
             image_edit_button.click(
@@ -4529,7 +4562,7 @@ def build_demo() -> gr.Blocks:
                     image_edit_shift,
                     image_edit_cfg,
                 ],
-                outputs=[image_edit_output_path, image_edit_status],
+                outputs=[image_edit_output_url, image_edit_status],
                 api_name="edit_image",
                 show_progress="minimal",
             )