TEST - Fixed list of version ids.

app.py

@@ -43,30 +43,35 @@ def download_file(url, file_path, folder, api_key=None):
         raise gr.Error(f"Error downloading file: {str(e)}")
 
         
-def get_files_by_username(username, api_key = None):
-    url = f"https://civitai.com/api/v1/models?username={username}&limit=100&nsfw=true"
-    if api_key: # Shows hidden models.
-        url += f"&token={api_key}"
+def get_files_by_username(username="loradude"):
+    # Fixed list of model IDs to check
+    LIST1 = [106794, 123456, 789012]  # Add/remove model IDs as needed
+    
     output = {}
-
-    while url:
-        response = requests.get(url, timeout=30)
-        data = response.json()
-        # Add current page items to the list
-        for model in data['items']:
-            for version in model['modelVersions']:
-                for file in version['files']:
-                    output[str(model['id']) + '/' + str(version['id']) + '/' + file['name']] = {
-                        'downloadUrl': file['downloadUrl'],
-                        'modelId': model['name'] + ' - ' + version['name'],
-                        'modelUrl': f"https://civitai.com/models/{model['id']}?modelVersionId={version['id']}",
-                        'author': model['creator']['username'],
-                        'authorUrl': f"https://civitai.com/user/{model['creator']['username']}",
-                        'mirrorUrl': f"https://civitaiarchive.com/models/{model['id']}?modelVersionId={version['id']}",
-                    }
-                    
-        metadata = data.get('metadata', {})
-        url = metadata.get('nextPage', None)
+    
+    for model_id in LIST1:
+        # Get model version data directly
+        version_url = f"https://civitai.com/api/v1/model-versions/{model_id}"
+        try:
+            response = requests.get(version_url, timeout=30)
+            response.raise_for_status()
+            version_data = response.json()
+            
+            # Process files (same structure as before)
+            for file in version_data['files']:
+                output[f"{version_data["modelId"]}/{version_data["id"]}/{file["name"]}"] = {
+                    'downloadUrl': file['downloadUrl'],
+                    'modelId': version_data["model"]["name"] + ' - ' + version_data["name"],
+                    'modelUrl': f"https://civitai.com/models/{version_data["modelId"]}?modelVersionId={version_data['id']}",
+                    'author': username,  # Hardcoded username
+                    'authorUrl': f"https://civitai.com/user/{username}",
+                    'mirrorUrl': f"https://civitaiarchive.com/models/{version_data["modelId"]}?modelVersionId={version_data['id']}",
+                }
+                
+        except Exception as e:
+            print(f"Failed to process model {model_id}: {str(e)}")
+            continue
+            
     return output
         
 def get_files_by_model_id(model_id):