fix file handling

app.py

@@ -1,109 +1,87 @@
-# created with great guidance from https://github.com/NimaBoscarino
-
 import gradio as gr
-
 import kornia as K
 from kornia.core import Tensor
-
+import torch
+import numpy as np
 
 # Define Functions
-def box_blur_fn(file, box_blur): 
- # load the image using the rust backend 
- img: Tensor = K.io.load_image(file.name, K.io.ImageLoadType.RGB32)
- img = img[None] # 1xCxHxW / fp32 / [0, 1]
-
+def process_image(file):
+ if isinstance(file, np.ndarray):
+ # If the input is already a numpy array, convert it to a tensor
+ img = K.image_to_tensor(file).float() / 255.0
+ else:
+ # If it's a file path, load it using kornia
+ img = K.io.load_image(file, K.io.ImageLoadType.RGB32)
+ return img.unsqueeze(0) # Add batch dimension: 1xCxHxW
+
+def box_blur_fn(file, box_blur):
+ img = process_image(file)
  x_out: Tensor = K.filters.box_blur(img, (int(box_blur), int(box_blur)))
- 
- return K.utils.tensor_to_image(x_out)
-
-
-def blur_pool2d_fn(file, blur_pool2d): 
- # load the image using the rust backend 
- img: Tensor = K.io.load_image(file.name, K.io.ImageLoadType.RGB32)
- img = img[None] # 1xCxHxW / fp32 / [0, 1]
+ return K.utils.tensor_to_image(x_out.squeeze())
 
+def blur_pool2d_fn(file, blur_pool2d):
+ img = process_image(file)
  x_out: Tensor = K.filters.blur_pool2d(img, int(blur_pool2d))
- 
- return K.utils.tensor_to_image(x_out)
-
-
-def gaussian_blur_fn(file, gaussian_blur2d): 
- # load the image using the rust backend 
- img: Tensor = K.io.load_image(file.name, K.io.ImageLoadType.RGB32)
- img = img[None] # 1xCxHxW / fp32 / [0, 1]
+ return K.utils.tensor_to_image(x_out.squeeze())
 
- x_out: Tensor = K.filters.gaussian_blur2d(img, 
+def gaussian_blur_fn(file, gaussian_blur2d):
+ img = process_image(file)
+ x_out: Tensor = K.filters.gaussian_blur2d(img,
  (int(gaussian_blur2d), int(gaussian_blur2d)),
- (float(gaussian_blur2d), float(gaussian_blur2d)))
- 
- return K.utils.tensor_to_image(x_out)
-
+ (float(gaussian_blur2d)/2, float(gaussian_blur2d)/2))
+ return K.utils.tensor_to_image(x_out.squeeze())
 
 def max_blur_pool2d_fn(file, max_blur_pool2d):
- # load the image using the rust backend 
- img: Tensor = K.io.load_image(file.name, K.io.ImageLoadType.RGB32)
- img = img[None] # 1xCxHxW / fp32 / [0, 1]
-
+ img = process_image(file)
  x_out: Tensor = K.filters.max_blur_pool2d(img, int(max_blur_pool2d))
- 
- return K.utils.tensor_to_image(x_out)
-
-
-def median_blur_fn(file, median_blur): 
- # load the image using the rust backend 
- img: Tensor = K.io.load_image(file.name, K.io.ImageLoadType.RGB32)
- img = img[None] # 1xCxHxW / fp32 / [0, 1]
+ return K.utils.tensor_to_image(x_out.squeeze())
 
+def median_blur_fn(file, median_blur):
+ img = process_image(file)
  x_out: Tensor = K.filters.median_blur(img, (int(median_blur), int(median_blur)))
-
- return K.utils.tensor_to_image(x_out)
-
+ return K.utils.tensor_to_image(x_out.squeeze())
 
 # Define Examples
 examples = [
- ["examples/monkey.jpg", 1, 1, 1, 1, 1],
- ["examples/pikachu.jpg", 1, 1, 1, 1, 1]
+ ["examples/monkey.jpg", 1],
+ ["examples/pikachu.jpg", 1]
 ]
 
-
 # Define Demos
 box_blur_demo = gr.Interface(
  box_blur_fn,
  [
- gr.Image(type="filepath"),
+ gr.Image(type="numpy"),
  gr.Slider(minimum=1, maximum=20, step=1, value=10, label="Box Blur")
  ],
  "image",
  examples=examples,
 )
 
-
 blur_pool2d_demo = gr.Interface(
  blur_pool2d_fn,
  [
- gr.Image(type="file"),
+ gr.Image(type="numpy"),
  gr.Slider(minimum=1, maximum=40, step=1, value=20, label="Blur Pool")
  ],
  "image",
  examples=examples,
 )
 
-
 gaussian_blur_demo = gr.Interface(
  gaussian_blur_fn,
  [
- gr.Image(type="file"),
+ gr.Image(type="numpy"),
  gr.Slider(minimum=1, maximum=30, step=2, value=15, label="Gaussian Blur")
  ],
  "image",
  examples=examples,
 )
 
-
 max_blur_pool2d_demo = gr.Interface(
  max_blur_pool2d_fn,
  [
- gr.Image(type="file"),
+ gr.Image(type="numpy"),
  gr.Slider(minimum=1, maximum=40, step=1, value=20, label="Max Pool")
  ],
  "image",
@@ -113,29 +91,28 @@ max_blur_pool2d_demo = gr.Interface(
 median_blur_demo = gr.Interface(
  median_blur_fn,
  [
- gr.Image(type="file"),
+ gr.Image(type="numpy"),
  gr.Slider(minimum=1, maximum=30, step=2, value=15, label="Median Blur")
  ],
  "image",
  examples=examples,
 )
 
-
 # Create Interface
 demo = gr.TabbedInterface(
  [
- box_blur_demo,
- blur_pool2d_demo,
- gaussian_blur_demo,
- max_blur_pool2d_demo,
- median_blur_demo
- ], 
+ box_blur_demo,
+ blur_pool2d_demo,
+ gaussian_blur_demo,
+ max_blur_pool2d_demo,
+ median_blur_demo
+ ],
  [
- "Box Blur", 
- "Blur Pool",
- "Gaussian Blur",
- "Max Pool",
- "Median Blur"
+ "Box Blur",
+ "Blur Pool",
+ "Gaussian Blur",
+ "Max Pool",
+ "Median Blur"
  ]
 )