Update build_logic.py

build_logic.py

@@ -23,25 +23,6 @@ logging.basicConfig(
 )
 logger = logging.getLogger(__name__)
 
-def list_user_repos(username):
-  api = HfApi()
-  user_repos = {}
-
-  # List models for the user
-  models = api.list_models(author=username)
-  user_repos['models'] = [model.id for model in models]
-
-  # List datasets for the user
-  datasets = api.list_datasets(author=username)
-  user_repos['datasets'] = [dataset.id for dataset in datasets]
-
-  # List Spaces for the user
-  spaces = api.list_spaces(author=username)
-  user_repos['spaces'] = [space.id for space in spaces]
-
-  return user_repos
-
-
 def _get_api_token(ui_token_from_textbox=None):
     env_token = os.getenv('HF_TOKEN')
     if env_token:
@@ -880,7 +861,23 @@ def list_user_spaces(hf_api_key, owner=None):
                  return None, f"Error auto-detecting owner for listing: {e}. Please specify Owner field."
 
         api = HfApi(token=token)
-        spaces = hf_list_repos(effective_owner,token=token)
+        #spaces = hf_list_repos(effective_owner,token=token)
+        
+        user_repos = {}
+        
+        # List models for the user
+        models = api.list_models(author=owner)
+        user_repos['models'] = [model.id for model in models]
+        
+        # List datasets for the user
+        datasets = api.list_datasets(author=owner)
+        user_repos['datasets'] = [dataset.id for dataset in datasets]
+        
+        # List Spaces for the user
+        spaces = api.list_spaces(author=owner)
+        user_repos['spaces'] = [space.id for space in spaces]
+
+        
         space_ids = [f"{r.author}/{r.id}" for r in spaces]
 
         logger.info(f"Successfully listed {len(space_ids)} spaces for {effective_owner}.")
@@ -896,4 +893,23 @@ def list_user_spaces(hf_api_key, owner=None):
         return None, f"HTTP Error ({status_code}) listing spaces for '{owner or 'authenticated user'}': {e_http.response.text if e_http.response else str(e_http)}"
     except Exception as e:
         logger.exception(f"Error listing spaces for {owner or 'authenticated user'}:")
-        return None, f"Error listing spaces: {e}"
+        return None, f"Error listing spaces: {e}"
+
+
+def list_user_spaces(username):
+  api = HfApi()
+  user_repos = {}
+
+  # List models for the user
+  models = api.list_models(author=username)
+  user_repos['models'] = [model.id for model in models]
+
+  # List datasets for the user
+  datasets = api.list_datasets(author=username)
+  user_repos['datasets'] = [dataset.id for dataset in datasets]
+
+  # List Spaces for the user
+  spaces = api.list_spaces(author=username)
+  user_repos['spaces'] = [space.id for space in spaces]
+
+  return user_repos