resize image before upload

this significantly speeds up the space when using e.g. mobile photos

src/app.py

@@ -19,6 +19,14 @@ with env.prefixed("ERASER_"):
 auth = None if API_KEY is None else httpx.BasicAuth("hf", API_KEY)
 
 
+def resize(image: Image.Image, shortest_side: int = 768) -> Image.Image:
+    if image.width <= shortest_side and image.height <= shortest_side:
+        return image
+    if image.width < image.height:
+        return image.resize(size=(shortest_side, int(shortest_side * image.height / image.width)))
+    return image.resize(size=(int(shortest_side * image.width / image.height), shortest_side))
+
+
 def process_bbox(
     prompts: dict[str, Any],
     request: gr.Request | None,
@@ -27,16 +35,21 @@ def process_bbox(
     assert isinstance(boxes := prompts["boxes"], list)
     assert len(boxes) == 1
     assert isinstance(box := boxes[0], dict)
-    data = {"bbox": ",".join([str(box[k]) for k in ["xmin", "ymin", "xmax", "ymax"]])}
     headers = {}
     if request:  # avoid DOS - can be None despite type hint!
         client_ip = request.headers.get("x-forwarded-for") or request.client.host
         headers = {"X-HF-Client-IP": client_ip}
+
+    resized_img = resize(img)
+    bbox = [box[k] for k in ["xmin", "ymin", "xmax", "ymax"]]
+    if resized_img.width != img.width:
+        bbox = [int(v * resized_img.width / img.width) for v in bbox]
+
     with io.BytesIO() as f:
-        img.save(f, format="JPEG")
+        resized_img.save(f, format="JPEG")
         r = httpx.post(
             API_URL,
-            data=data,
+            data={"bbox": ",".join([str(v) for v in bbox])},
             files={"file": f},
             verify=CA_BUNDLE or True,
             timeout=30.0,
@@ -58,16 +71,18 @@ def process_prompt(
     prompt: str,
     request: gr.Request | None,
 ) -> tuple[Image.Image, Image.Image]:
-    data = {"prompt": prompt}
     headers = {}
     if request:  # avoid DOS - can be None despite type hint!
         client_ip = request.headers.get("x-forwarded-for") or request.client.host
         headers = {"X-HF-Client-IP": client_ip}
+
+    resized_img = resize(img)
+
     with io.BytesIO() as f:
-        img.save(f, format="JPEG")
+        resized_img.save(f, format="JPEG")
         r = httpx.post(
             API_URL,
-            data=data,
+            data={"prompt": prompt},
             files={"file": f},
             verify=CA_BUNDLE or True,
             timeout=30.0,