Update app.py

app.py

@@ -373,48 +373,40 @@ def select_lora(selected_indices, loras_state):
 
 
 
-def add_custom_lora(custom_lora, selected_indices, current_loras, gallery):
-    if custom_lora:
-        try:
+def add_custom_lora(custom_lora, selected_indices, current_loras, gallery, debug_log):
+    logs = debug_log
+    try:
+        logs = append_debug_log(f"Adding custom LoRA: {custom_lora}", logs)
+        if custom_lora:
             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:
+            logs = append_debug_log(f"Loaded custom LoRA: {repo}", logs)
+
+            if image is not None and "http" in image and not is_repo_public(repo):
                 try:
                     image = download_file_mod(image)
                 except Exception as e:
-                    print(e)
+                    logs = append_debug_log(f"Error downloading image: {e}", logs)
                     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 if image else get_custom_image(),  # Integration der neuen Funktion
+                current_loras.append({
+                    "image": image or "/path/to/default-placeholder.png",
                     "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)
-            
-            # 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.")
-
-            return current_loras, gr.update(value=gallery_items)
-        except Exception as e:
-            print(e)
-            gr.Warning(str(e))
-            return current_loras, gr.update()
-    else:
-        return current_loras, gr.update()
+                })
+                logs = append_debug_log(f"New LoRA added: {title}", logs)
 
+            gallery_items = [(item["image"], item["title"]) for item in current_loras]
+            return current_loras, gr.update(value=gallery_items), gr.update(value=logs)
+        else:
+            logs = append_debug_log("No custom LoRA provided.", logs)
+            return current_loras, gallery, gr.update(value=logs)
+    except Exception as e:
+        logs = append_debug_log(f"Error: {e}", logs)
+        return current_loras, gallery, gr.update(value=logs)
 
 
 
@@ -1109,8 +1101,8 @@ with gr.Blocks(theme='NoCrypt/miku@>=1.2.2', fill_width=True, css=css, delete_ca
     )
     add_custom_lora_button.click(
         add_custom_lora,
-        inputs=[custom_lora, selected_indices, loras_state, gallery],
-        outputs=[loras_state, gallery, selected_info_1, selected_info_2, selected_indices, lora_scale_1, lora_scale_2, lora_image_1, lora_image_2]
+        inputs=[custom_lora, selected_indices, loras_state, gallery, debug_log],
+        outputs=[loras_state, gallery, debug_log]
     )
     remove_custom_lora_button.click(
         remove_custom_lora,