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()