From e4d421c782a181f421e2987f75b4610845e3ef55 Mon Sep 17 00:00:00 2001
From: tanaka takashi <isazaka@gmail.com>
Date: Wed, 24 Jun 2026 22:47:13 +0900
Subject: Harden web UI security

---
 src/ollama_web/tools/image.py | 72 -------------------------------------------
 1 file changed, 72 deletions(-)
 delete mode 100644 src/ollama_web/tools/image.py

(limited to 'src/ollama_web/tools/image.py')

diff --git a/src/ollama_web/tools/image.py b/src/ollama_web/tools/image.py
deleted file mode 100644
index 8bc14f5..0000000
--- a/src/ollama_web/tools/image.py
+++ /dev/null
@@ -1,72 +0,0 @@
-"""Image preprocessing helpers for ollama vision models."""
-
-from __future__ import annotations
-
-import io
-from pathlib import Path
-
-from PIL import Image
-
-
-_IMAGE_FORMATS: dict[str, str] = {
-    "png": "PNG",
-    "jpg": "JPEG",
-    "jpeg": "JPEG",
-    "gif": "GIF",
-    "bmp": "BMP",
-    "webp": "WEBP",
-    "tif": "TIFF",
-    "tiff": "TIFF",
-}
-
-
-def _format_from_name(name: str) -> str:
-    ext = Path(name).suffix.lstrip(".").lower()
-    return _IMAGE_FORMATS.get(ext, "JPEG")
-
-
-def resize_image(
-    data: bytes,
-    name: str = "image.jpg",
-    max_dimension: int = 1024,
-    quality: int = 85,
-) -> bytes:
-    """Resize and re-encode an image to reduce payload size for ollama.
-
-    Args:
-      data: Raw image bytes.
-      name: Original file name, used to guess the output format.
-      max_dimension: Maximum width or height in pixels.
-      quality: JPEG/WebP quality when re-encoding to a lossy format.
-
-    Returns:
-      Resized image bytes. If the input is already small, it may be returned
-      unchanged when re-encoding is not needed.
-    """
-    if not data:
-        return data
-
-    try:
-        with Image.open(io.BytesIO(data)) as img:
-            # Preserve orientation if EXIF data exists.
-            img = img.convert("RGB")
-
-            width, height = img.size
-            if width > max_dimension or height > max_dimension:
-                ratio = min(max_dimension / width, max_dimension / height)
-                new_size = (int(width * ratio), int(height * ratio))
-                img = img.resize(new_size, Image.Resampling.LANCZOS)
-
-            fmt = _format_from_name(name)
-            output = io.BytesIO()
-            save_kwargs: dict[str, object] = {}
-            if fmt == "JPEG":
-                save_kwargs["quality"] = quality
-                save_kwargs["optimize"] = True
-            elif fmt == "WEBP":
-                save_kwargs["quality"] = quality
-            img.save(output, format=fmt, **save_kwargs)
-            return output.getvalue()
-    except Exception:  # noqa: BLE001
-        # Fall back to the original bytes if anything goes wrong.
-        return data
\ No newline at end of file
-- 
cgit v1.2.3