pain

app.py

@@ -39,15 +39,17 @@ with gr.Blocks() as demo:
 
         return "\n".join(info)
 
-    # ==================================================================== #
-    # parameter name is also shown as parameter name in API                #
-    # Note: parameter type has to be manually handled when called from API #
-    # ==================================================================== #
-    def proc_image(image: str, width: int, height: int) -> list[str, Image.Image]:
+    # ===================================================== #
+    # parameter name is also shown as parameter name in API #
+    # ===================================================== #
+    def proc_image(
+        image: dict | str, width: int, height: int
+    ) -> list[str, Image.Image]:
 
-        print(type(image), image)
+        if isinstance(image, dict):
+            image = Image.open(image["path"])
 
-        if isinstance(image, str):
+        elif isinstance(image, str):
             if os.path.isfile(image):
                 image = Image.open(image)
             else:
@@ -55,6 +57,7 @@ with gr.Blocks() as demo:
                 # This works with base64 image via API #
                 # ==================================== #
                 image = Image.open(BytesIO(base64.b64decode(image)))
+
         else:
             raise ValueError("Unsupported Image Type")
 
@@ -164,6 +167,10 @@ with gr.Blocks() as demo:
         api_name="systemInfo",
     )
 
+    # ======================================================== #
+    # disable preprocess to allow arbitrary type from API      #
+    # Note: parameter type has to be manually handled this way #
+    # ======================================================== #
     yeet1.click(
         fn=proc_image,
         inputs=[foo1, foo2, foo3],
@@ -171,6 +178,8 @@ with gr.Blocks() as demo:
         show_progress="hidden",
         show_api=True,
         api_name="resizeImage",
+        preprocess=False,
+        postprocess=False,
     )
 
 demo.launch(show_error=True)