From 2806c0b9f75f7dd966887a96bb1de13020b26a9f Mon Sep 17 00:00:00 2001
From: Christian Kento Rasmussen <christian.kento@gmail.com>
Date: Fri, 12 Apr 2024 11:20:58 +0200
Subject: [PATCH] Fix drawing issue in gr.ImageEditor

---
 qim3d/gui/annotation_tool.py | 14 +++++++++-----
 1 file changed, 9 insertions(+), 5 deletions(-)

diff --git a/qim3d/gui/annotation_tool.py b/qim3d/gui/annotation_tool.py
index 591f96d0..95807399 100644
--- a/qim3d/gui/annotation_tool.py
+++ b/qim3d/gui/annotation_tool.py
@@ -15,13 +15,16 @@ app.launch()
 ```
 """
 
-import tifffile
-import tempfile
+import getpass
 import os
+import tempfile
 import time
-import getpass
-import numpy as np
+
 import gradio as gr
+import numpy as np
+import tifffile
+from PIL import Image
+
 import qim3d.utils
 from qim3d.io import load, save
 from qim3d.io.logger import log
@@ -112,7 +115,8 @@ class Interface:
 
                 with gr.Column(scale=6):
                     img_editor = gr.ImageEditor(
-                        value=img,
+                        # ! Temporary fix for drawing at wrong location https://github.com/gradio-app/gradio/pull/7959
+                        value={"background": img, "layers": [Image.new("RGBA", img.shape, (0, 0, 0, 0))], "composite": None} if img is not None else None,
                         type="numpy",
                         image_mode="RGB",
                         brush=brush,
-- 
GitLab