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 von Bild-URLs aus Repositories.
+    Optimierte Funktion zur Handhabung und Korrektur von Bild-URLs aus Repositories.
     """
     loras_json = []
 
@@ -292,73 +292,88 @@ def download_loras_images(loras_json_orig: list[dict]):
         repo = lora.get("repo", None)
         image_url = lora.get("image", None)
 
-        # Standardwerte und Fallbacks
+        # Standardwerte setzen
         lora["title"] = lora.get("title", "Unknown LoRA")
         lora["trigger_word"] = lora.get("trigger_word", "")
-        
-        # Prüfen und bereinigen von URLs
+
+        # Prüfen und korrigieren der Bild-URL
         if repo and image_url:
             if not image_url.startswith("http"):
-                # Bild-URL korrekt erstellen, falls nur der Dateiname vorliegt
+                # Korrekte URL aus Repo und Bildnamen erstellen
                 image_url = f"https://huggingface.co/{repo}/resolve/main/{image_url}"
-            else:
-                # Validieren der bestehenden URL
-                try:
-                    if requests.head(image_url).status_code != 200:
-                        raise ValueError(f"Invalid image URL: {image_url}")
-                except Exception as e:
-                    print(f"Error loading external image: {e}")
-                    image_url = None
-        
-        # Wenn kein Bild gefunden, Platzhalter setzen
+            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}")
+            except Exception as e:
+                print(f"Error validating image URL: {e}")
+                image_url = None  # Fallback bei ungültiger URL
+
+        # Platzhalter verwenden, falls kein gültiges Bild verfügbar ist
         lora["image"] = image_url if image_url else "/path/to/default-placeholder.png"
         loras_json.append(lora)
 
     return loras_json
 
 
+
+
 def add_custom_lora(custom_lora, selected_indices, current_loras, gallery):
+    """
+    Funktion zum Hinzufügen eines benutzerdefinierten LoRA (Low-Rank Adaptation).
+    """
     if custom_lora:
         try:
+            # Überprüfung und Aufbereitung der Eingaben
             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(e)
+                    print(f"Fehler beim Herunterladen des Bildes: {e}")
                     image = None
-            print(f"Loaded custom LoRA: {repo}")
+
+            print(f"Benutzerdefinierte LoRA geladen: {repo}")
+            
+            # Prüfen, ob die LoRA bereits vorhanden ist
             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(repo)
                     repo = download_file_mod(repo)
+                
+                # Neues LoRA-Objekt erstellen
                 new_item = {
-                    "image": image if image else "/home/user/app/custom.png",
+                    "image": image if image else "/path/to/default-placeholder.png",
                     "title": title,
                     "repo": repo,
                     "weights": path,
                     "trigger_word": trigger_word
                 }
-                print(f"New LoRA: {new_item}")
+                print(f"Neue LoRA hinzugefügt: {new_item}")
                 existing_item_index = len(current_loras)
                 current_loras.append(new_item)
-            
-            # Update gallery
+
+            # Galerie-Elemente aktualisieren
             gallery_items = [(item["image"], item["title"]) for item in current_loras]
-            # Update selected_indices if there's room
+
+            # Auswahl-Indizes aktualisieren
             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.")
+                gr.Warning("Du kannst maximal 2 LoRAs auswählen. Entferne eine, um eine neue hinzuzufügen.")
 
-            # Update selected_info and images
+            # 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']} ✨"
@@ -367,24 +382,16 @@ def add_custom_lora(custom_lora, selected_indices, current_loras, gallery):
                 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("Finished adding custom LoRA")
+
+            print("Benutzerdefinierte LoRA erfolgreich hinzugefügt.")
             return (
                 current_loras,
                 gr.update(value=gallery_items),
-                selected_info_1, 
+                selected_info_1,
                 selected_info_2,
-                selected_indices,
-                lora_scale_1,
-                lora_scale_2,
-                lora_image_1,
-                lora_image_2
-            )
-        except Exception as 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()
-    else:
-        return current_loras, gr.update(), gr.update(), gr.update(), selected_indices, gr.update(), gr.update(), gr.update(), gr.update()
+                sele
+
+
 
 def remove_custom_lora(selected_indices, current_loras, gallery):
     if current_loras: