Update app.py

app.py

@@ -91,6 +91,10 @@ MIME_MAP = {
     "avif": "image/avif",
     "heif": "image/heic",
     "tiff": "image/tiff",
+    "gif":  "image/gif",
+    "bmp":  "image/bmp",
+    "ico":  "image/x-icon",
+    "jp2":  "image/jp2",
 }
 
 # ---------------------------------------------------------------------------
@@ -108,12 +112,20 @@ def _compress_image(raw_bytes: bytes, quality: int, fmt: str) -> bytes:
     img = Image.open(io.BytesIO(raw_bytes))
 
     # Convert RGBA → RGB for formats that don't support alpha
-    if fmt in ("jpeg", "tiff") and img.mode in ("RGBA", "LA", "PA"):
+    if fmt in ("jpeg", "tiff", "bmp", "jp2") and img.mode in ("RGBA", "LA", "PA"):
         background = Image.new("RGB", img.size, (255, 255, 255))
         background.paste(img, mask=img.split()[-1])  # alpha composite
         img = background
-    elif fmt in ("jpeg",) and img.mode != "RGB":
+    elif fmt in ("jpeg", "bmp", "jp2") and img.mode not in ("RGB", "L"):
         img = img.convert("RGB")
+    elif fmt == "gif" and img.mode == "RGBA":
+        # GIF supports palette transparency; convert RGBA → P with transparency
+        img = img.convert("RGBA")  # ensure consistent mode
+        alpha = img.split()[-1]
+        img = img.convert("RGB").convert("P", palette=Image.ADAPTIVE, colors=255)
+        mask = Image.eval(alpha, lambda a: 255 if a <= 128 else 0)
+        img.paste(255, mask)  # map transparent pixels to index 255
+        # transparency index will be set during save
 
     out_buffer = io.BytesIO()
 
@@ -164,6 +176,63 @@ def _compress_image(raw_bytes: bytes, quality: int, fmt: str) -> bytes:
             format="TIFF",
             compression="tiff_lzw",  # lossless LZW compression
         )
+    elif fmt == "gif":
+        # Preserve animated GIF frames if present
+        n_frames = getattr(img, 'n_frames', 1)
+        if n_frames > 1:
+            frames = []
+            durations = []
+            for i in range(n_frames):
+                img.seek(i)
+                frame = img.copy()
+                if frame.mode != "P":
+                    frame = frame.convert("RGBA").convert("P", palette=Image.ADAPTIVE, colors=256)
+                frames.append(frame)
+                durations.append(img.info.get('duration', 100))
+            frames[0].save(
+                out_buffer,
+                format="GIF",
+                save_all=True,
+                append_images=frames[1:],
+                duration=durations,
+                loop=img.info.get('loop', 0),
+                optimize=True,
+            )
+        else:
+            if img.mode not in ("P", "L"):
+                img = img.convert("P", palette=Image.ADAPTIVE, colors=256)
+            img.save(
+                out_buffer,
+                format="GIF",
+                optimize=True,
+            )
+    elif fmt == "bmp":
+        if img.mode not in ("RGB", "L", "1"):
+            img = img.convert("RGB")
+        img.save(
+            out_buffer,
+            format="BMP",
+        )
+    elif fmt == "ico":
+        # ICO spec limits dimensions to 256×256
+        max_ico = 256
+        if img.width > max_ico or img.height > max_ico:
+            img.thumbnail((max_ico, max_ico), Image.LANCZOS)
+        if img.mode != "RGBA":
+            img = img.convert("RGBA")
+        img.save(
+            out_buffer,
+            format="ICO",
+        )
+    elif fmt == "jp2":
+        if img.mode not in ("RGB", "L", "RGBA"):
+            img = img.convert("RGB")
+        img.save(
+            out_buffer,
+            format="JPEG2000",
+            quality_mode="rates",
+            quality_layers=[quality],
+        )
     else:
         raise ValueError(f"Unsupported format: {fmt}")
 
@@ -190,7 +259,7 @@ async def root():
 async def process_stream(
     request: Request,
     quality: int = Query(default=80, ge=1, le=100),
-    format: str = Query(default="jpeg", pattern="^(jpeg|png|webp|avif|heif|tiff)$"),
+    format: str = Query(default="jpeg", pattern="^(jpeg|png|webp|avif|heif|tiff|gif|bmp|ico|jp2)$"),
 ):
     """
     Receive a raw image in the request body (Content-Type: image/*),