Update app.py

app.py

@@ -284,55 +284,39 @@ def randomize_loras(selected_indices, loras_state):
 
 def download_loras_images(loras_json_orig: list[dict]):
     """
-    Überarbeitete Funktion zur Handhabung von Vorschaubildern.
-    Prüft REPO-URLs zuerst, dann externe URLs, und verwendet das Platzhalterbild als Fallback.
+    Optimierte Funktion zur Handhabung von Bild-URLs aus Repositories.
     """
     loras_json = []
 
     for lora in loras_json_orig:
         repo = lora.get("repo", None)
+        image_url = lora.get("image", None)
 
-        # Standardwerte und Fallback setzen
+        # Standardwerte und Fallbacks
         lora["title"] = lora.get("title", "Unknown LoRA")
         lora["trigger_word"] = lora.get("trigger_word", "")
-        image_url = lora.get("image", None)
-
-        # Prüfen, ob die Bild-URL korrekt ist (vorzugsweise aus dem REPO)
-        if repo:
-            repo_image_url = f"https://huggingface.co/{repo}/resolve/main/{image_url}" if image_url else None
-            try:
-                if repo_image_url and requests.head(repo_image_url).status_code == 200:
-                    # REPO-URL verwenden, falls verfügbar
-                    lora["image"] = repo_image_url
-                else:
-                    raise ValueError(f"Image not found in repo: {repo_image_url}")
-            except Exception as e:
-                print(f"Error loading REPO image for '{repo}': {e}")
         
-        # Fallback zu externer URL (falls angegeben und gültig)
-        if "http" in image_url and lora["image"] != repo_image_url:
-            try:
-                if requests.head(image_url).status_code == 200:
-                    lora["image"] = image_url
-                else:
-                    raise ValueError("Invalid external image URL.")
-            except Exception as e:
-                print(f"Failed to use external image URL for '{repo}': {e}")
-
-        # Wenn keine gültige URL gefunden -> Platzhalter
-        if not lora.get("image"):
-            lora["image"] = "/home/user/app/custom.png"
-
+        # Prüfen und bereinigen von URLs
+        if repo and image_url:
+            if not image_url.startswith("http"):
+                # Bild-URL korrekt erstellen, falls nur der Dateiname vorliegt
+                image_url = f"https://huggingface.co/{repo}/resolve/main/{image_url}"
+            else:
+                # Validieren der bestehenden URL
+                try:
+                    if requests.head(image_url).status_code != 200:
+                        raise ValueError(f"Invalid image URL: {image_url}")
+                except Exception as e:
+                    print(f"Error loading external image: {e}")
+                    image_url = None
+        
+        # Wenn kein Bild gefunden, Platzhalter setzen
+        lora["image"] = image_url if image_url else "/path/to/default-placeholder.png"
         loras_json.append(lora)
 
     return loras_json
 
 
-
-
-
-
-
 def add_custom_lora(custom_lora, selected_indices, current_loras, gallery):
     if custom_lora:
         try:
@@ -874,8 +858,8 @@ with gr.Blocks(theme='NoCrypt/miku@>=1.2.2', fill_width=True, css=css, delete_ca
                 selected_info = gr.Markdown("")
                 gallery = gr.Gallery(
                     label="LoRA Gallery",
-                    elements=[
-                        (lora["image"], lora["title"]) for lora in loras  # `loras` enthält die Bild-URLs und Titel
+                    value=[
+                        (lora["image"], lora["title"]) for lora in loras
                     ],
                     columns=4,
                     interactive=False