Update app.py

app.py

@@ -287,24 +287,41 @@ def download_loras_images(loras_json_orig: list[dict]):
     loras_json = []
     for lora in loras_json_orig:
         repo = lora.get("repo", None)
-        if repo is None or not api.repo_exists(repo_id=repo, token=HF_TOKEN):
-            print(f"LoRA '{repo}' is not exsit.")
-            continue
-        if "title" not in lora.keys() or "trigger_word" not in lora.keys() or "image" not in lora.keys():
-            title, _repo, _path, trigger_word, image_def = check_custom_model(repo)
-            if "title" not in lora.keys(): lora["title"] = title
-            if "trigger_word" not in lora.keys(): lora["trigger_word"] = trigger_word
-            if "image" not in lora.keys(): lora["image"] = image_def
+
+        # Behandlung für Hugging Face-Repos
+        if repo != "civitai":
+            if repo is None or not api.repo_exists(repo_id=repo, token=HF_TOKEN):
+                print(f"LoRA '{repo}' does not exist.")
+                continue
+        else:
+            # Behandlung für Civitai-Links
+            download_url = lora.get("url", None)
+            if not download_url:
+                print(f"No download URL provided for Civitai LoRA: {lora.get('title', 'Unknown')}")
+                continue
+            try:
+                # Download der LoRA-Datei
+                lora_path = download_file_mod(download_url)
+                lora["weights"] = lora_path
+            except Exception as e:
+                print(f"Failed to download Civitai LoRA '{lora.get('title', 'Unknown')}' from '{download_url}': {e}")
+                continue
+
+        # Herunterladen oder Verarbeiten von Bildern
         image = lora.get("image", None)
-        try:
-            if not is_repo_public(repo) and image is not None and "http" in image and repo in image: image = download_file_mod(image)
-            lora["image"] = image if image else "/home/user/app/custom.png"
-        except Exception as e:
-            print(f"Failed to download LoRA '{repo}''s image '{image if image else ''}'. {e}")
-            lora["image"] = "/home/user/app/custom.png"
+        if image:
+            try:
+                if repo != "civitai" and not is_repo_public(repo) and "http" in image and repo in image:
+                    image = download_file_mod(image)
+            except Exception as e:
+                print(f"Failed to download image for LoRA '{repo}': {e}")
+                image = "/home/user/app/custom.png"
+        lora["image"] = image if image else "/home/user/app/custom.png"
         loras_json.append(lora)
+
     return loras_json
 
+
 def add_custom_lora(custom_lora, selected_indices, current_loras, gallery):
     if custom_lora:
         try: