test preview iamge model downlaod

files_cells/notebooks/en/downloading_en.ipynb

@@ -344,12 +344,51 @@
         "    \"adetailer\": adetailer_dir\n",
         "}\n",
         "\n",
-        "!mkdir -p {models_dir} {vaes_dir} {loras_dir} {embeddings_dir} {extensions_dir} {control_dir} {adetailer_dir}\n",
+        "# !mkdir -p {models_dir} {vaes_dir} {loras_dir} {embeddings_dir} {extensions_dir} {control_dir} {adetailer_dir}\n",
+        "directories = [value for key, value in prefixes.items()] # for unpucking zip files\n",
+        "!mkdir -p {\" \".join(directories)}\n",
         "\n",
         "url = \"\"\n",
         "hf_token = optional_huggingface_token if optional_huggingface_token else \"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\"\n",
         "user_header = f\"\\\"Authorization: Bearer {hf_token}\\\"\"\n",
         "\n",
+        "''' Get Image Preview | CivitAi '''\n",
+        "\n",
+        "def get_data_from_api(model_id): # get model data\n",
+        "    base_url = f\"https://civitai.com/api/v1/model-versions/{model_id}\"\n",
+        "    headers = {\"Content-Type\": \"application/json\"}\n",
+        "    try:\n",
+        "        response = requests.get(base_url, headers=headers)\n",
+        "        if response.status_code == 200:\n",
+        "            return response.json()\n",
+        "        else:\n",
+        "            print(f\"Failed to retrieve data. Status code: {response.status_code}\")\n",
+        "            return None\n",
+        "    except requests.exceptions.RequestException as e:\n",
+        "        print(f\"An error occurred: {e}\")\n",
+        "        return None\n",
+        "\n",
+        "def extract_file_and_image_info(data):\n",
+        "    files = data.get('files', [{}])\n",
+        "    model_name = files[0].get('name', None) # get original file name\n",
+        "    images = data.get('images', [{}])\n",
+        "    image_url = images[0].get('url', None) # get preview: first image\n",
+        "    return model_name, image_url\n",
+        "\n",
+        "def modify_image_url(image_url):\n",
+        "    parts = image_url.split('/')\n",
+        "    for i, part in enumerate(parts):\n",
+        "        if part.startswith('width='):\n",
+        "            width_value = int(part.split('=')[1])\n",
+        "            parts[i] = f'width={width_value * 2}' # resize for quality image\n",
+        "            break\n",
+        "    return '/'.join(parts)\n",
+        "\n",
+        "def generate_preview_filename(model_name, image_url):\n",
+        "    file_parts = model_name.split('.')\n",
+        "    image_format = image_url.split('.')[-1].split('?')[0]\n",
+        "    return f\"{file_parts[0]}.preview.{image_format}\" # assigning the original image format\n",
+        "\n",
         "''' main download code '''\n",
         "\n",
         "def handle_manual(url):\n",
@@ -374,11 +413,26 @@
         "    print(\"\\033[32m---\"*45 + f\"\\n\\033[33mURL: \\033[34m{url}\\n\\033[33mSAVE DIR: \\033[34m{dst_dir}\\n\\033[33mFILE NAME: \\033[34m{file_name}\\033[32m\\n~~~\\033[0m\")\n",
         "    # print(url, dst_dir, file_name)\n",
         "\n",
-        "    # I do it at my own risk..... Fucking CivitAi >:(\n",
+        "    # === CivitAi API ===\n",
         "    civitai_token = \"62c0c5956b2f9defbd844d754000180b\"\n",
         "    if 'civitai' in url and civitai_token:\n",
+        "        model_id = url.split('/')[-1]\n",
+        "        data = get_data_from_api(model_id)\n",
         "        url = f\"{url}?token={civitai_token}\"\n",
         "\n",
