Update build_logic.py

build_logic.py

@@ -23,13 +23,23 @@ logging.basicConfig(
 )
 logger = logging.getLogger(__name__)
 
-def hf_list_repos(user_name,token):
-    api=HfApi()
-    user_repos=[]
-    for space in api.list_spaces(user_name):
-        if space.id.startswith(user_name):  # Check if modelId starts with the username
-            user_repos.append(space.id)
-    return user_repos
+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):