Update app.py

app.py

@@ -202,7 +202,7 @@ ocr_lang=[
 
 
 
-def blur_im(img,bounds):
+def blur_im(img,bounds,target_lang,trans_lang):
     im = cv2.imread(img)
     im = cv2.cvtColor(im, cv2.COLOR_BGR2RGB)
     for bound in bounds:
@@ -216,7 +216,6 @@ def blur_im(img,bounds):
         im[y:y+h, x:x+w] = cv2.GaussianBlur(im[y:y+h, x:x+w],(51,51),0)
         
         
-    text = "New Text"
 
     im = Image.fromarray(im)
     for bound in bounds:
@@ -228,11 +227,8 @@ def blur_im(img,bounds):
         draw = ImageDraw.Draw(im)
         font_size=int(int(w)*0.2)
         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))
-        #kernel = np.ones((1, 1), np.uint8)
-        #im[y:y+h, x:x+w] = cv2.dilate(im[y:y+h, x:x+w], kernel, iterations=1)
-        #im[y:y+h, x:x+w] = cv2.erode(im[y:y+h, x:x+w], kernel, iterations=1)
-
     return im
     
 def draw_boxes(image, bounds, color='blue', width=1):
@@ -242,7 +238,7 @@ def draw_boxes(image, bounds, color='blue', width=1):
         draw.line([*p0, *p1, *p2, *p3, *p0], fill=color, width=width)
     return image
 
-def detect(img, target_lang,target_lang2=None):
+def detect(img, target_lang,trans_lang,target_lang2=None):
     if target_lang2 != None and target_lang2 != "":
         lang=f"{lang_id[target_lang]}"
         lang2=f"{lang_id[target_lang2]}"
@@ -257,7 +253,7 @@ def detect(img, target_lang,target_lang2=None):
     im_out=draw_boxes(im, bounds)
     #im.save('result.jpg')
 
-    blr_out=blur_im(img,bounds)
+    blr_out=blur_im(img,bounds,target_lang,trans_lang)
     return im_out,blr_out,pd.DataFrame(bounds),pd.DataFrame(bounds).iloc[:,1:]
 
 
@@ -269,8 +265,9 @@ with gr.Blocks() as robot:
         with gr.Column():
             im=gr.Image(type="filepath")
         with gr.Column():
-            target_lang = gr.Dropdown(label="Detect language", choices=list(lang_id.keys()),value="English")
-            target_lang2 = gr.Dropdown(label="Detect language2", choices=list(lang_id.keys()),value="")
+            target_lang = gr.Dropdown(label="Detect language:", choices=list(lang_id.keys()),value="English")
+            #target_lang2 = gr.Dropdown(label="Detect language2", choices=list(lang_id.keys()),value="")
+            trans_lang = gr.Dropdown(label="Translate to:", choices=list(lang_id.keys()),value="Chinese")
             go_btn=gr.Button()
     with gr.Row():
         with gr.Column():
@@ -282,6 +279,6 @@ with gr.Blocks() as robot:
         out_txt=gr.Textbox(lines=8)
         data_f=gr.Dataframe()
                         
-
-    go_btn.click(detect,[im,target_lang,target_lang2],[out_im,trans_im,out_txt,data_f])
+    go_btn.click(detect,[im,target_lang,trans_lang],[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()