+        "        if data:\n",
+        "            model_name, image_url = extract_file_and_image_info(data)\n",
+        "            if model_name and image_url:\n",
+        "                new_image_url = modify_image_url(image_url)\n",
+        "                image_file_name = generate_preview_filename(model_name if not file_name else file_name, new_image_url)\n",
+        "                save_img_path = f\"{dst_dir}/{image_file_name}\"\n",
+        "                !wget -O {save_img_path} {new_image_url} # download image\n",
+        "                # print(f\"\\n\\n\\n{save_img_path, new_image_url}\\n\\n\\n\")\n",
+        "            else:\n",
+        "                print(\"File name or image URL missing.\")\n",
+        "        else:\n",
+        "            print(\"Failed to retrieve data from the API.\")\n",
+        "\n",
         "    # -- GDrive --\n",
         "    if 'drive.google' in url:\n",
         "        if 'folders' in url:\n",
@@ -423,8 +477,7 @@
         "\n",
         "## unpucking zip files\n",
         "def unpucking_zip_files():\n",
-        "    directories = [models_dir, vaes_dir, embeddings_dir, loras_dir , extensions_dir, control_dir , adetailer_dir]\n",
-        "\n",
+        "    # directories = [models_dir, vaes_dir, embeddings_dir, loras_dir , extensions_dir, control_dir , adetailer_dir]\n",
         "    for directory in directories:\n",
         "        for root, dirs, files in os.walk(directory):\n",
         "            for file in files:\n",
@@ -561,21 +614,21 @@
         "else:\n",
         "    if any(not file.endswith('.txt') for file in os.listdir(models_dir)):\n",
         "        print(\"\\n\\033[33m➤ Models\\033[0m\")\n",
-        "        get_ipython().system(\"find {models_dir}/ -mindepth 1 ! -name '*.txt' -printf '%f\\\\n'\")\n",
+        "        !find {models_dir}/ -mindepth 1 -name '*.safetensors' -printf '%f\\n'\n",
         "    if any(not file.endswith('.txt') for file in os.listdir(vaes_dir)):\n",
         "        print(\"\\n\\033[33m➤ VAEs\\033[0m\")\n",
-        "        get_ipython().system(\"find {vaes_dir}/ -mindepth 1 ! -name '*.txt' -printf '%f\\\\n'\")\n",
+        "        !find {vaes_dir}/ -mindepth 1 -name '*.safetensors' -printf '%f\\n'\n",
         "    if any(not file.endswith('.txt') and not os.path.isdir(os.path.join(embeddings_dir, file)) for file in os.listdir(embeddings_dir)):\n",
         "        print(\"\\n\\033[33m➤ Embeddings\\033[0m\")\n",
-        "        get_ipython().system(\"find {embeddings_dir}/ -mindepth 1 -maxdepth 1 \\\\( -name '*.pt' -or -name '*.safetensors' \\\\) -printf '%f\\\\n'\")\n",
+        "        !find {embeddings_dir}/ -mindepth 1 -maxdepth 1 \\( -name '*.pt' -or -name '*.safetensors' \\) -printf '%f\\n'\n",
         "    if any(not file.endswith('.txt') for file in os.listdir(loras_dir)):\n",
         "        print(\"\\n\\033[33m➤ LoRAs\\033[0m\")\n",
-        "        get_ipython().system(\"find {loras_dir}/ -mindepth 1 ! -name '*.keep' -printf '%f\\\\n'\")\n",
+        "        !find {loras_dir}/ -mindepth 1 -name '*.safetensors' -printf '%f\\n'\n",
         "    print(f\"\\n\\033[33m➤ Extensions\\033[0m\")\n",
-        "    get_ipython().system(\"find {extensions_dir}/ -mindepth 1 -maxdepth 1 ! -name '*.txt' -printf '%f\\\\n'\")\n",
+        "    !find {extensions_dir}/ -mindepth 1 -maxdepth 1 ! -name '*.txt' -printf '%f\\n'\n",
         "    if any(not file.endswith(('.txt', '.yaml')) for file in os.listdir(control_dir)):\n",
         "        print(\"\\n\\033[33m➤ ControlNet\\033[0m\")\n",
