Added compress function

mcp_server.py

@@ -6,6 +6,7 @@ from src.utils.generate_image import generate_image
 from src.utils.apply_filter import apply_filter
 from src.utils.add_text import add_text_to_image
 from src.utils.describe import describe_image
+from src.utils.compress import compress_image
 
 mcp = FastMCP("Youtube Service")
 
@@ -29,6 +30,7 @@ mcp.add_tool(visualize_base64_image)
 mcp.add_tool(generate_image)
 mcp.add_tool(apply_filter)
 mcp.add_tool(add_text_to_image)
+mcp.add_tool(compress_image)
 
 if __name__ == "__main__":
     mcp.run()

src/utils/compress.py

@@ -0,0 +1,49 @@
+from PIL import Image
+import os
+from typing import Literal, Optional
+
+def compress_image(
+    input_path: str,
+    output_path: str,
+    quality: int = 85,
+    format: Literal["JPEG", "PNG", "WEBP"] = "JPEG",
+    max_width: Optional[int] = None,
+    max_height: Optional[int] = None
+) -> str:
+    """
+    Compress an image file.
+    
+    Args:
+        input_path: Path to input image
+        output_path: Path for compressed output
+        quality: Compression quality 1-95 (for JPEG/WEBP)
+        format: Output format
+        max_width: Maximum width (optional)
+        max_height: Maximum height (optional)
+    """
+    try:
+        if not os.path.splitext(output_path)[1]:
+            extension_map = {"JPEG": ".jpg", "PNG": ".png", "WEBP": ".webp"}
+            output_path = output_path + extension_map[format]
+        
+        with Image.open(input_path) as img:
+            if format == "JPEG" and img.mode in ("RGBA", "P"):
+                img = img.convert("RGB")
+            
+            if max_width or max_height:
+                img.thumbnail((max_width or img.width, max_height or img.height), Image.Resampling.LANCZOS)
+            
+            save_kwargs = {"format": format, "optimize": True}
+            if format in ["JPEG", "WEBP"]:
+                save_kwargs["quality"] = quality
+            
+            img.save(output_path, **save_kwargs)
+
+        original_size = os.path.getsize(input_path) / 1024 / 1024  
+        compressed_size = os.path.getsize(output_path) / 1024 / 1024 
+        reduction = (1 - compressed_size/original_size) * 100
+        
+        return f"✅ Compressed successfully!\nOriginal: {original_size:.2f}MB → Compressed: {compressed_size:.2f}MB\nReduction: {reduction:.1f}%"
+    
+    except Exception as e:
+        return f"❌ Error: {str(e)}"