use webp images

invokeai/app/api/routers/model_manager.py

@@ -287,21 +287,18 @@ async def update_model_record(
             "description": "The model image was fetched successfully",
         },
         400: {"description": "Bad request"},
-        404: {"description": "The model could not be found"},
+        404: {"description": "The model image could not be found"}
     },
     status_code=200,
 )
 async def get_model_image(
     key: str = Path(description="The name of model image file to get"),
 ) -> FileResponse:
-    """Gets a full-resolution image file"""
+    """Gets an image file that previews the model"""
 
     try:
         path = ApiDependencies.invoker.services.model_images.get_path(key)
 
-        if not path:
-            raise HTTPException(status_code=404)
-
         response = FileResponse(
             path,
             media_type="image/png",

invokeai/app/services/model_images/model_images_base.py

@@ -12,13 +12,13 @@ class ModelImagesBase(ABC):
         pass
 
     @abstractmethod
-    def get_path(self, model_key: str) -> Path | None:
+    def get_path(self, model_key: str) -> Path:
         """Gets the internal path to a model image."""
         pass
 
     @abstractmethod
     def get_url(self, model_key: str) -> str | None:
-        """Gets the URL to a model image."""
+        """Gets the URL to fetch a model image."""
         pass
 
     @abstractmethod
@@ -27,7 +27,7 @@ class ModelImagesBase(ABC):
         image: PILImageType,
         model_key: str,
     ) -> None:
-        """Saves a model image. Returns a tuple of the image name and created timestamp."""
+        """Saves a model image."""
         pass
 
     @abstractmethod

invokeai/app/services/model_images/model_images_default.py

@@ -1,7 +1,7 @@
 from pathlib import Path
 from typing import Union
 
-from PIL import Image, PngImagePlugin
+from PIL import Image
 from PIL.Image import Image as PILImageType
 from send2trash import send2trash
 
@@ -45,25 +45,16 @@ class ModelImagesService(ModelImagesBase):
     ) -> None:
         try:
             self.__validate_storage_folders()
-            logger = self.__invoker.services.logger
-            image_path = self.__model_images_folder / (model_key + '.png')
-            logger.debug(f"Saving image for model {model_key} to image_path {image_path}")
-            
-            pnginfo = PngImagePlugin.PngInfo()  
+            image_path = self.__model_images_folder / (model_key + '.webp')
             image = make_thumbnail(image, 256)
 
-            image.save(
-                image_path,
-                "PNG",
-                pnginfo=pnginfo,
-                compress_level=self.__invoker.services.configuration.png_compress_level,
-            )
+            image.save(image_path, format="webp")
 
         except Exception as e:
             raise ModelImageFileSaveException from e
 
     def get_path(self, model_key: str) -> Path:
-        path = self.__model_images_folder / (model_key + '.png')
+        path = self.__model_images_folder / (model_key + '.webp')
 
         return path
     

invokeai/backend/model_manager/config.py

@@ -20,7 +20,6 @@ Validation errors will raise an InvalidModelConfigException error.
 
 """
 
-from pathlib import Path
 import time
 from enum import Enum
 from typing import Literal, Optional, Type, Union

invokeai/frontend/package-lock.json

@@ -1,6 +0,0 @@
-{
-  "name": "frontend",
-  "lockfileVersion": 2,
-  "requires": true,
-  "packages": {}
-}

invokeai/frontend/web/src/features/modelManagerV2/subpanels/ModelManagerPanel/ModelImage.tsx

@@ -9,7 +9,7 @@ type Props = {
 
 const ModelImage = ({ image_url }: Props) => {
 
-  if (!image_url) return <Box height="50px" width="50px" />;
+  if (!image_url) return <Box height="50px" minWidth="50px" />;
 
     return (
       <Image
@@ -18,6 +18,8 @@ const ModelImage = ({ image_url }: Props) => {
         objectPosition="50% 50%"
         height="50px"
         width="50px"
+        minHeight="50px"
+        minWidth="50px"
         borderRadius="base"
       />
     );

invokeai/frontend/web/src/features/modelManagerV2/subpanels/ModelPanel/Fields/ModelImageUpload.tsx

@@ -64,7 +64,7 @@ type Props = {
     if (!model_key) {
       return;
     }
-    setImage(undefined);
+    setImage(null);
     deleteModelImage(model_key)
     .unwrap()
     .then(() => {
@@ -102,11 +102,11 @@ type Props = {
         position="relative"
       >
         <Image
-          onError={() => setImage(undefined)}
           src={image}
-          objectFit="contain"
-          maxW="full"
-          maxH="100px"
+          objectFit="cover"
+          objectPosition="50% 50%"
+          height="100px"
+          width="100px"
           borderRadius="base"
         />
         <IconButton