-        "        get_ipython().system(\"find {control_dir}/ -mindepth 1 ! -name '*.yaml' -printf '%f\\\\n' | sed 's/^[^_]*_[^_]*_[^_]*_\\\\(.*\\\\)_fp16\\\\.safetensors$/\\\\1/'\")\n",
+        "        !find {control_dir}/ -mindepth 1 ! -name '*.yaml' -printf '%f\\n' | sed 's/^[^_]*_[^_]*_[^_]*_\\([^_]*\\)_fp16\\.safetensors$/\\1/'\n",
         "\n",
         "\n",
         "# === OTHER ===\n",

files_cells/notebooks/ru/downloading_ru.ipynb

@@ -344,12 +344,51 @@
         "    \"adetailer\": adetailer_dir\n",
         "}\n",
         "\n",
-        "!mkdir -p {models_dir} {vaes_dir} {loras_dir} {embeddings_dir} {extensions_dir} {control_dir} {adetailer_dir}\n",
+        "# !mkdir -p {models_dir} {vaes_dir} {loras_dir} {embeddings_dir} {extensions_dir} {control_dir} {adetailer_dir}\n",
+        "directories = [value for key, value in prefixes.items()] # for unpucking zip files\n",
+        "!mkdir -p {\" \".join(directories)}\n",
         "\n",
         "url = \"\"\n",
         "hf_token = optional_huggingface_token if optional_huggingface_token else \"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\"\n",
         "user_header = f\"\\\"Authorization: Bearer {hf_token}\\\"\"\n",
         "\n",
+        "''' Get Image Preview | CivitAi '''\n",
+        "\n",
+        "def get_data_from_api(model_id): # get model data\n",
+        "    base_url = f\"https://civitai.com/api/v1/model-versions/{model_id}\"\n",
+        "    headers = {\"Content-Type\": \"application/json\"}\n",
+        "    try:\n",
+        "        response = requests.get(base_url, headers=headers)\n",
+        "        if response.status_code == 200:\n",
+        "            return response.json()\n",
+        "        else:\n",
+        "            print(f\"Failed to retrieve data. Status code: {response.status_code}\")\n",
+        "            return None\n",
+        "    except requests.exceptions.RequestException as e:\n",
+        "        print(f\"An error occurred: {e}\")\n",
+        "        return None\n",
+        "\n",
+        "def extract_file_and_image_info(data):\n",
+        "    files = data.get('files', [{}])\n",
+        "    model_name = files[0].get('name', None) # get original file name\n",
+        "    images = data.get('images', [{}])\n",
+        "    image_url = images[0].get('url', None) # get preview: first image\n",
+        "    return model_name, image_url\n",
+        "\n",
+        "def modify_image_url(image_url):\n",
+        "    parts = image_url.split('/')\n",
+        "    for i, part in enumerate(parts):\n",
+        "        if part.startswith('width='):\n",
+        "            width_value = int(part.split('=')[1])\n",
+        "            parts[i] = f'width={width_value * 2}' # resize for quality image\n",
+        "            break\n",
+        "    return '/'.join(parts)\n",
+        "\n",
+        "def generate_preview_filename(model_name, image_url):\n",
+        "    file_parts = model_name.split('.')\n",
+        "    image_format = image_url.split('.')[-1].split('?')[0]\n",
+        "    return f\"{file_parts[0]}.preview.{image_format}\" # assigning the original image format\n",
+        "\n",
         "''' main download code '''\n",
         "\n",
         "def handle_manual(url):\n",
@@ -374,11 +413,26 @@
         "    print(\"\\033[32m---\"*45 + f\"\\n\\033[33mURL: \\033[34m{url}\\n\\033[33mSAVE DIR: \\033[34m{dst_dir}\\n\\033[33mFILE NAME: \\033[34m{file_name}\\033[32m\\n~~~\\033[0m\")\n",
         "    # print(url, dst_dir, file_name)\n",
         "\n",
