add image editor

app.py

@@ -40,6 +40,7 @@ def image_to_base64(image):
 
 @spaces.GPU()
 def process_image(image, task, ocr_type=None, ocr_box=None, ocr_color=None):
+ def process_image(image, task, ocr_type=None, ocr_box=None, ocr_color=None):
     if image is None:
         return "Error: No image provided", None, None
 
@@ -53,6 +54,8 @@ def process_image(image, task, ocr_type=None, ocr_box=None, ocr_color=None):
             if composite_image is not None:
                 if isinstance(composite_image, np.ndarray):
                     cv2.imwrite(image_path, cv2.cvtColor(composite_image, cv2.COLOR_RGB2BGR))
+                elif isinstance(composite_image, Image.Image):
+                    composite_image.save(image_path)
                 else:
                     return "Error: Unsupported image format from ImageEditor", None, None
             else: