Update app.py

app.py

@@ -345,15 +345,15 @@ def draw_mask(ksize, th, v, d, evt: gr.EventData):
     d["layers"][0] = cv2.fillPoly(d["layers"][0], [pts], (0,0,0,255))
     d["composite"] = cv2.fillPoly(d["composite"], [pts], (0,0,0,255))
 
-    #img = cv2.cvtColor(d["background"], cv2.COLOR_RGB2GRAY)
-    #rows, cols = img.shape
-    #crow, ccol = rows//2, cols//2
-    #f = np.fft.fft2(img)
-    #fshift = np.fft.fftshift(f)
-    #fshift[crow-int(ksize/2):crow+int(ksize/2+1), ccol-int(ksize/2):ccol+int(ksize/2+1)] = 0
-    #f_ishift = np.fft.ifftshift(fshift)
-    #img_back = np.fft.ifft2(f_ishift)
-    #img_back = np.real(img_back).astype(np.uint8)
+    img = cv2.cvtColor(d["background"], cv2.COLOR_RGB2GRAY)
+    rows, cols = img.shape
+    crow, ccol = rows//2, cols//2
+    f = np.fft.fft2(img)
+    fshift = np.fft.fftshift(f)
+    fshift[crow-int(ksize/2):crow+int(ksize/2+1), ccol-int(ksize/2):ccol+int(ksize/2+1)] = 0
+    f_ishift = np.fft.ifftshift(fshift)
+    img_back = np.fft.ifft2(f_ishift)
+    img_back = np.real(img_back).astype(np.uint8)
     
 
     #scale = 1
@@ -370,10 +370,10 @@ def draw_mask(ksize, th, v, d, evt: gr.EventData):
 
     #grad = cv2.addWeighted(abs_grad_x, 0.5, abs_grad_y, 0.5, 0)
     #bw = cv2.cvtColor(cv2.threshold(255-img_back, th, 255, cv2.THRESH_BINARY)[1], cv2.COLOR_GRAY2RGB)
-    #bw = cv2.cvtColor(255-img_back, cv2.COLOR_GRAY2RGB)
+    bw = cv2.cvtColor(255-img_back, cv2.COLOR_GRAY2RGB)
 
-    #d["background"] = bw
-    #d["composite"] = bw
+    d["background"] = bw
+    d["composite"] = bw
     print(d["composite"])
     
     return gr.ImageEditor(value=d)
@@ -430,7 +430,7 @@ with gr.Blocks(css=css) as demo:
             output_mask = gr.ImageEditor(layers=True, sources=None, type="numpy", interactive=True, transforms=(None,), eraser=gr.Eraser(), brush=gr.Brush(default_size=0, colors=['black', '#505050', '#a0a0a0', 'white']), elem_id="image_edit")
             ksize = gr.Slider(label="Kernel", value=128, minimum=0, maximum=1024, step=1)
             threshold = gr.Slider(label="Threshold", value=128, minimum=0, maximum=255, step=1)
-            selector = gr.HTML(value="""<div id='select_cursor' style='width:1px;height:1px;position:absolute;border:1px dotted #ffffff;'></div>
+            selector = gr.HTML(value="""<div id='select_cursor' style='width:1px;height:1px;position:absolute;border:3px dotted #ffffff;'></div>
             <a href='#' id='selector' onclick='this.style.fontWeight=\"bold\";
             document.body.appendChild(document.getElementById(\"select_cursor\"));
             document.getElementById(\"image_edit\").getElementsByTagName(\"canvas\")[0].oncontextmenu = function(e){e.preventDefault();}
@@ -466,12 +466,15 @@ with gr.Blocks(css=css) as demo:
                 }
                 document.getElementById(\"mouse\").getElementsByTagName(\"textarea\")[0].value = \"[\" + p + \"]\";
 
+                var evt = document.createEvent(\"Event\");
+                evt.initEvent(\"input\", true, false);
+                document.getElementById(\"mouse\").getElementsByTagName(\"textarea\")[0].dispatchEvent(evt);
+
                 document.getElementById(\"select_cursor\").style.left = e.clientX-1 + \"px\";
                 document.getElementById(\"select_cursor\").style.top = e.clientY-1 + \"px\";
               }
             }
-            document.getElementById(\"image_edit\").getElementsByTagName(\"canvas\")[0].ondblclick = function(e) {
-              e.preventDefault();
+            document.getElementById(\"image_edit\").getElementsByTagName(\"canvas\")[0].onblur = function(e) {
               document.getElementById(\"mouse\").getElementsByTagName(\"textarea\")[0].value = \"[]\";
 
               document.getElementById(\"select_cursor\").style.left = 0;