fix imports, add missing controlnet deps

api/onnx_web/__init__.py

@@ -40,7 +40,7 @@ from .image.source_filter import (
     source_filter_hed,
     source_filter_mlsd,
     source_filter_normal,
-    source_filter_pose,
+    source_filter_openpose,
     source_filter_scribble,
     source_filter_segment,
 )

api/onnx_web/diffusers/run.py

@@ -231,7 +231,9 @@ def run_img2img_pipeline(
 
     # filter the source image
     if source_filter is not None:
-        source = get_source_filters(source_filter)(source)
+        f = get_source_filters().get(source_filter, None)
+        if f is not None:
+            source = f(server, source)
 
     pipe = load_pipeline(
         server,

api/onnx_web/image/mask_filter.py

@@ -1,6 +1,6 @@
 from PIL import Image, ImageChops, ImageFilter
 
-from .params import Point
+from ..params import Point
 
 
 def mask_filter_none(

api/onnx_web/image/noise_source.py

@@ -2,7 +2,7 @@ import numpy as np
 from numpy import random
 from PIL import Image, ImageFilter
 
-from .params import Point
+from ..params import Point
 
 
 def get_pixel_index(x: int, y: int, width: int) -> int:

api/onnx_web/image/utils.py

@@ -4,7 +4,7 @@ from PIL import Image, ImageChops, ImageOps
 
 from .mask_filter import mask_filter_none
 from .noise_source import noise_source_histogram
-from .params import Border, Size
+from ..params import Border, Size
 
 
 # very loosely based on https://github.com/AUTOMATIC1111/stable-diffusion-webui/blob/master/scripts/outpainting_mk_2.py#L175-L232

api/requirements/base.txt

@@ -7,6 +7,7 @@ accelerate
 coloredlogs
 controlnet_aux
 diffusers
+mediapipe
 onnx
 # onnxruntime has many platform-specific packages
 safetensors