api...?

app.py

@@ -42,24 +42,13 @@ with gr.Blocks() as demo:
  # ===================================================== #
  # parameter name is also shown as parameter name in API #
  # ===================================================== #
- def proc_image(
- image: dict | str, width: int, height: int
- ) -> list[str, Image.Image]:
+ def proc_image(image: str, width: int, height: int) -> list[str, Image.Image]:
+ assert isinstance(image, str)
 
- if isinstance(image, dict):
- image = Image.open(image["path"])
-
- elif isinstance(image, str):
- if os.path.isfile(image):
- image = Image.open(image)
- else:
- # ==================================== #
- # This works with base64 image via API #
- # ==================================== #
- image = Image.open(BytesIO(base64.b64decode(image)))
-
- else:
- raise ValueError("Unsupported Image Type")
+ # ==================================== #
+ # This works with base64 image via API #
+ # ==================================== #
+ image = Image.open(BytesIO(base64.b64decode(image)))
 
  if isinstance(width, float):
  width = int(width)
@@ -72,7 +61,11 @@ with gr.Blocks() as demo:
 
  image = image.convert("RGB").resize((width, height))
 
- return [log, image]
+ with BytesIO() as buffer:
+ image.save(buffer, format="PNG")
+ b64_image = base64.b64encode(buffer.getvalue()).decode()
+
+ return [log, b64_image]
 
  with gr.Row():
 
@@ -81,18 +74,15 @@ with gr.Blocks() as demo:
  # ==================================== #
  # variable name does not matter in API #
  # ==================================== #
- foo1 = gr.Image(
- value=None,
- image_mode="RGB",
- sources="upload",
- type="filepath",
- label="Input",
+ foo1 = gr.Text(
+ value="",
+ label="Input Image",
  show_label=True,
- show_download_button=False,
+ info="Image in base64",
  visible=True,
  interactive=True,
- show_share_button=False,
- show_fullscreen_button=False,
+ lines=1,
+ max_lines=1,
  )
 
  # =============================================== #
@@ -119,18 +109,15 @@ with gr.Blocks() as demo:
  interactive=True,
  )
 
- bar = gr.Image(
- value=None,
- image_mode="RGB",
- sources="upload",
- type="pil",
- label="Output",
+ bar = gr.Text(
+ value="",
+ label="Output Image",
  show_label=True,
- show_download_button=False,
+ info="Image in base64",
  visible=True,
  interactive=False,
- show_share_button=False,
- show_fullscreen_button=False,
+ lines=1,
+ max_lines=1,
  )
 
  yeet1 = gr.Button("Resize")
@@ -156,7 +143,6 @@ with gr.Blocks() as demo:
 
  # =================================== #
  # api_name is used as endpoint in API #
- # order is based on event defination #
  # =================================== #
  yeet2.click(
  fn=whatever,
@@ -167,10 +153,9 @@ 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 #
- # ======================================================== #
+ # =============================================== #
+ # API order is based on event declaration in code #
+ # =============================================== #
  yeet1.click(
  fn=proc_image,
  inputs=[foo1, foo2, foo3],
@@ -178,8 +163,6 @@ with gr.Blocks() as demo:
  show_progress="hidden",
  show_api=True,
  api_name="resizeImage",
- preprocess=False,
- postprocess=False,
  )
 
 demo.launch(show_error=True)