Favourites support.

app.py

@@ -144,13 +144,43 @@ def get_files_by_model_id(model_id, api_key=None):
     except requests.exceptions.RequestException as e:
         raise gr.Error("Something went wrong in fetching CivitAI API")
 
-def process_url(url, profile, user_repo_id, oauth_token, folder, api_key=None, nsfw=None, hidden=None, indimages = False):
+def get_files_all(api_key=None, nsfw=None, hidden=None, favourites=None):
+    url = f"https://civitai.com/api/v1/models?limit=100"
+    output = {}
+    images = {}
+    headers = {}
+    if nsfw is not None:
+        url = url + f"&nsfw={BOOLPARM[nsfw]}"
+    if hidden is not None:
+        url = url + f"&hidden={BOOLPARM[hidden]}"
+    if favourites: # fav=false does nothing.
+        url = url + f"&favorites={BOOLPARM[favourites]}"
+    gr.Info(f"SBM url: {url}")
+    if api_key:
+        headers['Authorization'] = f'Bearer {api_key}'
+
+    while url:
+        response = requests.get(url, headers=headers, timeout=180)
+        data = response.json()
+        # Add current page items to the list
+        for model in data['items']:
+            (dfiles, dimages) = get_model_meta(model)
+            output.update(dfiles)
+            images.update(dimages)
+                    
+        metadata = data.get('metadata', {})
+        url = metadata.get('nextPage', None)
+    return (output, images)
+
+def process_url(url, profile, user_repo_id, oauth_token, folder, api_key=None, nsfw=None, hidden=None, favourites=None, indimages = False):
     if url.startswith("https://civitai.com/models/"):
         model_id = url.split('/')[4]
         (files, images) = get_files_by_model_id(model_id, api_key)
     elif url.startswith("https://civitai.com/user/"):
         username = url.split('/')[4]
-        (files, images) = get_files_by_username(username, api_key, nsfw, hidden)
+        (files, images) = get_files_by_username(username, api_key, nsfw, hidden, favourites)
+    elif favourites: # Can access all favourites if desired.
+        (files, images) = get_files_all(api_key, nsfw, hidden, favourites)
     else:
         raise gr.Error("Unknown CivitAI URL format, please provide model URL or user profile URL")
 
@@ -268,7 +298,8 @@ def add_mirror(repo_id):
 
 
 
-def upload_civit_to_hf(profile: Optional[gr.OAuthProfile], oauth_token: gr.OAuthToken, url, destination_repo, civitai_api_key=None, nsfw=None, hidden=None, indimages=False):
+def upload_civit_to_hf(profile: Optional[gr.OAuthProfile], oauth_token: gr.OAuthToken, url, destination_repo, civitai_api_key=None,
+                       nsfw=None, hidden=None, favourites=None, indimages=False):
     if not profile.name:
         raise gr.Error("Are you sure you are logged in?")
     
@@ -279,6 +310,9 @@ def upload_civit_to_hf(profile: Optional[gr.OAuthProfile], oauth_token: gr.OAuth
     if not re.match(r'^[a-zA-Z0-9_-]+$', destination_repo):
         raise gr.Error("Destination repository name must contain only alphanumeric characters, underscores, and hyphens")
 
+    if favourites and not civitai_api_key:
+        raise gr.Error("You must provide an api key to access your favourites.")
+
     # Convert bool flags.
     nsfw = BOOLPARM[nsfw]
     hidden = BOOLPARM[hidden]
@@ -299,7 +333,7 @@ def upload_civit_to_hf(profile: Optional[gr.OAuthProfile], oauth_token: gr.OAuth
         update_repo_visibility(repo_id=user_repo_id, private=False, token=oauth_token.token)
 
         gr.Info(f"SBM get images upload {indimages}")
-        (files, images) = process_url(url, profile, user_repo_id, oauth_token.token, folder, civitai_api_key, nsfw, hidden, indimages)
+        (files, images) = process_url(url, profile, user_repo_id, oauth_token.token, folder, civitai_api_key, nsfw, hidden, favourites, indimages)
         if not files or len(files.keys()) == 0:
             raise gr.Error("No files were copied. Something went wrong.")
         
@@ -380,6 +414,12 @@ Once uploaded, it will add this repository to CivitaiArchive.com as a mirror.
             label="Hidden models",
             info="Optional: Include, exclude or do not specify inclusion of hidden models.",
         )
+        include_favorites = gr.Checkbox(
+            value=False,
+            interactive=True, # Mandatory, otherwise disables the control.
+            label="Favourite models",
+            info="Optional: Include only favourite (liked) models. This requires a hash key."
+        )
         upload_images = gr.Checkbox(
             value=True,
             interactive=True, # Mandatory, otherwise disables the control.
@@ -395,7 +435,7 @@ Once uploaded, it will add this repository to CivitaiArchive.com as a mirror.
     
     submit_button_civit.click(
         fn=upload_civit_to_hf,
-        inputs=[submit_source_civit, destination_repo, civitai_api_key, include_nsfw, include_hidden, upload_images],
+        inputs=[submit_source_civit, destination_repo, civitai_api_key, include_nsfw, include_hidden, include_favorites, upload_images],
         outputs=[output]
     )