Update app.py

app.py

@@ -284,7 +284,7 @@ def randomize_loras(selected_indices, loras_state):
 
 def download_loras_images(loras_json_orig: list[dict]):
     """
-    Optimierte Funktion zur Handhabung und Korrektur von Bild-URLs aus Repositories.
+    Überarbeitete Funktion zur Verarbeitung von Bild-URLs aus JSON-Daten und Repositories.
     """
     loras_json = []
 
@@ -292,26 +292,28 @@ def download_loras_images(loras_json_orig: list[dict]):
         repo = lora.get("repo", None)
         image_url = lora.get("image", None)
 
-        # Standardwerte setzen
+        # Standardwerte und Fallbacks
         lora["title"] = lora.get("title", "Unknown LoRA")
         lora["trigger_word"] = lora.get("trigger_word", "")
 
-        # Prüfen und korrigieren der Bild-URL
+        # URL-Validierung und Korrektur
         if repo and image_url:
             if not image_url.startswith("http"):
-                # Korrekte URL aus Repo und Bildnamen erstellen
+                # URL für Bilder aus Repos korrekt erstellen
                 image_url = f"https://huggingface.co/{repo}/resolve/main/{image_url}"
             try:
-                # Validieren, ob die URL erreichbar ist
                 response = requests.head(image_url)
                 if response.status_code != 200:
-                    raise ValueError(f"Image URL not reachable: {image_url}")
+                    raise ValueError(f"Bild-URL nicht erreichbar: {image_url}")
             except Exception as e:
-                print(f"Error validating image URL: {e}")
-                image_url = None  # Fallback bei ungültiger URL
+                print(f"Fehler beim Laden des Bildes: {e}")
+                image_url = None
+        else:
+            # Falls keine Bild-URL oder Repo angegeben, Platzhalter verwenden
+            image_url = "/path/to/default-placeholder.png"
 
-        # Platzhalter verwenden, falls kein gültiges Bild verfügbar ist
-        lora["image"] = image_url if image_url else "/path/to/default-placeholder.png"
+        # Finales Bild in das JSON-Objekt einfügen
+        lora["image"] = image_url
         loras_json.append(lora)
 
     return loras_json
@@ -319,6 +321,7 @@ def download_loras_images(loras_json_orig: list[dict]):
 
 
 
+
 def add_custom_lora(custom_lora, selected_indices, current_loras, gallery):
     """
     Funktion zum Hinzufügen eines benutzerdefinierten LoRA (Low-Rank Adaptation).
@@ -403,20 +406,33 @@ def add_custom_lora(custom_lora, selected_indices, current_loras, gallery):
 
 
 
-def update_gallery_with_loras(loras):
+def update_gallery_with_loras(selected_indices, loras_state, gallery):
     """
-    Aktualisiert die Galerie mit den aktuellen LoRAs, auch bei fehlerhaften Einträgen.
+    Aktualisiert die Galerie basierend auf der Auswahl. Implementiert die Vorschau-Logik.
     """
-    gallery_items = []
-    for lora in loras:
-        try:
-            image = lora.get("image", "/path/to/default-placeholder.png")
-            title = lora.get("title", "Unknown LoRA")
-            gallery_items.append((image, title))
-        except Exception as e:
-            print(f"Fehler bei der Aktualisierung der Galerie: {e}")
-            gallery_items.append(("/path/to/default-placeholder.png", "Invalid LoRA Entry"))
-    return gallery_items
+    if not selected_indices:
+        # Galerieansicht: Keine Auswahl
+        gallery_items = [(lora["image"], lora["title"]) for lora in loras_state]
+        return gr.update(value=gallery_items), gr.update(visible=False), gr.update(visible=True)
+
+    # Vorschauansicht: Ein Bild wurde ausgewählt
+    selected_lora = loras_state[selected_indices[0]]  # Nur das erste ausgewählte Bild
+    preview_image = selected_lora["image"]
+    preview_title = selected_lora["title"]
+    preview_trigger_word = selected_lora.get("trigger_word", "")
+    preview_button_visible = True
+
+    # Micro-Thumbnails erstellen
+    micro_thumbnails = [(lora["image"], "") for lora in loras_state]
+
+    return (
+        gr.update(value=[(preview_image, preview_title)], visible=True),
+        gr.update(value=micro_thumbnails, visible=True),
+        gr.update(visible=False),  # Galerie deaktivieren
+        gr.update(value=preview_trigger_word, visible=True),
+        gr.update(visible=preview_button_visible),
+    )
+