-        "    # I do it at my own risk..... Fucking CivitAi >:(\n",
+        "    # === CivitAi API ===\n",
         "    civitai_token = \"62c0c5956b2f9defbd844d754000180b\"\n",
         "    if 'civitai' in url and civitai_token:\n",
+        "        model_id = url.split('/')[-1]\n",
+        "        data = get_data_from_api(model_id)\n",
         "        url = f\"{url}?token={civitai_token}\"\n",
         "\n",
+        "        if data:\n",
+        "            model_name, image_url = extract_file_and_image_info(data)\n",
+        "            if model_name and image_url:\n",
+        "                new_image_url = modify_image_url(image_url)\n",
+        "                image_file_name = generate_preview_filename(model_name if not file_name else file_name, new_image_url)\n",
+        "                save_img_path = f\"{dst_dir}/{image_file_name}\"\n",
+        "                !wget -O {save_img_path} {new_image_url} # download image\n",
+        "                # print(f\"\\n\\n\\n{save_img_path, new_image_url}\\n\\n\\n\")\n",
+        "            else:\n",
+        "                print(\"File name or image URL missing.\")\n",
+        "        else:\n",
+        "            print(\"Failed to retrieve data from the API.\")\n",
+        "\n",
         "    # -- GDrive --\n",
         "    if 'drive.google' in url:\n",
         "        if 'folders' in url:\n",
@@ -388,7 +442,7 @@
         "               !gdown \"{url}\" -O {dst_dir}/{file_name} --fuzzy -c\n",
         "            else:\n",
         "               !gdown \"{url}\" -O {dst_dir} --fuzzy -c\n",
-        "    # -- Huggin Face --\n",
+        "    # -- Hugging Face --\n",
         "    elif 'huggingface' in url:\n",
         "        if '/blob/' in url:\n",
         "            url = url.replace('/blob/', '/resolve/')\n",
@@ -423,8 +477,7 @@
         "\n",
         "## unpucking zip files\n",
         "def unpucking_zip_files():\n",
-        "    directories = [models_dir, vaes_dir, embeddings_dir, loras_dir , extensions_dir, control_dir , adetailer_dir]\n",
-        "\n",
+        "    # directories = [models_dir, vaes_dir, embeddings_dir, loras_dir , extensions_dir, control_dir , adetailer_dir]\n",
         "    for directory in directories:\n",
         "        for root, dirs, files in os.walk(directory):\n",
         "            for file in files:\n",
@@ -561,21 +614,21 @@
         "else:\n",
         "    if any(not file.endswith('.txt') for file in os.listdir(models_dir)):\n",
         "        print(\"\\n\\033[33m➤ Models\\033[0m\")\n",
-        "        get_ipython().system(\"find {models_dir}/ -mindepth 1 ! -name '*.txt' -printf '%f\\\\n'\")\n",
+        "        !find {models_dir}/ -mindepth 1 -name '*.safetensors' -printf '%f\\n'\n",
         "    if any(not file.endswith('.txt') for file in os.listdir(vaes_dir)):\n",
         "        print(\"\\n\\033[33m➤ VAEs\\033[0m\")\n",
-        "        get_ipython().system(\"find {vaes_dir}/ -mindepth 1 ! -name '*.txt' -printf '%f\\\\n'\")\n",
+        "        !find {vaes_dir}/ -mindepth 1 -name '*.safetensors' -printf '%f\\n'\n",
         "    if any(not file.endswith('.txt') and not os.path.isdir(os.path.join(embeddings_dir, file)) for file in os.listdir(embeddings_dir)):\n",
         "        print(\"\\n\\033[33m➤ Embeddings\\033[0m\")\n",
