Update app.py

app.py

@@ -514,6 +514,32 @@ def _log_call_end(func_name: str, output_desc: str) -> None:
         print(f"[TOOL RESULT] {func_name} (failed to log output: {e})", flush=True)
 
 
+# ==============================
+# HF Space file URL helper
+# ==============================
+
+def _public_file_url(local_path: str) -> str:
+    """Return an absolute Hugging Face Space URL for a locally saved file.
+
+    This ensures that download links point to the Space host (SPACE_HOST) rather than any
+    reverse-proxy origin (e.g., mcp.nymbo.net), avoiding 404s when proxies don’t expose /file=.
+
+    If SPACE_HOST is not set (e.g., running locally), falls back to a relative "/file=" URL.
+    """
+    try:
+        from urllib.parse import quote
+        host = os.getenv("SPACE_HOST", "").strip()
+        abs_path = os.path.abspath(local_path)
+        quoted = quote(abs_path)
+        if host:
+            return f"https://{host}/file={quoted}"
+        # Local/dev fallback
+        return f"/file={quoted}"
+    except Exception:
+        # Last-resort: return the original path to avoid breaking UI
+        return local_path
+
+
 # ==============================
 # Deep Research helpers: slow-host detection
 # ==============================
@@ -1776,7 +1802,7 @@ kokoro_interface = gr.Interface(
         "Supports unlimited text length by processing all segments. Voice examples: 'af_heart' (US female), 'am_onyx' (US male), "
         "'bf_emma' (British female), 'af_sky' (US female), 'af_nicole' (US female), "
         "Parameters: text (str), speed (float 0.5–2.0, default 1.25x), voice (str from 54 available options, default 'af_heart'). "
-        "Return the generated media to the user in this format `![Alt text](URL)`"
+        "Return the generated media to the user in this format `[Generated Speech](URL)`"
     ),
     flagging_mode="never",
 )
@@ -2164,8 +2190,10 @@ def Generate_Video(  # <-- MCP tool #6 (Generate Video)
                 size = os.path.getsize(path)
             except Exception:
                 size = -1
-            _log_call_end("Generate_Video", f"provider={provider} path={os.path.basename(path)} bytes={size}")
-            return path
+            # Always publish HF Space absolute URL to avoid proxy hosts (e.g., mcp.nymbo.net)
+            url = _public_file_url(path)
+            _log_call_end("Generate_Video", f"provider={provider} path={os.path.basename(path)} bytes={size} url={url}")
+            return url
         except Exception as e:
             last_error = e
             continue
@@ -2211,7 +2239,7 @@ video_generation_interface = gr.Interface(
         "'time-lapse of clouds moving across a blue sky'. Default model: Wan2.2-T2V-A14B (2-6 second videos). "
         "Parameters: prompt (str), model_id (str), negative_prompt (str), steps (int), cfg_scale (float), seed (int), "
         "width/height (int), fps (int), duration (float in seconds). Returns MP4 file path. "
-        "Return the generated media to the user in this format `![Alt text](URL)`"
+        "Return the generated media to the user in this format `[Generated Video](URL)`"
     ),
     flagging_mode="never",
     # Only expose to MCP when HF token is provided; UI tab is always visible
@@ -2699,10 +2727,18 @@ def Deep_Research(
     # Build outputs
     links_text = "\n".join([f"[{i+1}] {u}" for i, u in enumerate(pages.keys())])
     file_path = _write_report_tmp(report)
+    file_url = _public_file_url(file_path)
+    # Append a canonical HF download link so clients avoid proxy origins when clicking
+    try:
+        report = report.rstrip() + f"\n\n---\nDownload the full report as text: [research_report.txt]({file_url})\n"
+    except Exception:
+        pass
     elapsed = time.time() - start_ts
     # Print explicit timing and include in structured log output
     print(f"[TIMING] Deep_Research elapsed: {elapsed:.2f}s", flush=True)
-    _log_call_end("Deep_Research", f"urls={len(pages)} file={os.path.basename(file_path)} duration={elapsed:.2f}s")
+    _log_call_end("Deep_Research", f"urls={len(pages)} file={os.path.basename(file_path)} duration={elapsed:.2f}s url={file_url}")
+    # Keep return signature for existing UI (report, links, file path). MCP clients can prefer the URL by
+    # reading the api_description instruction or deriving it from the path using the same rule.
     return report, links_text, file_path
 
 
@@ -2735,7 +2771,8 @@ deep_research_interface = gr.Interface(
         "Runs 1–5 DDG searches (URLs only), caps total results to 50 (when exceeding, each query returns 10). "
         "Fetches all URLs (3000 chars each) and calls the Researcher to write a research report. "
         "Returns the report (Markdown), the list of sources, and a downloadable text file path. "
-        "Provide the user with one-paragraph summary of the research report and the txt file in this format `![research_report.txt](URL)`"
+        "When sharing a link, always use the Space URL constructed as: https://{SPACE_HOST}/file=<abs_path> (we handle this automatically). "
+        "Provide the user with one-paragraph summary of the research report and the txt file in this format `[research_report.txt](URL)`"
     ),
     flagging_mode="never",
     show_api=bool(HF_TEXTGEN_TOKEN),