diff --git a/__init__.py b/__init__.py index 0602c69..47513aa 100644 --- a/__init__.py +++ b/__init__.py @@ -293,6 +293,8 @@ def get_image_info(image): def image_format_is_equal(f1, f2): if not isinstance(f1, str) or not isinstance(f2, str): return False + if f1[0] == ".": f1 = f1[1:] + if f2[0] == ".": f2 = f2[1:] f1 = f1.upper() f2 = f2.upper() return f1 == f2 or (f1 == "JPG" and f2 == "JPEG") or (f1 == "JPEG" and f2 == "JPG") @@ -484,9 +486,22 @@ def download_model_preview(formdata): image_data = file.read() with open(image_path, "wb") as f: f.write(image_data) + print("Saved file: " + image_path) delete_same_name_files(path_without_extension, preview_extensions, image_extension) + # detect and fix wrong file extension + image_format = None + with Image.open(image_path) as image: + image_format = image.format + image_dir_and_name, image_ext = os.path.splitext(image_path) + if not image_format_is_equal(image_format, image_ext): + corrected_image_path = image_dir_and_name + "." + image_format.lower() + os.rename(image_path, corrected_image_path) + print("Saved file: " + corrected_image_path) + image_path = corrected_image_path + return image_path # return in-case need corrected path + @server.PromptServer.instance.routes.post("/model-manager/preview/set") async def set_model_preview(request):