-        "        get_ipython().system(\"find {embeddings_dir}/ -mindepth 1 -maxdepth 1 \\\\( -name '*.pt' -or -name '*.safetensors' \\\\) -printf '%f\\\\n'\")\n",
+        "        !find {embeddings_dir}/ -mindepth 1 -maxdepth 1 \\( -name '*.pt' -or -name '*.safetensors' \\) -printf '%f\\n'\n",
         "    if any(not file.endswith('.txt') for file in os.listdir(loras_dir)):\n",
         "        print(\"\\n\\033[33m➤ LoRAs\\033[0m\")\n",
-        "        get_ipython().system(\"find {loras_dir}/ -mindepth 1 ! -name '*.keep' -printf '%f\\\\n'\")\n",
+        "        !find {loras_dir}/ -mindepth 1 -name '*.safetensors' -printf '%f\\n'\n",
         "    print(f\"\\n\\033[33m➤ Extensions\\033[0m\")\n",
-        "    get_ipython().system(\"find {extensions_dir}/ -mindepth 1 -maxdepth 1 ! -name '*.txt' -printf '%f\\\\n'\")\n",
+        "    !find {extensions_dir}/ -mindepth 1 -maxdepth 1 ! -name '*.txt' -printf '%f\\n'\n",
         "    if any(not file.endswith(('.txt', '.yaml')) for file in os.listdir(control_dir)):\n",
         "        print(\"\\n\\033[33m➤ ControlNet\\033[0m\")\n",
-        "        get_ipython().system(\"find {control_dir}/ -mindepth 1 ! -name '*.yaml' -printf '%f\\\\n' | sed 's/^[^_]*_[^_]*_[^_]*_\\\\(.*\\\\)_fp16\\\\.safetensors$/\\\\1/'\")\n",
+        "        !find {control_dir}/ -mindepth 1 ! -name '*.yaml' -printf '%f\\n' | sed 's/^[^_]*_[^_]*_[^_]*_\\([^_]*\\)_fp16\\.safetensors$/\\1/'\n",
         "\n",
         "\n",
         "# === OTHER ===\n",

files_cells/python/en/downloading_en.py

@@ -325,12 +325,51 @@ prefixes = {
     "adetailer": adetailer_dir
 }
 
-get_ipython().system('mkdir -p {models_dir} {vaes_dir} {loras_dir} {embeddings_dir} {extensions_dir} {control_dir} {adetailer_dir}')
+# !mkdir -p {models_dir} {vaes_dir} {loras_dir} {embeddings_dir} {extensions_dir} {control_dir} {adetailer_dir}
+directories = [value for key, value in prefixes.items()] # for unpucking zip files
+get_ipython().system('mkdir -p {" ".join(directories)}')
 
 url = ""
 hf_token = optional_huggingface_token if optional_huggingface_token else "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"
 user_header = f"\"Authorization: Bearer {hf_token}\""
 
+''' Get Image Preview | CivitAi '''
+
+def get_data_from_api(model_id): # get model data
+    base_url = f"https://civitai.com/api/v1/model-versions/{model_id}"
+    headers = {"Content-Type": "application/json"}
+    try:
+        response = requests.get(base_url, headers=headers)
+        if response.status_code == 200:
+            return response.json()
+        else:
+            print(f"Failed to retrieve data. Status code: {response.status_code}")
+            return None
+    except requests.exceptions.RequestException as e:
+        print(f"An error occurred: {e}")
+        return None
+
+def extract_file_and_image_info(data):
+    files = data.get('files', [{}])
+    model_name = files[0].get('name', None) # get original file name
+    images = data.get('images', [{}])
+    image_url = images[0].get('url', None) # get preview: first image
+    return model_name, image_url
+
+def modify_image_url(image_url):
+    parts = image_url.split('/')
+    for i, part in enumerate(parts):
+        if part.startswith('width='):
+            width_value = int(part.split('=')[1])
+            parts[i] = f'width={width_value * 2}' # resize for quality image
+            break
+    return '/'.join(parts)
+
+def generate_preview_filename(model_name, image_url):
+    file_parts = model_name.split('.')
+    image_format = image_url.split('.')[-1].split('?')[0]
+    return f"{file_parts[0]}.preview.{image_format}" # assigning the original image format
+
 ''' main download code '''
 
 def handle_manual(url):
