Update app.py

app.py

@@ -1,8 +1,14 @@
 import gradio as gr
-from huggingface_hub import create_repo, list_models
+import requests
+from huggingface_hub import (
+    create_repo,
+    list_models,
+    upload_file,
+)
+
+# — CALLBACKS —
 
 def show_profile(profile: gr.OAuthProfile | None) -> str:
-    """Update login status."""
     if profile is None:
         return "*Not logged in.*"
     return f"✅ Logged in as **{profile.username}**"
@@ -11,15 +17,17 @@ def list_private_models(
     profile: gr.OAuthProfile | None,
     oauth_token: gr.OAuthToken | None
 ) -> str:
-    """Demo: list user’s private/public models."""
     if profile is None or oauth_token is None:
         return "Please log in to see your models."
     models = [
-        f"{model.id} ({'private' if model.private else 'public'})"
-        for model in list_models(author=profile.username, token=oauth_token.token)
+        f"{m.id} ({'private' if m.private else 'public'})"
+        for m in list_models(author=profile.username, token=oauth_token.token)
     ]
-    return "No models found." if not models \
+    return (
+        "No models found."
+        if not models
         else "Models:\n\n" + "\n - ".join(models)
+    )
 
 def create_space(
     repo_name: str,
@@ -27,10 +35,6 @@ def create_space(
     profile: gr.OAuthProfile | None,
     oauth_token: gr.OAuthToken | None
 ) -> tuple[str, str, str]:
-    """
-    Create (or get) a Hugging Face Space with the chosen SDK template,
-    returning (repo_id, logs, iframe).
-    """
     if profile is None or oauth_token is None:
         return "", "⚠️ Please log in first.", "<p>No Space created yet.</p>"
     repo_id = f"{profile.username}/{repo_name}"
@@ -46,46 +50,141 @@ def create_space(
     iframe = f'<iframe src="{url}" width="100%" height="500px"></iframe>'
     return repo_id, logmsg, iframe
 
-with gr.Blocks(title="HF OAuth + Space Creator") as demo:
+def enable_button(
+    profile: gr.OAuthProfile | None,
+    oauth_token: gr.OAuthToken | None,
+    repo_id: str
+):
+    """Enable an action button only once logged in & repo exists."""
+    return gr.update(interactive=bool(profile and oauth_token and repo_id))
+
+def upload_file_to_space(
+    file: gr.components.FileData | None,
+    path_in_repo: str,
+    repo_id: str,
+    oauth_token: gr.OAuthToken | None
+) -> str:
+    if oauth_token is None:
+        return "⚠️ Please log in first."
+    if not repo_id:
+        return "⚠️ Please create a Space first."
+    if file is None:
+        return "⚠️ No file selected."
+    # Gradio FileData gives us a path on disk:
+    upload_file(
+        path_or_fileobj=file.name,
+        path_in_repo=path_in_repo,
+        repo_id=repo_id,
+        token=oauth_token.token,
+        repo_type="space"
+    )
+    return f"✅ Uploaded `{path_in_repo}` to `{repo_id}`"  # :contentReference[oaicite:0]{index=0}
+
+def get_space_logs(
+    repo_id: str,
+    profile: gr.OAuthProfile | None,
+    oauth_token: gr.OAuthToken | None
+) -> str:
+    if profile is None or oauth_token is None:
+        return "⚠️ Please log in and create a Space first."
+    # This endpoint is currently undocumented; usage may change :contentReference[oaicite:1]{index=1}.
+    endpoint = f"https://huggingface.co/api/spaces/{repo_id}/logs"
+    headers = {"Authorization": f"Bearer {oauth_token.token}"}
+    resp = requests.get(endpoint, headers=headers)
+    if resp.status_code != 200:
+        return f"❌ Failed to fetch logs ({resp.status_code})"
+    entries = resp.json()  # expecting a list of {text, timestamp,…}
+    return "\n".join(e.get("text", "") for e in entries)
+
+# — BUILD UI —
+
+with gr.Blocks(title="HF OAuth + Space Manager") as demo:
     gr.Markdown(
-        "## Sign in with Hugging Face + Create a Space\n\n"
-        "1. Click **Sign in**.\n"
-        "2. Pick a template, enter a name, and **Create Space**.\n\n"
+        "## Sign in with Hugging Face + Manage Your Space\n\n"
+        "1. **Sign in**\n"
+        "2. **Create** a Space (Gradio/Streamlit)\n"
+        "3. **Upload** files to it\n"
+        "4. **Fetch** its runtime logs\n\n"
         "---"
     )
 
-    # — Login UI —
+    # — LOGIN STATUS & MODEL LIST —
     login_btn  = gr.LoginButton(variant="huggingface", size="lg")
     status_md  = gr.Markdown("*Not logged in.*")
     models_md  = gr.Markdown()
+    demo.load(show_profile,              inputs=None,           outputs=status_md)
+    login_btn.click(show_profile,        inputs=None,           outputs=status_md)
+    demo.load(list_private_models,       inputs=None,           outputs=models_md)
+    login_btn.click(list_private_models, inputs=None,           outputs=models_md)
 
-    demo.load(show_profile,             inputs=None, outputs=status_md)
-    login_btn.click(show_profile,       inputs=None, outputs=status_md)
-    demo.load(list_private_models,      inputs=None, outputs=models_md)
-    login_btn.click(list_private_models,inputs=None, outputs=models_md)
-
-    # — Create Space UI —
-    repo_name      = gr.Textbox(label="New Space name", placeholder="my-space-name")
-    sdk_selector   = gr.Radio(
+    # — CREATE SPACE —
+    repo_name    = gr.Textbox(label="New Space name", placeholder="my-space")
+    sdk_selector = gr.Radio(
         choices=["gradio", "streamlit"],
         value="gradio",
         label="Space template (SDK)"
     )
-    create_btn     = gr.Button("Create Space", interactive=False)
-    session_id     = gr.Textbox(visible=False)
-    logs           = gr.Textbox(label="Logs", interactive=False, lines=3)
-    preview_iframe = gr.HTML("<p>No Space created yet.</p>")
+    create_btn   = gr.Button("Create Space", interactive=False)
+    session_id   = gr.Textbox(visible=False)  # holds the repo_id
+    logs         = gr.Textbox(label="Logs", interactive=False, lines=3)
+    preview      = gr.HTML("<p>No Space created yet.</p>")
 
-    def enable_create(profile: gr.OAuthProfile | None):
-        return gr.update(interactive=profile is not None)
-
-    demo.load(enable_create,       inputs=None, outputs=create_btn)
-    login_btn.click(enable_create, inputs=None, outputs=create_btn)
+    demo.load(enable_button, 
+              inputs=[login_btn, login_btn, session_id], 
+              outputs=create_btn)
+    login_btn.click(enable_button, 
+                    inputs=[login_btn, login_btn, session_id], 
+                    outputs=create_btn)
 
     create_btn.click(
         fn=create_space,
-        inputs=[repo_name, sdk_selector],
-        outputs=[session_id, logs, preview_iframe]
+        inputs=[repo_name, sdk_selector, login_btn, login_btn],
+        outputs=[session_id, logs, preview]
+    )
+
+    # — UPLOAD FILES —
+    path_in_repo = gr.Textbox(label="Path in Space", value="app.py")
+    file_uploader= gr.File(label="Select file to upload")
+    upload_btn   = gr.Button("Upload File", interactive=False)
+    upload_logs  = gr.Textbox(label="Upload Logs", interactive=False, lines=2)
+
+    # Enable upload once the space exists and you’re signed in
+    session_id.change(
+        fn=enable_button,
+        inputs=[login_btn, login_btn, session_id],
+        outputs=[upload_btn]
+    )
+    login_btn.click(
+        fn=enable_button,
+        inputs=[login_btn, login_btn, session_id],
+        outputs=[upload_btn]
+    )
+
+    upload_btn.click(
+        fn=upload_file_to_space,
+        inputs=[file_uploader, path_in_repo, session_id, login_btn],
+        outputs=[upload_logs]
+    )
+
+    # — FETCH SPACE LOGS —
+    logs_btn      = gr.Button("Get Space Logs", interactive=False)
+    space_logs_md = gr.Textbox(label="Space Logs", interactive=False, lines=10)
+
+    session_id.change(
+        fn=enable_button,
+        inputs=[login_btn, login_btn, session_id],
+        outputs=[logs_btn]
+    )
+    login_btn.click(
+        fn=enable_button,
+        inputs=[login_btn, login_btn, session_id],
+        outputs=[logs_btn]
+    )
+
+    logs_btn.click(
+        fn=get_space_logs,
+        inputs=[session_id, login_btn, login_btn],
+        outputs=[space_logs_md]
     )
 
 if __name__ == "__main__":