Update app.py

app.py

@@ -284,8 +284,9 @@ def randomize_loras(selected_indices, loras_state):
 
 def download_loras_images(loras_json_orig: list[dict]):
     """
-    Überarbeitete Funktion zur Verarbeitung von Bild-URLs aus JSON-Daten und Repositories.
+    Optimierte Funktion zur Handhabung von Bild-URLs aus Repositories mit Fallback-Logik.
     """
+    default_placeholder = "/path/to/default-placeholder.png"  # Platzhalterbild für fehlende Bilder
     loras_json = []
 
     for lora in loras_json_orig:
@@ -295,30 +296,72 @@ def download_loras_images(loras_json_orig: list[dict]):
         # Standardwerte und Fallbacks
         lora["title"] = lora.get("title", "Unknown LoRA")
         lora["trigger_word"] = lora.get("trigger_word", "")
+        resolved_image_url = None
 
-        # URL-Validierung und Korrektur
-        if repo and image_url:
-            if not image_url.startswith("http"):
-                # URL für Bilder aus Repos korrekt erstellen
-                image_url = f"https://huggingface.co/{repo}/resolve/main/{image_url}"
+        # 1. Prüfen und Laden des Repository-Bildes
+        if repo:
+            repo_image_url = f"https://huggingface.co/{repo}/resolve/main/{image_url}" if image_url else None
             try:
-                response = requests.head(image_url)
-                if response.status_code != 200:
-                    raise ValueError(f"Bild-URL nicht erreichbar: {image_url}")
+                if repo_image_url and requests.head(repo_image_url).status_code == 200:
+                    resolved_image_url = repo_image_url
             except Exception as e:
-                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"
+                print(f"Fehler beim Laden des Repo-Bildes: {repo_image_url}: {e}")
+
+        # 2. Fallback: Laden des Bildes aus der JSON-URL (Hotlink)
+        if not resolved_image_url and image_url:
+            try:
+                if requests.head(image_url).status_code == 200:
+                    resolved_image_url = image_url
+            except Exception as e:
+                print(f"Fehler beim Laden des Hotlink-Bildes: {image_url}: {e}")
 
-        # Finales Bild in das JSON-Objekt einfügen
-        lora["image"] = image_url
+        # 3. Fallback: Platzhalterbild verwenden
+        lora["image"] = resolved_image_url if resolved_image_url else default_placeholder
         loras_json.append(lora)
 
     return loras_json
 
 
+def handle_gallery_click(selected_index, loras_state):
+    """
+    Handhabt den Klick auf ein Bild in der Galerie.
+    """
+    if selected_index is None:
+        # Kein Bild ausgewählt, zurück zur Galerie-Ansicht
+        return (
+            gr.update(visible=True),  # Galerie sichtbar
+            gr.update(visible=False),  # Großes Bild unsichtbar
+            None,  # Keine LoRA ausgewählt
+            None   # Keine LoRA-Skalierung
+        )
+
+    # LoRA-Daten für das ausgewählte Bild
+    selected_lora = loras_state[selected_index]
+    large_view_image = selected_lora["image"]
+    lora_title = selected_lora["title"]
+
+    return (
+        gr.update(visible=False),  # Galerie unsichtbar
+        gr.update(visible=True, value=large_view_image),  # Großes Bild sichtbar
+        selected_index,  # Ausgewählter Index
+        lora_title  # Titel des LoRA
+    )
+
+
+def toggle_large_view(selected_index, loras_state):
+    """
+    Schaltet zwischen Galerie-Ansicht und Großansicht um.
+    """
+    if selected_index is not None:
+        # Wechsel zur Galerie-Ansicht
+        return gr.update(visible=True), gr.update(visible=False)
+    else:
+        # Wechsel zur Großansicht
+        selected_lora = loras_state[selected_index]
+        large_view_image = selected_lora["image"]
+        return gr.update(visible=False), gr.update(visible=True, value=large_view_image)
+
+
 
 
 
@@ -908,15 +951,49 @@ 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("")
+                # Galerie-Komponente
                 gallery = gr.Gallery(
                     label="LoRA Gallery",
-                    value=[
-                        (lora["image"], lora["title"]) for lora in loras
-                    ],
+                    value=[(lora["image"], lora["title"]) for lora in loras],  # Loras initial
                     columns=4,
-                    interactive=False
+                    interactive=True  # Galerie interaktiv
+                )
+
+                # Großansicht für das ausgewählte Bild
+                    large_view = gr.Image(
+                    label="Selected Image",
+                    visible=False,  # Standardmäßig nicht sichtbar
+                    interactive=False  # Keine Interaktivität
+                )
+
+                # Select-Button, um das ausgewählte Bild zu übernehmen
+                select_button = gr.Button(
+                "Select",
+                visible=False  # Nur sichtbar, wenn ein Bild ausgewählt ist
                 )
 
+                # Event-Handler: Klick auf ein Galerie-Bild
+                gallery.select(
+                    handle_gallery_click,  # Funktion zum Verarbeiten des Galerie-Klicks
+                    inputs=[gallery, loras_state],  # Eingabe: Galerie-Index und State
+                    outputs=[gallery, large_view, select_button]  # Ausgabe: Galerie, Großansicht, Button
+                )
+
+                # Event-Handler: Klick auf den Select-Button
+                select_button.click(
+                    add_custom_lora,  # Vorhandene Funktion zur LoRA-Auswahl
+                    inputs=[selected_indices, loras_state],
+                    outputs=[gallery, large_view, select_button]
+                )
+
+                # Event-Handler: Klick auf das große Bild zum Zurückwechseln
+                large_view.click(
+                    toggle_large_view,  # Funktion zum Umschalten der Ansicht
+                    inputs=[selected_indices, loras_state],
+                    outputs=[gallery, large_view]
+                )
+
+
                 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)