Update app.py

app.py

@@ -25,7 +25,29 @@ def get_my_spaces(hf_token):
         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:
@@ -109,7 +131,9 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
             placeholder="hf_...",
             scale=3,
         )
-        list_spaces_button = gr.Button("List My Public Spaces", variant="secondary", scale=1)
+        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
@@ -123,7 +147,7 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
         )
 
     make_private_button = gr.Button("Set Selected Spaces Private", variant="primary")
-    make_paused_button = gr.Button("Set Selected Spaces Paused", variant="secondary")
+    make_paused_button = gr.Button("Set Selected Spaces Paused", variant="primary")
     status_output = gr.Textbox(label="Status", interactive=False, lines=5)
 
     # --- Event Listeners ---
@@ -133,13 +157,23 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
         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.