Update app.py

app.py

@@ -180,10 +180,15 @@ ocr_id = {
 }
 
 
-def blur_im(img,bounds,target_lang,trans_lang,ocr_sens,font_fac):
+def blur_im(img,bounds,target_lang,trans_lang,ocr_sens,font_fac,t_color):
     im = cv2.imread(img)
     im = cv2.cvtColor(im, cv2.COLOR_BGR2RGB)
-   
+    if t_color == "Black":
+        t_fill = (0,0,0)
+        pass
+    elif t_color == "White":
+        t_fill = (255,255,255)
+        pass
     for bound in bounds:
         if bound[2]>=(ocr_sens):
             p0, p1, p2, p3 = bound[0]
@@ -192,7 +197,12 @@ def blur_im(img,bounds,target_lang,trans_lang,ocr_sens,font_fac):
             w = int(p2[0]) - int(x)
             h = int(p2[1]) - int(y)
             kernel = np.ones((3, 3), np.uint8)
-            im[y:y+h, x:x+w] = cv2.dilate(im[y:y+h, x:x+w], kernel, iterations=3)
+            if t_color=="Black":
+                im[y:y+h, x:x+w] = cv2.dilate(im[y:y+h, x:x+w], kernel, iterations=3)
+                pass
+            elif t_color=="White":
+                im[y:y+h, x:x+w] = cv2.erode(im[y:y+h, x:x+w], kernel, iterations=3)
+                pass
             im[y:y+h, x:x+w] = cv2.GaussianBlur(im[y:y+h, x:x+w],(51,51),0)
         else:
             pass
@@ -208,7 +218,7 @@ def blur_im(img,bounds,target_lang,trans_lang,ocr_sens,font_fac):
             text = this(bound[1],target_lang,trans_lang)
             font_size=int(int(h)*font_fac)
             font = ImageFont.truetype("./fonts/unifont-15.0.01.ttf", int(font_size))
-            draw.text((x, y),text, font = font, fill=(0,0,0))
+            draw.text((x, y),text, font = font, fill=t_fill)
         else:
             pass
     return im
@@ -274,7 +284,9 @@ with gr.Blocks() as robot:
                     with gr.Column():
                         ocr_sens=gr.Slider(0.1, 1, step=0.05,value=0.25,label="Detect Min Confidence")
                         font_fac=gr.Slider(0.1, 1, step =0.1,value=0.4,label="Font Scale")
-                        ocr_space=gr.Slider(1,10, step=1,value=5,label="Future Function")            
+                        ocr_space=gr.Slider(1,10, step=1,value=5,label="Future Function")     
+                        text_color=gr.Radio(["Black", "White"])
+
             go_btn=gr.Button("Go")
     with gr.Row():
         with gr.Column():
@@ -286,6 +298,6 @@ with gr.Blocks() as robot:
         out_txt=gr.Textbox(lines=8)
         data_f=gr.Dataframe()
                         
-    go_btn.click(detect,[im,target_lang,trans_lang,ocr_sens,font_fac],[out_im,trans_im,out_txt,data_f])
+    go_btn.click(detect,[im,target_lang,trans_lang,ocr_sens,font_fac,text_color],[out_im,trans_im,out_txt,data_f])
     #go_btn.click(detect,[im,target_lang,target_lang2],[out_im,trans_im,out_txt,data_f])
 robot.queue(concurrency_count=10).launch()