Update app.py

app.py

@@ -283,48 +283,46 @@ 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)
+    """
+    Überarbeitete Funktion zur Handhabung von Vorschaubildern.
+    Prüft REPO-URLs zuerst, dann externe URLs, und verwendet das Platzhalterbild als Fallback.
+    """
     loras_json = []
 
     for lora in loras_json_orig:
         repo = lora.get("repo", None)
 
-        # Standardwerte setzen
+        # Standardwerte und Fallback setzen
         lora["title"] = lora.get("title", "Unknown LoRA")
         lora["trigger_word"] = lora.get("trigger_word", "")
-        lora["image"] = lora.get("image", "/home/user/app/custom.png")  # Fallback
+        image_url = lora.get("image", None)
 
-        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
+        # 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:
-                repo_image_url = f"https://huggingface.co/{repo}/resolve/main/{lora['image']}"
-                response = requests.head(repo_image_url)
-                if response.status_code == 200:
-                    # URL innerhalb des REPOs funktioniert
+                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:
-                    print(f"Repo image not found: {repo_image_url}")
+                    raise ValueError(f"Image not found in repo: {repo_image_url}")
             except Exception as e:
-                print(f"Error checking repo image: {e}")
-
-        # 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
+                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.")
-            else:
-                raise ValueError("Invalid or missing image URL.")
-        except Exception as e:
-            # Fehler beim Bild-Download -> Standardbild verwenden
-            print(f"Failed to download image for LoRA '{repo}': {e}")
+            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"
 
-        # Fügen Sie das LoRA in die Liste ein
         loras_json.append(lora)
 
     return loras_json
@@ -334,6 +332,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:
@@ -873,8 +872,15 @@ with gr.Blocks(theme='NoCrypt/miku@>=1.2.2', fill_width=True, css=css, delete_ca
         with gr.Row():
             with gr.Column():
                 selected_info = gr.Markdown("")
-                gallery = gr.Gallery([(item["image"], item["title"]) for item in loras], label="LoRA Gallery", allow_preview=False,
-                                     columns=4, elem_id="gallery", show_share_button=False, interactive=False)
+                gallery = gr.Gallery(
+                    label="LoRA Gallery",
+                    elements=[
+                        (lora["image"], lora["title"]) for lora in loras  # `loras` enthält die Bild-URLs und Titel
+                    ],
+                    columns=4,
+                    interactive=False
+                )
+
                 with gr.Group():
                     with gr.Row(elem_id="custom_lora_structure"):
                         custom_lora = gr.Textbox(label="Custom LoRA", info="LoRA Hugging Face path or *.safetensors public URL", placeholder="multimodalart/vintage-ads-flux", scale=3, min_width=150)