Spaces:

mgbam
/

builder

Running

App Files Files Community

mgbam commited on 24 days ago

Commit

80bfd4a

verified ·

1 Parent(s): b2a87aa

Update deploy.py

Browse files

Files changed (1) hide show

deploy.py +21 -99

deploy.py CHANGED Viewed

@@ -14,11 +14,10 @@ from huggingface_hub import HfApi, duplicate_space
 # ------------------------------------------------------------------
 def send_to_sandbox(code: str) -> str:
-    """Wrap HTML code in a sandboxed iframe via a data URI."""
     wrapped = f"""
     <!DOCTYPE html>
     <html>
-    <head><meta charset="UTF-8"><meta name="viewport" content="width=device-width, initial-scale=1.0"></head>
     <body>{code}</body>
     </html>
     """
@@ -30,18 +29,9 @@ def send_to_sandbox(code: str) -> str:
     )
 def demo_card_click(e: gr.EventData) -> str:
-    """Return the description of a demo card when clicked."""
-    try:
-        idx = (
-            e._data.get("index")
-            or e._data.get("component", {}).get("index")
-            or e._data.get("target", {}).get("index")
-        )
-    except Exception:
-        idx = 0
     from constants import DEMO_LIST
-    if not (0 <= idx < len(DEMO_LIST)):
-        idx = 0
     return DEMO_LIST[idx]["description"]
 # ------------------------------------------------------------------
@@ -49,7 +39,6 @@ def demo_card_click(e: gr.EventData) -> str:
 # ------------------------------------------------------------------
 def wrap_html_in_gradio_app(html_code: str) -> str:
-    """Generate a minimal Gradio app.py that renders given HTML."""
     safe = html_code.replace('"""', r'\"\"\"')
     return (
         "import gradio as gr\n\n"
@@ -61,40 +50,23 @@ def wrap_html_in_gradio_app(html_code: str) -> str:
     )
 def deploy_to_spaces(code: str) -> None:
-    """Open a new tab to create a Hugging Face Space with a Gradio app."""
     if not code.strip():
         return
     app_py = wrap_html_in_gradio_app(code)
-    params = urllib.parse.urlencode({
-        "name": "new-space",
-        "sdk": "gradio"
-    })
-    files = {
-        "files[0][path]": "app.py",
-        "files[0][content]": app_py
-    }
-    files_params = urllib.parse.urlencode(files)
     url = f"https://huggingface.co/new-space?{params}&{files_params}"
     webbrowser.open_new_tab(url)
 def wrap_html_in_static_app(html_code: str) -> str:
-    """Return raw HTML for a static Hugging Face Space."""
     return html_code
 def deploy_to_spaces_static(code: str) -> None:
-    """Open a new tab to create a static HTML Space."""
     if not code.strip():
         return
     html = wrap_html_in_static_app(code)
-    params = urllib.parse.urlencode({
-        "name": "new-space",
-        "sdk": "static"
-    })
-    files = {
-        "files[0][path]": "index.html",
-        "files[0][content]": html
-    }
-    files_params = urllib.parse.urlencode(files)
     url = f"https://huggingface.co/new-space?{params}&{files_params}"
     webbrowser.open_new_tab(url)
@@ -103,39 +75,21 @@ def deploy_to_spaces_static(code: str) -> None:
 # ------------------------------------------------------------------
 def check_hf_space_url(url: str):
-    """Validate a HF Spaces URL and extract username/project."""
     import re
-    pattern = re.compile(r'^(?:https?://)?(?:huggingface\.co|hf\.co)/spaces/([\w-]+)/([\w-]+)$', re.IGNORECASE)
     m = pattern.match(url.strip())
-    if not m:
-        return False, None, None
-    return True, m.group(1), m.group(2)
 def fetch_hf_space_content(username: str, project: str) -> str:
-    """Download the main file of a Space and return its content."""
     api = HfApi()
     info = api.space_info(f"{username}/{project}")
     sdk = info.sdk
-    # Choose main file based on SDK
-    if sdk == "static":
-        main_file = "index.html"
-    elif sdk == "gradio":
-        main_file = "app.py"
-    else:
-        # Fallback: try common filenames
-        for fname in ["app.py", "index.html", "streamlit_app.py", "main.py"]:
-            try:
-                _ = api.hf_hub_download(repo_id=f"{username}/{project}", filename=fname, repo_type="space")
-                main_file = fname
-                break
-            except:
-                continue
     path = api.hf_hub_download(repo_id=f"{username}/{project}", filename=main_file, repo_type="space")
     with open(path, "r", encoding="utf-8") as f:
         return f.read()
 def load_project_from_url(url: str):
-    """Return (status_message, code_content) for importing a Space URL."""
     valid, user, proj = check_hf_space_url(url)
     if not valid:
         return "Error: Invalid Hugging Face Space URL.", ""
@@ -145,54 +99,22 @@ def load_project_from_url(url: str):
     except Exception as e:
         return f"Error fetching project: {e}", ""
-def deploy_to_user_space(code, space_name, sdk_choice, profile: gr.OAuthProfile | None = None, token: gr.OAuthToken | None = None):
-    """Create or update a user's own HF Space with proper authentication."""
     if not profile or not token or not token.token or token.token.startswith("hf_"):
         return gr.update(value="Please log in with a valid Hugging Face write token.", visible=True)
     api = HfApi(token=token.token)
-    is_update = "/" in space_name.strip()
-    if is_update:
-        repo_id = space_name.strip()
-    else:
-        repo_id = f"{profile.username}/{space_name.strip()}"
-    sdk_map = {
-        "Gradio (Python)": "gradio",
-        "Streamlit (Python)": "docker",
-        "Static (HTML)": "static",
-        "Transformers.js": "static"
-    }
-    sdk = sdk_map.get(sdk_choice, "gradio")
-    # Create or update repo
-    if not is_update and sdk != "docker":
         api.create_repo(repo_id=repo_id, repo_type="space", space_sdk=sdk, exist_ok=True)
-    # Duplicate and upload logic for docker (Streamlit) or transformers.js
-    if sdk == "docker":
-        dup = duplicate_space(from_id="streamlit/streamlit-template-space", to_id=repo_id, token=token.token, exist_ok=True)
-        with tempfile.NamedTemporaryFile("w", suffix=".py", delete=False) as f:
-            f.write(code)
-            path = f.name
-        api.upload_file(path_or_fileobj=path, path_in_repo="src/streamlit_app.py", repo_id=repo_id, repo_type="space")
-        os.unlink(path)
-    elif sdk_choice == "Transformers.js":
-        dup = duplicate_space(from_id="static-templates/transformers.js", to_id=repo_id, token=token.token, exist_ok=True)
-        from utils import parse_transformers_js_output  # ensure we have that helper
-        files = parse_transformers_js_output(code)
-        for name, content in files.items():
-            with tempfile.NamedTemporaryFile("w", suffix=f".{name.split('.')[-1]}", delete=False) as f:
-                f.write(content)
-                path = f.name
-            api.upload_file(path_or_fileobj=path, path_in_repo=name, repo_id=repo_id, repo_type="space")
-            os.unlink(path)
-    else:
-        # Static or Gradio: upload single file
-        filename = "index.html" if sdk == "static" else "app.py"
-        with tempfile.NamedTemporaryFile("w", suffix=f".{filename.split('.')[-1]}", delete=False) as f:
-            f.write(code)
-            path = f.name
-        api.upload_file(path_or_fileobj=path, path_in_repo=filename, repo_id=repo_id, repo_type="space")
-        os.unlink(path)
     return gr.update(value=f"✅ Deployed to https://huggingface.co/spaces/{repo_id}", visible=True)

 # ------------------------------------------------------------------
 def send_to_sandbox(code: str) -> str:
     wrapped = f"""
     <!DOCTYPE html>
     <html>
+    <head><meta charset=\"UTF-8\"><meta name=\"viewport\" content=\"width=device-width, initial-scale=1.0\"></head>
     <body>{code}</body>
     </html>
     """
     )
 def demo_card_click(e: gr.EventData) -> str:
+    idx = e.index if hasattr(e, 'index') else 0
     from constants import DEMO_LIST
+    idx = idx if 0 <= idx < len(DEMO_LIST) else 0
     return DEMO_LIST[idx]["description"]
 # ------------------------------------------------------------------
 # ------------------------------------------------------------------
 def wrap_html_in_gradio_app(html_code: str) -> str:
     safe = html_code.replace('"""', r'\"\"\"')
     return (
         "import gradio as gr\n\n"
     )
 def deploy_to_spaces(code: str) -> None:
     if not code.strip():
         return
     app_py = wrap_html_in_gradio_app(code)
+    params = urllib.parse.urlencode({"name": "new-space", "sdk": "gradio"})
+    files_params = urllib.parse.urlencode({"files[0][path]": "app.py", "files[0][content]": app_py})
     url = f"https://huggingface.co/new-space?{params}&{files_params}"
     webbrowser.open_new_tab(url)
 def wrap_html_in_static_app(html_code: str) -> str:
     return html_code
 def deploy_to_spaces_static(code: str) -> None:
     if not code.strip():
         return
     html = wrap_html_in_static_app(code)
+    params = urllib.parse.urlencode({"name": "new-space", "sdk": "static"})
+    files_params = urllib.parse.urlencode({"files[0][path]": "index.html", "files[0][content]": html})
     url = f"https://huggingface.co/new-space?{params}&{files_params}"
     webbrowser.open_new_tab(url)
 # ------------------------------------------------------------------
 def check_hf_space_url(url: str):
     import re
+    pattern = re.compile(r'^(?:https?://)?(?:huggingface\\.co|hf\\.co)/spaces/([\\w-]+)/([\\w-]+)$', re.IGNORECASE)
     m = pattern.match(url.strip())
+    return (False, None, None) if not m else (True, m.group(1), m.group(2))
 def fetch_hf_space_content(username: str, project: str) -> str:
     api = HfApi()
     info = api.space_info(f"{username}/{project}")
     sdk = info.sdk
+    main_file = "index.html" if sdk == "static" else "app.py"
     path = api.hf_hub_download(repo_id=f"{username}/{project}", filename=main_file, repo_type="space")
     with open(path, "r", encoding="utf-8") as f:
         return f.read()
 def load_project_from_url(url: str):
     valid, user, proj = check_hf_space_url(url)
     if not valid:
         return "Error: Invalid Hugging Face Space URL.", ""
     except Exception as e:
         return f"Error fetching project: {e}", ""
+def deploy_to_user_space(code, space_name, sdk_choice, profile: Optional[gr.OAuthProfile] = None, token: Optional[gr.OAuthToken] = None):
     if not profile or not token or not token.token or token.token.startswith("hf_"):
         return gr.update(value="Please log in with a valid Hugging Face write token.", visible=True)
     api = HfApi(token=token.token)
+    repo_id = space_name.strip() if "/" in space_name.strip() else f"{profile.username}/{space_name.strip()}"
+    sdk = {"Gradio (Python)": "gradio", "Streamlit (Python)": "docker", "Static (HTML)": "static", "Transformers.js": "static"}.get(sdk_choice, "gradio")
+    if "/" not in space_name and sdk != "docker":
         api.create_repo(repo_id=repo_id, repo_type="space", space_sdk=sdk, exist_ok=True)
+    filename = "index.html" if sdk == "static" else "app.py"
+    with tempfile.NamedTemporaryFile("w", suffix=f".{filename.split('.')[-1]}", delete=False) as f:
+        f.write(code)
+        path = f.name
+    api.upload_file(path_or_fileobj=path, path_in_repo=filename, repo_id=repo_id, repo_type="space")
+    os.unlink(path)
     return gr.update(value=f"✅ Deployed to https://huggingface.co/spaces/{repo_id}", visible=True)