Update app.py

app.py

@@ -7,6 +7,7 @@ from huggingface_hub import InferenceClient
 import tempfile
 import json
 import uuid
+import re
 
 # Project by Nymbo - Converted to MCP Server
 
@@ -14,6 +15,40 @@ import uuid
 API_TOKEN = os.getenv("HF_READ_TOKEN")
 timeout = 100
 
+# Helper: build a public base URL for the running app (HF Spaces or local)
+def _slugify_for_subdomain(s: str) -> str:
+    s = s.strip().lower()
+    s = s.replace(".", "-").replace("_", "-").replace(" ", "-")
+    s = re.sub(r"[^a-z0-9-]", "-", s)
+    s = re.sub(r"-+", "-", s).strip("-")
+    return s
+
+def _build_public_base_url() -> str:
+    # Allow explicit override via env
+    for var in ("HF_SPACE_URL", "SPACE_URL", "PUBLIC_SPACE_URL"):
+        val = os.getenv(var)
+        if val:
+            return val.rstrip("/")
+
+    space_id = os.getenv("SPACE_ID")
+    if space_id:
+        # If a full URL was provided, use it directly
+        if space_id.startswith("http://") or space_id.startswith("https://"):
+            return space_id.rstrip("/")
+        # Typical HF Spaces SPACE_ID is "owner/space-name"
+        if "/" in space_id:
+            owner, space = space_id.split("/", 1)
+            sub = f"{_slugify_for_subdomain(owner)}-{_slugify_for_subdomain(space)}"
+        else:
+            # Fall back to slugifying the whole string
+            sub = _slugify_for_subdomain(space_id)
+        return f"https://{sub}.hf.space"
+
+    # Local fallback
+    host = os.getenv("GRADIO_SERVER_NAME", "localhost")
+    port = os.getenv("GRADIO_SERVER_PORT", os.getenv("PORT", "7860"))
+    return f"http://{host}:{port}"
+
 def flux_krea_generate(
     prompt: str, 
     negative_prompt: str = "(deformed, distorted, disfigured), poorly drawn, bad anatomy, wrong anatomy, extra limb, missing limb, floating limbs, (mutated hands and fingers), disconnected limbs, mutation, mutated, ugly, disgusting, blurry, amputation, misspellings, typos",
@@ -98,16 +133,9 @@ def flux_krea_generate(
                     # This matches the format you saw: /gradio_api/file=<file_path>
                     gradio_file_url = f"/gradio_api/file={temp_file.name}"
                     
-                    # For hosted spaces, we also need the full URL
-                    # Try to detect if we're running on Spaces
-                    space_name = os.getenv("SPACE_ID")
-                    if space_name:
-                        # Running on HF Spaces
-                        full_url = f"https://{space_name}.hf.space{gradio_file_url}"
-                    else:
-                        # Running locally
-                        server_port = os.getenv("GRADIO_SERVER_PORT", "7860")
-                        full_url = f"http://localhost:{server_port}{gradio_file_url}"
+                    # Build a proper public base URL (HF Spaces or local)
+                    base_url = _build_public_base_url()
+                    full_url = f"{base_url}{gradio_file_url}"
                     
                     print(f'\033[1mMCP Generation {generation_id} completed with {provider}!\033[0m')
                     print(f'🌐 Image accessible at: {full_url}')