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)