Update app.py

app.py

@@ -358,7 +358,7 @@ def add_custom_lora(custom_lora, selected_indices, current_loras, gallery):
                 current_loras.append(new_item)
 
             # Galerie-Elemente aktualisieren
-            gallery_items = [(item["image"], item["title"]) for item in current_loras]
+            gallery_items = update_gallery_with_loras(current_loras)
 
             # Auswahl-Indizes aktualisieren
             if len(selected_indices) < 2:
@@ -389,7 +389,36 @@ def add_custom_lora(custom_lora, selected_indices, current_loras, gallery):
                 gr.update(value=gallery_items),
                 selected_info_1,
                 selected_info_2,
-                sele
+                selected_indices,
+                lora_scale_1,
+                lora_scale_2,
+                lora_image_1,
+                lora_image_2
+            )
+        except Exception as e:
+            print(f"Fehler beim Hinzufügen der benutzerdefinierten LoRA: {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(loras):
+    """
+    Aktualisiert die Galerie mit den aktuellen LoRAs, auch bei fehlerhaften Einträgen.
+    """
+    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
+
+