Upload processor/zipper.py with huggingface_hub

processor/zipper.py

@@ -0,0 +1,45 @@
+import cv2
+import zipfile
+import io
+import numpy as np
+from typing import List
+
+def standardize_and_zip(images: List[tuple[str, np.ndarray]]) -> io.BytesIO:
+    """
+    Takes a list of tuples (filename, image). Resizes images to standard size
+    (e.g., 600x800), encodes them as PNG/JPG, and zips them.
+    Returns an in-memory ZIP file stream.
+    """
+    TARGET_WIDTH = 600
+    TARGET_HEIGHT = 800
+    
+    zip_buffer = io.BytesIO()
+    
+    with zipfile.ZipFile(zip_buffer, "a", zipfile.ZIP_DEFLATED, False) as zip_file:
+        for idx, (filename, img) in enumerate(images):
+            h, w = img.shape[:2]
+            
+            # Determine orientation to preserve aspect ratio
+            if w > h:
+                # Landscape
+                new_w = 800
+                new_h = int(h * (800 / w))
+            else:
+                # Portrait
+                new_h = 800
+                new_w = int(w * (800 / h))
+                
+            resized = cv2.resize(img, (new_w, new_h), interpolation=cv2.INTER_CUBIC)
+            
+            # Encode as PNG to preserve transparent background
+            success, buffer = cv2.imencode(".png", resized)
+            if success:
+                # Keep original name or rename
+                name_without_ext = filename.rsplit(".", 1)[0]
+                new_filename = f"{name_without_ext}_processed.png"
+                
+                zip_file.writestr(new_filename, buffer.tobytes())
+
+    # Rewind buffer
+    zip_buffer.seek(0)
+    return zip_buffer