Hotfix for https://huggingface.co/spaces/finegrain/finegrain-object-eraser/discussions/9 (#10)

- Hotfix for https://huggingface.co/spaces/finegrain/finegrain-object-eraser/discussions/9 (110daf2d24d404ae001745ce96319ea34b92d16e)

src/app.py

@@ -88,6 +88,15 @@ async def _process(ctx: EditorAPIContext, params: ProcessParams) -> Image.Image:
     f.seek(0)
     return Image.open(f)
 
+async def _run(params: ProcessParams) -> Image.Image:
+    # hotfix for https://huggingface.co/spaces/finegrain/finegrain-object-eraser/discussions/9
+    # happens because _sse_task is not None in sse_start
+    ctx = _ctx()
+    try:
+        return ctx.run_one_sync(_process, params)
+    except AssertionError:
+        ctx.reset()
+        return ctx.run_one_sync(_process, params)
 
 def process_bbox(prompts: dict[str, Any]) -> tuple[Image.Image, Image.Image]:
     assert isinstance(img := prompts["image"], Image.Image)
@@ -100,8 +109,7 @@ def process_bbox(prompts: dict[str, Any]) -> tuple[Image.Image, Image.Image]:
     if resized_img.width != img.width:
         bbox = [int(v * resized_img.width / img.width) for v in bbox]
 
-    output_image = _ctx().run_one_sync(
-        _process,
+    output_image = _run(
         ProcessParams(
             image=resized_img,
             bbox=(bbox[0], bbox[1], bbox[2], bbox[3]),
@@ -116,8 +124,7 @@ def on_change_bbox(prompts: dict[str, Any] | None):
 
 def process_prompt(img: Image.Image, prompt: str) -> tuple[Image.Image, Image.Image]:
     resized_img = resize(img)
-    output_image = _ctx().run_one_sync(
-        _process,
+    output_image = _run(
         ProcessParams(image=resized_img, prompt=prompt),
     )
     return (img, output_image)