Spaces:

mgbam
/

builder

Running

App Files Files Community

mgbam commited on 25 days ago

Commit

3e1c4b9

verified ·

1 Parent(s): 1dcead8

Update deploy.py

Browse files

Files changed (1) hide show

deploy.py +178 -78

deploy.py CHANGED Viewed

@@ -1,98 +1,198 @@
 import gradio as gr
-from typing import Tuple, Optional
-def send_to_sandbox(code):
-    wrapped_code=f"""
     <!DOCTYPE html>
     <html>
-    <head>
-        <meta charset="UTF-8">
-        ...
-    </body>
     </html>
     """
-    encoded_html=base64.b64encode(wrapped_code.encode('utf-8')).decode('utf-8')
-    data_uri=f"data:text/html;charset=utf-8;base64,{encoded_html}"
-    return f'<iframe src="{data_uri}" width="100%" height="920px" sandbox="allow-scripts allow-same-origin allow-forms allow-popups allow-modals allow-presentation" allow="display-capture"></iframe>'
-def demo_card_click(e: gr.EventData):
-    ...
-    return DEMO_LIST[index]['description']
-def wrap_html_in_gradio_app(html_code):
-    safe_html=html_code.replace('"""',r'\"\"\"')
     return (
-        'import gradio as gr\n\n'
-        'def show_html():\n'
-        f'    return """{safe_html}"""\n\n'
-        'demo = gr.Interface(fn=show_html, inputs=None, outputs=gr.HTML())\n\n'
-        'if __name__ == "__main__":\n'
-        '    demo.launch()\n'
     )
-def deploy_to_spaces(code):
-    if not code.strip(): return
-    app_py=wrap_html_in_gradio_app(code.strip())
-    ...
-    webbrowser.open_new_tab(full_url)
-def wrap_html_in_static_app(html_code):
     return html_code
-def deploy_to_spaces_static(code):
-    if not code.strip(): return
-    app_html=wrap_html_in_static_app(code.strip())
-    ...
-    webbrowser.open_new_tab(full_url)
-def check_hf_space_url(url: str) -> Tuple[bool, Optional[str], Optional[str]]:
     import re
-    url_pattern=re.compile(
-        r'^(https?://)?(huggingface\.co|hf\.co)/spaces/([\w-]+)/([\w-]+)$',
-        re.IGNORECASE
-    )
-    match=url_pattern.match(url.strip())
-    if match:
-        return True, match.group(3), match.group(4)
-    return False, None, None
-def fetch_hf_space_content(username: str, project_name: str) -> str:
     try:
-        api=HfApi()
-        space_info=api.space_info(f"{username}/{project_name}")
-        sdk=space_info.sdk
-        ...
-        return f"""IMPORTED PROJECT FROM HUGGING FACE SPACE
-==============================================
-Space: {username}/{project_name}
-SDK: {sdk}
-Main File: {main_file}
-{file_content}"""
     except Exception as e:
-        return f"Error fetching space content: {str(e)}"
-def load_project_from_url(url: str) -> Tuple[str, str]:
-    is_valid,username,project_name=check_hf_space_url(url)
-    if not is_valid:
-        return "Error: Please enter a valid Hugging Face Spaces URL.\n\nExpected format: https://huggingface.co/spaces/username/project",""
-    content=fetch_hf_space_content(username,project_name)
-    if content.startswith("Error:"):
-        return content,""
-    lines=content.split('\n')
-    code_start=0
-    for i,line in enumerate(lines):
-        if line.strip() and not line.startswith(('=','IMPORTED','Space:','SDK:','Main File:')):
-            code_start=i
-            break
-    code_content='\n'.join(lines[code_start:])
-    return f"✅ Successfully imported project from {username}/{project_name}",code_content
-def deploy_to_user_space(code, space_name, sdk_name, profile=None, token=None):
-    import shutil
-    if not code.strip():
-        return gr.update(value="No code to deploy.", visible=True)
-    if profile is None or token is None:
-        return gr.update(value="Please log in with your Hugging Face account to deploy to your own Space. Otherwise, use the default deploy (opens in new tab).", visible=True)
-    ...
-    return gr.update(value=f"✅ {action_text}! [Open your Space here]({space_url})", visible=True)

+# deploy.py
+import os
+import base64
+import webbrowser
+import urllib.parse
+import tempfile
 import gradio as gr
+from huggingface_hub import HfApi, duplicate_space
+# ------------------------------------------------------------------
+# Utilities for live‐preview sandbox in the Gradio UI
+# ------------------------------------------------------------------
+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>
     """
+    b64 = base64.b64encode(wrapped.encode("utf-8")).decode("utf-8")
+    return (
+        f'<iframe src="data:text/html;base64,{b64}" '
+        'width="100%" height="920px" sandbox="allow-scripts allow-same-origin '
+        'allow-forms allow-popups allow-modals allow-presentation" allow="display-capture"></iframe>'
+    )
+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"]
+# ------------------------------------------------------------------
+# Functions for importing existing Hugging Face Spaces projects
+# ------------------------------------------------------------------
+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"
+        "def show_html():\n"
+        f'    return """{safe}"""\n\n'
+        "demo = gr.Interface(fn=show_html, inputs=None, outputs=gr.HTML())\n\n"
+        "if __name__ == '__main__':\n"
+        "    demo.launch()\n"
     )
+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)
+# ------------------------------------------------------------------
+# Functions for loading and updating user Spaces
+# ------------------------------------------------------------------
+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.", ""
     try:
+        content = fetch_hf_space_content(user, proj)
+        return f"✅ Imported {user}/{proj}", content
     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)