Spaces:
Sleeping
Sleeping
| import gradio as gr | |
| from huggingface_hub import HfApi | |
| from huggingface_hub.utils import HfHubHTTPError | |
| # --- Core Functions --- | |
| # Don't pause or privatize this_space | |
| this_space="broadfield-dev/HF-privatize" | |
| def get_my_spaces(hf_token): | |
| """Fetches a list of the user's public Hugging Face Spaces.""" | |
| if not hf_token: | |
| # Return empty values for all outputs | |
| return [], "Please enter your Hugging Face token.", [] | |
| try: | |
| api = HfApi(token=hf_token) | |
| # Get the username to list spaces accurately | |
| username = api.whoami()["name"] | |
| spaces = api.list_spaces(author=username) | |
| # Filter for only public spaces | |
| public_spaces = [space.id for space in spaces if not space.private] | |
| if not public_spaces: | |
| return [], "Successfully connected, but you have no public spaces.", [] | |
| # Return the list of spaces for the checkboxes and the state | |
| return public_spaces, f"Successfully listed {len(public_spaces)} public spaces.", public_spaces | |
| except HfHubHTTPError as e: | |
| # Handle common authentication errors | |
| return [], f"Authentication error: Please check your token. Details: {e}", [] | |
| except Exception as e: | |
| return [], f"An error occurred: {e}", [] | |
| def get_my_spaces_all(hf_token): | |
| """Fetches a list of the user's public Hugging Face Spaces.""" | |
| if not hf_token: | |
| # Return empty values for all outputs | |
| return [], "Please enter your Hugging Face token.", [] | |
| try: | |
| api = HfApi(token=hf_token) | |
| # Get the username to list spaces accurately | |
| username = api.whoami()["name"] | |
| spaces = api.list_spaces(author=username) | |
| # Filter for only public spaces | |
| public_spaces = [space.id for space in spaces] | |
| if not public_spaces: | |
| return [], "Successfully connected, but you have no public spaces.", [] | |
| # Return the list of spaces for the checkboxes and the state | |
| return public_spaces, f"Successfully listed {len(public_spaces)} public spaces.", public_spaces | |
| except HfHubHTTPError as e: | |
| # Handle common authentication errors | |
| return [], f"Authentication error: Please check your token. Details: {e}", [] | |
| except Exception as e: | |
| return [], f"An error occurred: {e}", [] | |
| def make_spaces_paused(hf_token, selected_spaces, progress=gr.Progress()): | |
| """Sets the visibility of selected Hugging Face Spaces to private.""" | |
| if not hf_token: | |
| return "Please enter your Hugging Face token." | |
| if not selected_spaces: | |
| return "Please select at least one space to Pause." | |
| progress(0, desc="Starting...") | |
| try: | |
| api = HfApi(token=hf_token) | |
| log_messages = [] | |
| total_spaces = len(selected_spaces) | |
| for i, space_id in enumerate(selected_spaces): | |
| progress((i + 1) / total_spaces, desc=f"Processing '{space_id}'...") | |
| try: | |
| if not space_id == this_space: | |
| api.pause_space(repo_id=space_id) | |
| log_messages.append(f"✅ Successfully made '{space_id}' paused.") | |
| except Exception as e: | |
| log_messages.append(f"❌ Failed to make '{space_id}' paused: {e}") | |
| return "\n".join(log_messages) | |
| except HfHubHTTPError as e: | |
| return f"Authentication error: Please check your token. Details: {e}" | |
| except Exception as e: | |
| return f"An error occurred: {e}" | |
| def make_spaces_private(hf_token, selected_spaces, progress=gr.Progress()): | |
| """Sets the visibility of selected Hugging Face Spaces to private.""" | |
| if not hf_token: | |
| return "Please enter your Hugging Face token." | |
| if not selected_spaces: | |
| return "Please select at least one space to make private." | |
| progress(0, desc="Starting...") | |
| try: | |
| api = HfApi(token=hf_token) | |
| log_messages = [] | |
| total_spaces = len(selected_spaces) | |
| for i, space_id in enumerate(selected_spaces): | |
| progress((i + 1) / total_spaces, desc=f"Processing '{space_id}'...") | |
| try: | |
| api.update_repo_settings(repo_id=space_id, private=True, repo_type="space") | |
| log_messages.append(f"✅ Successfully made '{space_id}' private.") | |
| except Exception as e: | |
| log_messages.append(f"❌ Failed to make '{space_id}' private: {e}") | |
| return "\n".join(log_messages) | |
| except HfHubHTTPError as e: | |
| return f"Authentication error: Please check your token. Details: {e}" | |
| except Exception as e: | |
| return f"An error occurred: {e}" | |
| # --- Gradio UI --- | |
| with gr.Blocks(theme=gr.themes.Soft()) as demo: | |
| gr.Markdown( | |
| """ | |
| # 🛡️ Hugging Face Space Privacy & Pause Control | |
| Quickly make your public Hugging Face Spaces private and paused. | |
| 1. Create and paste a Hugging Face access token with **write** permission. | |
| 2. Click "List My Public Spaces". | |
| 3. Select the spaces you want to make private or paused (or use "Select All"). | |
| 4. Click the final button to update their settings. | |
| """ | |
| ) | |
| with gr.Row(): | |
| hf_token_input = gr.Textbox( | |
| label="Hugging Face Write Token", | |
| type="password", | |
| placeholder="hf_...", | |
| scale=3, | |
| ) | |
| with gr.Column(): | |
| list_spaces_button = gr.Button("List My Public Spaces", variant="secondary", scale=1) | |
| list_spaces_button_all = gr.Button("List All My Spaces", variant="secondary", scale=1) | |
| with gr.Group(): | |
| # This (invisible) state component is the key to the select_all fix | |
| spaces_list_state = gr.State([]) | |
| select_all_checkbox = gr.Checkbox(label="Select All / Deselect All") | |
| spaces_checkboxes = gr.CheckboxGroup( | |
| label="Your Public Spaces", | |
| info="Only your public spaces are shown here." | |
| ) | |
| make_private_button = gr.Button("Set Selected Spaces Private", variant="primary") | |
| make_paused_button = gr.Button("Set Selected Spaces Paused", variant="primary") | |
| status_output = gr.Textbox(label="Status", interactive=False, lines=5) | |
| # --- Event Listeners --- | |
| def list_spaces_action(token): | |
| """Called when the list button is clicked. Updates the checkboxes and the state.""" | |
| spaces, message, state_list = get_my_spaces(token) | |
| # Also reset the select_all checkbox to unchecked | |
| return gr.update(choices=spaces, value=[]), message, state_list, gr.update(value=False) | |
| def list_spaces_action_all(token): | |
| """Called when the list button is clicked. Updates the checkboxes and the state.""" | |
| spaces, message, state_list = get_my_spaces_all(token) | |
| # Also reset the select_all checkbox to unchecked | |
| return gr.update(choices=spaces, value=[]), message, state_list, gr.update(value=False) | |
| list_spaces_button.click( | |
| fn=list_spaces_action, | |
| inputs=hf_token_input, | |
| outputs=[spaces_checkboxes, status_output, spaces_list_state, select_all_checkbox] | |
| ) | |
| list_spaces_button_all.click( | |
| fn=list_spaces_action_all, | |
| inputs=hf_token_input, | |
| outputs=[spaces_checkboxes, status_output, spaces_list_state, select_all_checkbox] | |
| ) | |
| def select_all_action(select_all_checked, full_list_of_spaces): | |
| """ | |
| Called when the "Select All" checkbox is changed. | |
| Uses the list from the state to update the checkbox group. | |
| """ | |
| if select_all_checked: | |
| return gr.update(value=full_list_of_spaces) # Select all | |
| else: | |
| return gr.update(value=[]) # Deselect all | |
| select_all_checkbox.change( | |
| fn=select_all_action, | |
| inputs=[select_all_checkbox, spaces_list_state], # Input is from the state now | |
| outputs=spaces_checkboxes | |
| ) | |
| make_private_button.click( | |
| fn=make_spaces_private, | |
| inputs=[hf_token_input, spaces_checkboxes], | |
| outputs=status_output | |
| ) | |
| make_paused_button.click( | |
| fn=make_spaces_paused, | |
| inputs=[hf_token_input, spaces_checkboxes], | |
| outputs=status_output | |
| ) | |
| demo.launch() |