Update app.py

app.py

@@ -283,42 +283,48 @@ def randomize_loras(selected_indices, loras_state):
     return selected_info_1, selected_info_2, selected_indices, lora_scale_1, lora_scale_2, lora_image_1, lora_image_2, random_prompt
 
 def download_loras_images(loras_json_orig: list[dict]):
+    api = HfApi(token=HF_TOKEN)
     loras_json = []
+
     for lora in loras_json_orig:
         repo = lora.get("repo", None)
-        image = lora.get("image", None)
 
         # Standardwerte setzen
         lora["title"] = lora.get("title", "Unknown LoRA")
         lora["trigger_word"] = lora.get("trigger_word", "")
-        default_image = "/home/user/app/custom.png"
+        lora["image"] = lora.get("image", "/home/user/app/custom.png")  # Fallback
 
-        try:
-            # 1. Falls Repo vorhanden, prüfe auf ein Bild im Repo
-            if repo:
-                repo_image_url = f"https://huggingface.co/{repo}/resolve/main/preview.png"
+        if repo and api.repo_exists(repo_id=repo, token=HF_TOKEN):
+            # Wenn das Repo existiert, prüfen, ob das Bild von dort geladen werden kann
+            try:
+                repo_image_url = f"https://huggingface.co/{repo}/resolve/main/{lora['image']}"
                 response = requests.head(repo_image_url)
                 if response.status_code == 200:
-                    lora["image"] = repo_image_url  # Repo-Bild gefunden
+                    # URL innerhalb des REPOs funktioniert
+                    lora["image"] = repo_image_url
                 else:
-                    print(f"Repo preview.png not found for {repo}, checking external image.")
+                    print(f"Repo image not found: {repo_image_url}")
+            except Exception as e:
+                print(f"Error checking repo image: {e}")
 
-            # 2. Prüfe die angegebene externe Bild-URL, falls vorhanden
-            if "http" in image:
+        # Prüfen auf externe URL (falls angegeben und gültig)
+        image = lora.get("image", None)
+        try:
+            if image and "http" in image:
+                # Bild-URL testen, falls extern
                 response = requests.head(image)
                 if response.status_code == 200:
-                    lora["image"] = image  # Externe Bild-URL ist gültig
+                    lora["image"] = image
                 else:
-                    print(f"External image URL responded with status {response.status_code}, using default.")
-
-            # 3. Falls keine gültige URL gefunden wurde, nutze das Standardbild
-            if not lora["image"]:
-                raise ValueError("No valid image found, using default.")
-
+                    raise ValueError("Invalid external image URL.")
+            else:
+                raise ValueError("Invalid or missing image URL.")
         except Exception as e:
-            print(f"Failed to retrieve image for LoRA '{repo}': {e}")
-            lora["image"] = default_image  # Fallback auf Standardbild
+            # Fehler beim Bild-Download -> Standardbild verwenden
+            print(f"Failed to download image for LoRA '{repo}': {e}")
+            lora["image"] = "/home/user/app/custom.png"
 
+        # Fügen Sie das LoRA in die Liste ein
         loras_json.append(lora)
 
     return loras_json
@@ -327,6 +333,7 @@ def download_loras_images(loras_json_orig: list[dict]):
 
 
 
+
 def add_custom_lora(custom_lora, selected_indices, current_loras, gallery):
     if custom_lora:
         try: