Update app.py

app.py

@@ -321,17 +321,22 @@ def download_loras_images(loras_json_orig: list[dict]):
 
     return loras_json
 
+
 def handle_gallery_click(evt: gr.SelectData, loras_state):
     """Behandelt Klicks auf Galerie-Elemente."""
     selected_index = evt.index
     selected_lora = loras_state[selected_index]
 
-    # Großansicht vorbereiten
-    large_image = selected_lora["image"]
-    title = selected_lora["title"]
+    # Daten des ausgewählten LoRA-Elements abrufen
+    large_image = selected_lora.get("image", "/path/to/default-placeholder.png")
+    title = selected_lora.get("title", "Unknown LoRA")
 
     # Galerie ausblenden, Großansicht und Select-Button einblenden
-    return gr.update(visible=False), gr.update(value=large_image, visible=True), gr.update(visible=True)
+    return (
+        gr.update(visible=False),  # Galerie ausblenden
+        gr.update(value=large_image, visible=True),  # Großansicht anzeigen
+        gr.update(visible=True)  # Select-Button anzeigen
+    )
 
 def toggle_large_view(selected_indices, loras_state):
     """Kehrt von der Großansicht zur Galerie-Ansicht zurück."""
@@ -356,90 +361,68 @@ def toggle_large_view(selected_index, loras_state):
 
 
 
-def add_custom_lora(custom_lora, selected_indices, current_loras, gallery):
+ def add_custom_lora(custom_lora, selected_indices, current_loras, gallery):
     """
-    Funktion zum Hinzufügen eines benutzerdefinierten LoRA (Low-Rank Adaptation).
+    Fügt eine benutzerdefinierte LoRA hinzu und aktualisiert die Galerie.
     """
     if custom_lora:
         try:
-            # Überprüfung und Aufbereitung der Eingaben
+            # Überprüfen und Informationen zur LoRA abrufen
             title, repo, path, trigger_word, image = check_custom_model(custom_lora)
-
-            if image is not None and "http" in image and not is_repo_public(repo) and repo in image:
-                try:
-                    # Lade das Bild herunter, wenn es eine URL ist
-                    image = download_file_mod(image)
-                except Exception as e:
-                    print(f"Fehler beim Herunterladen des Bildes: {e}")
-                    image = None
-
-            print(f"Benutzerdefinierte LoRA geladen: {repo}")
             
-            # Prüfen, ob die LoRA bereits vorhanden ist
+            # URL-Validierung und Fallback-Logik anwenden
+            if image and "http" in image:
+                image = validate_and_fetch_image_url(repo, image)
+            elif repo:
+                image = f"https://huggingface.co/{repo}/resolve/main/{image}" if image else "/path/to/default-placeholder.png"
+            
+            # Prüfen, ob die LoRA bereits existiert
             existing_item_index = next((index for (index, item) in enumerate(current_loras) if item['repo'] == repo), None)
             if existing_item_index is None:
-                # Wenn es sich um eine *.safetensors-Datei handelt und die URL korrekt ist
-                if repo.endswith(".safetensors") and repo.startswith("http"):
-                    repo = download_file_mod(repo)
-                
-                # Neues LoRA-Objekt erstellen
                 new_item = {
-                    "image": image if image else "/path/to/default-placeholder.png",
+                    "image": image,
                     "title": title,
                     "repo": repo,
                     "weights": path,
                     "trigger_word": trigger_word
                 }
-                print(f"Neue LoRA hinzugefügt: {new_item}")
-                existing_item_index = len(current_loras)
                 current_loras.append(new_item)
+                existing_item_index = len(current_loras) - 1
 
             # Galerie-Elemente aktualisieren
-            gallery_items = update_gallery_with_loras(current_loras)
-
-            # Auswahl-Indizes aktualisieren
+            gallery_items = [(item["image"], item["title"]) for item in current_loras]
+            
+            # Indizes aktualisieren
             if len(selected_indices) < 2:
                 selected_indices.append(existing_item_index)
             else:
-                gr.Warning("Du kannst maximal 2 LoRAs auswählen. Entferne eine, um eine neue hinzuzufügen.")
-
-            # Ausgewählte Informationen und Bilder aktualisieren
-            selected_info_1 = "Select a LoRA 1"
-            selected_info_2 = "Select a LoRA 2"
-            lora_scale_1 = 1.15
-            lora_scale_2 = 1.15
-            lora_image_1 = None
-            lora_image_2 = None
-
-            if len(selected_indices) >= 1:
-                lora1 = current_loras[selected_indices[0]]
-                selected_info_1 = f"### LoRA 1 Selected: {lora1['title']} ✨"
-                lora_image_1 = lora1['image'] if lora1['image'] else None
-            if len(selected_indices) >= 2:
-                lora2 = current_loras[selected_indices[1]]
-                selected_info_2 = f"### LoRA 2 Selected: {lora2['title']} ✨"
-                lora_image_2 = lora2['image'] if lora2['image'] else None
-
-            print("Benutzerdefinierte LoRA erfolgreich hinzugefügt.")
+                gr.Warning("You can select up to 2 LoRAs. Remove one to select a new one.")
+
+            # Aktuelle Auswahl aktualisieren
+            selected_info_1 = f"### LoRA 1 Selected: {current_loras[selected_indices[0]]['title']}" if len(selected_indices) >= 1 else "Select a LoRA 1"
+            selected_info_2 = f"### LoRA 2 Selected: {current_loras[selected_indices[1]]['title']}" if len(selected_indices) >= 2 else "Select a LoRA 2"
+
             return (
                 current_loras,
                 gr.update(value=gallery_items),
                 selected_info_1,
                 selected_info_2,
                 selected_indices,
-                lora_scale_1,
-                lora_scale_2,
-                lora_image_1,
-                lora_image_2
+                gr.update(value=1.15),
+                gr.update(value=1.15),
+                current_loras[selected_indices[0]]['image'] if len(selected_indices) >= 1 else None,
+                current_loras[selected_indices[1]]['image'] if len(selected_indices) >= 2 else None
             )
         except Exception as e:
-            print(f"Fehler beim Hinzufügen der benutzerdefinierten LoRA: {e}")
+            print(f"Error adding custom LoRA: {e}")
+            gr.Warning(str(e))
             return current_loras, gr.update(), gr.update(), gr.update(), selected_indices, gr.update(), gr.update(), gr.update(), gr.update()
     else:
         return current_loras, gr.update(), gr.update(), gr.update(), selected_indices, gr.update(), gr.update(), gr.update(), gr.update()
 
 
 
+
 def update_gallery_with_loras(selected_indices, loras_state, gallery):
     """
     Aktualisiert die Galerie basierend auf der Auswahl. Implementiert die Vorschau-Logik.