@@ -355,11 +394,26 @@ def manual_download(url, dst_dir, file_name):
     print("\033[32m---"*45 + f"\n\033[33mURL: \033[34m{url}\n\033[33mSAVE DIR: \033[34m{dst_dir}\n\033[33mFILE NAME: \033[34m{file_name}\033[32m\n~~~\033[0m")
     # print(url, dst_dir, file_name)
 
-    # I do it at my own risk..... Fucking CivitAi >:(
+    # === CivitAi API ===
     civitai_token = "62c0c5956b2f9defbd844d754000180b"
     if 'civitai' in url and civitai_token:
+        model_id = url.split('/')[-1]
+        data = get_data_from_api(model_id)
         url = f"{url}?token={civitai_token}"
 
+        if data:
+            model_name, image_url = extract_file_and_image_info(data)
+            if model_name and image_url:
+                new_image_url = modify_image_url(image_url)
+                image_file_name = generate_preview_filename(model_name if not file_name else file_name, new_image_url)
+                save_img_path = f"{dst_dir}/{image_file_name}"
+                get_ipython().system('wget -O {save_img_path} {new_image_url} # download image')
+                # print(f"\n\n\n{save_img_path, new_image_url}\n\n\n")
+            else:
+                print("File name or image URL missing.")
+        else:
+            print("Failed to retrieve data from the API.")
+
     # -- GDrive --
     if 'drive.google' in url:
         if 'folders' in url:
@@ -404,8 +458,7 @@ def download(url):
 
 ## unpucking zip files
 def unpucking_zip_files():
-    directories = [models_dir, vaes_dir, embeddings_dir, loras_dir , extensions_dir, control_dir , adetailer_dir]
-
+    # directories = [models_dir, vaes_dir, embeddings_dir, loras_dir , extensions_dir, control_dir , adetailer_dir]
     for directory in directories:
         for root, dirs, files in os.walk(directory):
             for file in files:
@@ -539,21 +592,21 @@ if detailed_download == "off":
 else:
     if any(not file.endswith('.txt') for file in os.listdir(models_dir)):
         print("\n\033[33m➤ Models\033[0m")
-        get_ipython().system("find {models_dir}/ -mindepth 1 ! -name '*.txt' -printf '%f\\n'")
+        get_ipython().system("find {models_dir}/ -mindepth 1 -name '*.safetensors' -printf '%f\\n'")
     if any(not file.endswith('.txt') for file in os.listdir(vaes_dir)):
         print("\n\033[33m➤ VAEs\033[0m")
-        get_ipython().system("find {vaes_dir}/ -mindepth 1 ! -name '*.txt' -printf '%f\\n'")
+        get_ipython().system("find {vaes_dir}/ -mindepth 1 -name '*.safetensors' -printf '%f\\n'")
     if any(not file.endswith('.txt') and not os.path.isdir(os.path.join(embeddings_dir, file)) for file in os.listdir(embeddings_dir)):
         print("\n\033[33m➤ Embeddings\033[0m")
         get_ipython().system("find {embeddings_dir}/ -mindepth 1 -maxdepth 1 \\( -name '*.pt' -or -name '*.safetensors' \\) -printf '%f\\n'")
     if any(not file.endswith('.txt') for file in os.listdir(loras_dir)):
         print("\n\033[33m➤ LoRAs\033[0m")
-        get_ipython().system("find {loras_dir}/ -mindepth 1 ! -name '*.keep' -printf '%f\\n'")
+        get_ipython().system("find {loras_dir}/ -mindepth 1 -name '*.safetensors' -printf '%f\\n'")
     print(f"\n\033[33m➤ Extensions\033[0m")
     get_ipython().system("find {extensions_dir}/ -mindepth 1 -maxdepth 1 ! -name '*.txt' -printf '%f\\n'")
     if any(not file.endswith(('.txt', '.yaml')) for file in os.listdir(control_dir)):
         print("\n\033[33m➤ ControlNet\033[0m")
