Create app.py

app.py

@@ -0,0 +1,173 @@
+import gradio as gr
+from huggingface_hub import HfApi, HfFolder
+from huggingface_hub.utils import HfHubHTTPError
+
+# Helper function to initialize the HfApi client
+def get_hf_api(token):
+    if not token:
+        raise gr.Error("Hugging Face API token is required.")
+    return HfApi(token=token)
+
+# --- Spaces Functions ---
+def list_user_spaces(token, author):
+    try:
+        api = get_hf_api(token)
+        spaces = api.list_spaces(author=author)
+        return "\n".join([space.id for space in spaces]) if spaces else "No spaces found for this user."
+    except HfHubHTTPError as e:
+        return f"Error: {e}"
+
+def create_space(token, repo_id, repo_type, space_sdk):
+    try:
+        api = get_hf_api(token)
+        repo_url = api.create_repo(repo_id=repo_id, repo_type=repo_type, space_sdk=space_sdk)
+        return f"Space created successfully! URL: {repo_url}"
+    except HfHubHTTPError as e:
+        return f"Error: {e}"
+
+def delete_space(token, repo_id):
+    try:
+        api = get_hf_api(token)
+        api.delete_repo(repo_id=repo_id, repo_type='space')
+        return f"Space '{repo_id}' deleted successfully."
+    except HfHubHTTPError as e:
+        return f"Error: {e}"
+
+# --- Models Functions ---
+def list_user_models(token, author):
+    try:
+        api = get_hf_api(token)
+        models = api.list_models(author=author)
+        return "\n".join([model.id for model in models]) if models else "No models found for this user."
+    except HfHubHTTPError as e:
+        return f"Error: {e}"
+
+def delete_model(token, repo_id):
+    try:
+        api = get_hf_api(token)
+        api.delete_repo(repo_id=repo_id, repo_type='model')
+        return f"Model '{repo_id}' deleted successfully."
+    except HfHubHTTPError as e:
+        return f"Error: {e}"
+
+# --- Datasets Functions ---
+def list_user_datasets(token, author):
+    try:
+        api = get_hf_api(token)
+        datasets = api.list_datasets(author=author)
+        return "\n".join([dataset.id for dataset in datasets]) if datasets else "No datasets found for this user."
+    except HfHubHTTPError as e:
+        return f"Error: {e}"
+
+def delete_dataset(token, repo_id):
+    try:
+        api = get_hf_api(token)
+        api.delete_repo(repo_id=repo_id, repo_type='dataset')
+        return f"Dataset '{repo_id}' deleted successfully."
+    except HfHubHTTPError as e:
+        return f"Error: {e}"
+
+# --- User Info Functions ---
+def whoami(token):
+    try:
+        api = get_hf_api(token)
+        user_info = api.whoami()
+        return user_info
+    except HfHubHTTPError as e:
+        return f"Error: {e}"
+
+def get_user_info(token, user):
+    try:
+        api = get_hf_api(token)
+        user_info = api.get_user_info(user)
+        return user_info
+    except HfHubHTTPError as e:
+        return f"Error: {e}"
+
+
+with gr.Blocks(theme=gr.themes.Soft()) as demo:
+    gr.Markdown("# Hugging Face Hub Toolkit")
+    gr.Markdown("A Gradio interface for interacting with the Hugging Face Hub. Enter your Hugging Face API token with 'write' permissions to use the tools.")
+
+    with gr.Row():
+        hf_token = gr.Textbox(label="Hugging Face API Token", type="password", placeholder="hf_...")
+
+    with gr.Tabs():
+        with gr.TabItem("Spaces"):
+            with gr.Row():
+                with gr.Column():
+                    gr.Markdown("### List User Spaces")
+                    list_spaces_author = gr.Textbox(label="Author (Username)")
+                    list_spaces_btn = gr.Button("List Spaces")
+                    list_spaces_output = gr.Textbox(label="User Spaces", lines=10)
+                with gr.Column():
+                    gr.Markdown("### Create a New Space")
+                    create_space_id = gr.Textbox(label="Space Repo ID (e.g., username/spacename)")
+                    create_space_sdk = gr.Dropdown(label="Space SDK", choices=["gradio", "streamlit", "docker", "static"])
+                    create_space_btn = gr.Button("Create Space")
+                    create_space_output = gr.Textbox(label="Result")
+                with gr.Column():
+                    gr.Markdown("### Delete a Space")
+                    delete_space_id = gr.Textbox(label="Space Repo ID to Delete")
+                    delete_space_btn = gr.Button("Delete Space")
+                    delete_space_output = gr.Textbox(label="Result")
+
+        with gr.TabItem("Models"):
+            with gr.Row():
+                with gr.Column():
+                    gr.Markdown("### List User Models")
+                    list_models_author = gr.Textbox(label="Author (Username)")
+                    list_models_btn = gr.Button("List Models")
+                    list_models_output = gr.Textbox(label="User Models", lines=10)
+                with gr.Column():
+                    gr.Markdown("### Delete a Model")
+                    delete_model_id = gr.Textbox(label="Model Repo ID to Delete")
+                    delete_model_btn = gr.Button("Delete Model")
+                    delete_model_output = gr.Textbox(label="Result")
+
+        with gr.TabItem("Datasets"):
+            with gr.Row():
+                with gr.Column():
+                    gr.Markdown("### List User Datasets")
+                    list_datasets_author = gr.Textbox(label="Author (Username)")
+                    list_datasets_btn = gr.Button("List Datasets")
+                    list_datasets_output = gr.Textbox(label="User Datasets", lines=10)
+                with gr.Column():
+                    gr.Markdown("### Delete a Dataset")
+                    delete_dataset_id = gr.Textbox(label="Dataset Repo ID to Delete")
+                    delete_dataset_btn = gr.Button("Delete Dataset")
+                    delete_dataset_output = gr.Textbox(label="Result")
+
+        with gr.TabItem("User Info"):
+            with gr.Row():
+                with gr.Column():
+                    gr.Markdown("### Who Am I?")
+                    whoami_btn = gr.Button("Get My Info")
+                    whoami_output = gr.JSON(label="My User Info")
+                with gr.Column():
+                    gr.Markdown("### Get User Info")
+                    get_user_info_user = gr.Textbox(label="Username")
+                    get_user_info_btn = gr.Button("Get User Info")
+                    get_user_info_output = gr.JSON(label="User Info")
+
+    # --- Event Handlers ---
+    # Spaces
+    list_spaces_btn.click(list_user_spaces, inputs=[hf_token, list_spaces_author], outputs=list_spaces_output)
+    create_space_btn.click(create_space, inputs=[hf_token, create_space_id, gr.Textbox(value="space", visible=False), create_space_sdk], outputs=create_space_output)
+    delete_space_btn.click(delete_space, inputs=[hf_token, delete_space_id], outputs=delete_space_output)
+
+    # Models
+    list_models_btn.click(list_user_models, inputs=[hf_token, list_models_author], outputs=list_models_output)
+    delete_model_btn.click(delete_model, inputs=[hf_token, delete_model_id], outputs=delete_model_output)
+
+    # Datasets
+    list_datasets_btn.click(list_user_datasets, inputs=[hf_token, list_datasets_author], outputs=list_datasets_output)
+    delete_dataset_btn.click(delete_dataset, inputs=[hf_token, delete_dataset_id], outputs=delete_dataset_output)
+
+    # User Info
+    whoami_btn.click(whoami, inputs=hf_token, outputs=whoami_output)
+    get_user_info_btn.click(get_user_info, inputs=[hf_token, get_user_info_user], outputs=get_user_info_output)
+
+
+if __name__ == "__main__":
+    demo.launch()