add image editor

app.py

@@ -40,10 +40,9 @@ 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
-
+    
     unique_id = str(uuid.uuid4())
     image_path = os.path.join(UPLOAD_FOLDER, f"{unique_id}.png")
     result_path = os.path.join(RESULTS_FOLDER, f"{unique_id}.html")
@@ -127,11 +126,10 @@ def update_inputs(task):
 def ocr_demo(image, task, ocr_type, ocr_box, ocr_color):
     res, html_content, unique_id = process_image(image, task, ocr_type, ocr_box, ocr_color)
     
-    if res.startswith("Error:"):
+    if isinstance(res, str) and res.startswith("Error:"):
         return res, None
 
     res = res.replace("\\title", "\\title ")
-
     res = f"$$ {res} $$"
     
     if html_content: