Update app.py

app.py

@@ -364,63 +364,47 @@ def select_lora(selected_indices, loras_state):
 
 
 def add_custom_lora(custom_lora, selected_indices, current_loras, gallery):
-    """
-    Fügt eine benutzerdefinierte LoRA hinzu und aktualisiert die Galerie.
-    """
     if custom_lora:
         try:
-            # Überprüfen und Informationen zur LoRA abrufen
             title, repo, path, trigger_word, image = check_custom_model(custom_lora)
-            
-            # 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
+            if image is not None and "http" in image and not is_repo_public(repo) and repo in image:
+                try:
+                    image = download_file_mod(image)
+                except Exception as e:
+                    print(e)
+                    image = None
+            print(f"Loaded custom LoRA: {repo}")
             existing_item_index = next((index for (index, item) in enumerate(current_loras) if item['repo'] == repo), None)
             if existing_item_index is None:
+                if repo.endswith(".safetensors") and repo.startswith("http"):
+                    repo = download_file_mod(repo)
                 new_item = {
-                    "image": image,
+                    "image": image if image else get_custom_image(),  # Integration der neuen Funktion
                     "title": title,
                     "repo": repo,
                     "weights": path,
                     "trigger_word": trigger_word
                 }
+                print(f"New LoRA: {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 = [(item["image"], item["title"]) for item in current_loras]
             
-            # Indizes aktualisieren
+            # Update gallery
+            gallery_items = [(item["image"], item["title"]) for item in current_loras]
+            # Update selected_indices if there's room
             if len(selected_indices) < 2:
                 selected_indices.append(existing_item_index)
             else:
-                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,
-                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
-            )
+                gr.Warning("You can select up to 2 LoRAs, remove one to select a new one.")
+
+            return current_loras, gr.update(value=gallery_items)
         except Exception as e:
-            print(f"Error adding custom LoRA: {e}")
+            print(e)
             gr.Warning(str(e))
-            return current_loras, gr.update(), gr.update(), gr.update(), selected_indices, gr.update(), gr.update(), gr.update(), gr.update()
+            return current_loras, gr.update()
     else:
-        return current_loras, gr.update(), gr.update(), gr.update(), selected_indices, gr.update(), gr.update(), gr.update(), gr.update()
+        return current_loras, gr.update()
+
 
 
 
@@ -453,7 +437,20 @@ def update_gallery_with_loras(selected_indices, loras_state, gallery):
     )
 
 
-
+def get_custom_image():
+    """
+    Liefert ein Bild für den Fallback.
+    Prüft zuerst das Repository, dann die URL, und setzt sonst ein Platzhalterbild.
+    """
+    try:
+        # Prüfen, ob ein benutzerdefiniertes Bild hochgeladen wurde
+        # Platzhalter: "/path/to/default-placeholder.png"
+        placeholder = "/path/to/default-placeholder.png"
+        # Logik für benutzerdefiniertes Bild hinzufügen
+        return placeholder
+    except Exception as e:
+        print(f"Error fetching custom image: {e}")
+        return "/path/to/default-placeholder.png"  # Sicherer Fallback