-        get_ipython().system("find {control_dir}/ -mindepth 1 ! -name '*.yaml' -printf '%f\\n' | sed 's/^[^_]*_[^_]*_[^_]*_\\(.*\\)_fp16\\.safetensors$/\\1/'")
+        get_ipython().system("find {control_dir}/ -mindepth 1 ! -name '*.yaml' -printf '%f\\n' | sed 's/^[^_]*_[^_]*_[^_]*_\\([^_]*\\)_fp16\\.safetensors$/\\1/'")
 
 
 # === OTHER ===

files_cells/python/ru/downloading_ru.py

@@ -325,12 +325,51 @@ prefixes = {
     "adetailer": adetailer_dir
 }
 
-get_ipython().system('mkdir -p {models_dir} {vaes_dir} {loras_dir} {embeddings_dir} {extensions_dir} {control_dir} {adetailer_dir}')
+# !mkdir -p {models_dir} {vaes_dir} {loras_dir} {embeddings_dir} {extensions_dir} {control_dir} {adetailer_dir}
+directories = [value for key, value in prefixes.items()] # for unpucking zip files
+get_ipython().system('mkdir -p {" ".join(directories)}')
 
 url = ""
 hf_token = optional_huggingface_token if optional_huggingface_token else "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"
 user_header = f"\"Authorization: Bearer {hf_token}\""
 
+''' Get Image Preview | CivitAi '''
+
+def get_data_from_api(model_id): # get model data
+    base_url = f"https://civitai.com/api/v1/model-versions/{model_id}"
+    headers = {"Content-Type": "application/json"}
+    try:
+        response = requests.get(base_url, headers=headers)
+        if response.status_code == 200:
+            return response.json()
+        else:
+            print(f"Failed to retrieve data. Status code: {response.status_code}")
+            return None
+    except requests.exceptions.RequestException as e:
+        print(f"An error occurred: {e}")
+        return None
+
+def extract_file_and_image_info(data):
+    files = data.get('files', [{}])
+    model_name = files[0].get('name', None) # get original file name
+    images = data.get('images', [{}])
+    image_url = images[0].get('url', None) # get preview: first image
+    return model_name, image_url
+
+def modify_image_url(image_url):
+    parts = image_url.split('/')
+    for i, part in enumerate(parts):
+        if part.startswith('width='):
+            width_value = int(part.split('=')[1])
+            parts[i] = f'width={width_value * 2}' # resize for quality image
+            break
+    return '/'.join(parts)
+
+def generate_preview_filename(model_name, image_url):
+    file_parts = model_name.split('.')
+    image_format = image_url.split('.')[-1].split('?')[0]
+    return f"{file_parts[0]}.preview.{image_format}" # assigning the original image format
+
 ''' main download code '''
 
 def handle_manual(url):
@@ -355,11 +394,26 @@ def manual_download(url, dst_dir, file_name):
     print("\033[32m---"*45 + f"\n\033[33mURL: \033[34m{url}\n\033[33mSAVE DIR: \033[34m{dst_dir}\n\033[33mFILE NAME: \033[34m{file_name}\033[32m\n~~~\033[0m")
     # print(url, dst_dir, file_name)
 
-    # I do it at my own risk..... Fucking CivitAi >:(
+    # === CivitAi API ===
     civitai_token = "62c0c5956b2f9defbd844d754000180b"
     if 'civitai' in url and civitai_token:
+        model_id = url.split('/')[-1]
+        data = get_data_from_api(model_id)
         url = f"{url}?token={civitai_token}"
 
