Update app.py

app.py

@@ -183,15 +183,17 @@ def blur_im(img,bounds,target_lang,trans_lang):
     im = cv2.imread(img)
     im = cv2.cvtColor(im, cv2.COLOR_BGR2RGB)
     for bound in bounds:
-        
-        p0, p1, p2, p3 = bound[0]
-        x = int(p0[0])
-        y = int(p0[1])
-        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)
-        im[y:y+h, x:x+w] = cv2.GaussianBlur(im[y:y+h, x:x+w],(51,51),0)
+        if bound[2]>=0.3:
+            p0, p1, p2, p3 = bound[0]
+            x = int(p0[0])
+            y = int(p0[1])
+            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)
+            im[y:y+h, x:x+w] = cv2.GaussianBlur(im[y:y+h, x:x+w],(51,51),0)
+        else:
+            pass
     im = Image.fromarray(im)
     for bound in bounds:
         if bound[2]>=0.3:
@@ -201,7 +203,7 @@ def blur_im(img,bounds,target_lang,trans_lang):
             w = int(p2[0]) - int(x)
             h = int(p2[1]) - int(y)
             draw = ImageDraw.Draw(im)
-            font_size=int(int(w)*0.15)
+            font_size=int(int(w)*0.1)
             font = ImageFont.truetype("./fonts/unifont-15.0.01.ttf", int(font_size))
             text = this(bound[1],target_lang,trans_lang)
             draw.text((x, y),text, font = font, fill=(0,0,0))
@@ -209,9 +211,13 @@ def blur_im(img,bounds,target_lang,trans_lang):
             pass
     return im
     
-def draw_boxes(image, bounds, color='blue', width=1):
+def draw_boxes(image, bounds, width=1):
     draw = ImageDraw.Draw(image)
     for bound in bounds:
+        if bound[2]>=0.3:
+            color = "green"
+        else:
+            color = "red"
         p0, p1, p2, p3 = bound[0]
         draw.line([*p0, *p1, *p2, *p3, *p0], fill=color, width=width)
     return image
@@ -233,10 +239,6 @@ def detect(img, target_lang,trans_lang,target_lang2=None):
 
     blr_out=blur_im(img,bounds,target_lang,trans_lang)
     return im_out,blr_out,pd.DataFrame(bounds),pd.DataFrame(bounds).iloc[:,1:]
-
-
-    
-
     
 with gr.Blocks() as robot:
     with gr.Row():