+        if data:
+            model_name, image_url = extract_file_and_image_info(data)
+            if model_name and image_url:
+                new_image_url = modify_image_url(image_url)
+                image_file_name = generate_preview_filename(model_name if not file_name else file_name, new_image_url)
+                save_img_path = f"{dst_dir}/{image_file_name}"
+                get_ipython().system('wget -O {save_img_path} {new_image_url} # download image')
+                # print(f"\n\n\n{save_img_path, new_image_url}\n\n\n")
+            else:
+                print("File name or image URL missing.")
+        else:
+            print("Failed to retrieve data from the API.")
+
     # -- GDrive --
     if 'drive.google' in url:
         if 'folders' in url:
@@ -369,7 +423,7 @@ def manual_download(url, dst_dir, file_name):
                get_ipython().system('gdown "{url}" -O {dst_dir}/{file_name} --fuzzy -c')
             else:
                get_ipython().system('gdown "{url}" -O {dst_dir} --fuzzy -c')
-    # -- Huggin Face --
+    # -- Hugging Face --
     elif 'huggingface' in url:
         if '/blob/' in url:
             url = url.replace('/blob/', '/resolve/')
@@ -404,8 +458,7 @@ def download(url):
 
 ## unpucking zip files
 def unpucking_zip_files():
-    directories = [models_dir, vaes_dir, embeddings_dir, loras_dir , extensions_dir, control_dir , adetailer_dir]
-
+    # directories = [models_dir, vaes_dir, embeddings_dir, loras_dir , extensions_dir, control_dir , adetailer_dir]
     for directory in directories:
         for root, dirs, files in os.walk(directory):
             for file in files:
@@ -539,21 +592,21 @@ if detailed_download == "off":
 else:
     if any(not file.endswith('.txt') for file in os.listdir(models_dir)):
         print("\n\033[33m➤ Models\033[0m")
-        get_ipython().system("find {models_dir}/ -mindepth 1 ! -name '*.txt' -printf '%f\\n'")
+        get_ipython().system("find {models_dir}/ -mindepth 1 -name '*.safetensors' -printf '%f\\n'")
     if any(not file.endswith('.txt') for file in os.listdir(vaes_dir)):
         print("\n\033[33m➤ VAEs\033[0m")
-        get_ipython().system("find {vaes_dir}/ -mindepth 1 ! -name '*.txt' -printf '%f\\n'")
+        get_ipython().system("find {vaes_dir}/ -mindepth 1 -name '*.safetensors' -printf '%f\\n'")
     if any(not file.endswith('.txt') and not os.path.isdir(os.path.join(embeddings_dir, file)) for file in os.listdir(embeddings_dir)):
         print("\n\033[33m➤ Embeddings\033[0m")
         get_ipython().system("find {embeddings_dir}/ -mindepth 1 -maxdepth 1 \\( -name '*.pt' -or -name '*.safetensors' \\) -printf '%f\\n'")
     if any(not file.endswith('.txt') for file in os.listdir(loras_dir)):
         print("\n\033[33m➤ LoRAs\033[0m")
-        get_ipython().system("find {loras_dir}/ -mindepth 1 ! -name '*.keep' -printf '%f\\n'")
+        get_ipython().system("find {loras_dir}/ -mindepth 1 -name '*.safetensors' -printf '%f\\n'")
     print(f"\n\033[33m➤ Extensions\033[0m")
     get_ipython().system("find {extensions_dir}/ -mindepth 1 -maxdepth 1 ! -name '*.txt' -printf '%f\\n'")
     if any(not file.endswith(('.txt', '.yaml')) for file in os.listdir(control_dir)):
         print("\n\033[33m➤ ControlNet\033[0m")
-        get_ipython().system("find {control_dir}/ -mindepth 1 ! -name '*.yaml' -printf '%f\\n' | sed 's/^[^_]*_[^_]*_[^_]*_\\(.*\\)_fp16\\.safetensors$/\\1/'")
+        get_ipython().system("find {control_dir}/ -mindepth 1 ! -name '*.yaml' -printf '%f\\n' | sed 's/^[^_]*_[^_]*_[^_]*_\\([^_]*\\)_fp16\\.safetensors$/\\1/'")
 
 
 # === OTHER ===