Omnibus commited on
Commit
2099764
β€’
1 Parent(s): 186d961

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +15 -3
app.py CHANGED
@@ -198,7 +198,13 @@ ocr_lang=[
198
 
199
 
200
 
 
 
 
201
 
 
 
 
202
  def draw_boxes(image, bounds, color='blue', width=1):
203
  draw = ImageDraw.Draw(image)
204
  for bound in bounds:
@@ -214,12 +220,18 @@ def detect(img, target_lang,target_lang2=None):
214
  else:
215
  lang=[f"{lang_id[target_lang]}"]
216
  pass
217
-
218
  reader = easyocr.Reader(lang)
219
  bounds = reader.readtext(img)
220
  im = PIL.Image.open(img)
221
  im_out=draw_boxes(im, bounds)
222
- return im_out,pd.DataFrame(bounds),pd.DataFrame(bounds).iloc[:,1:]
 
 
 
 
 
 
223
  with gr.Blocks() as robot:
224
  with gr.Row():
225
  with gr.Column():
@@ -241,5 +253,5 @@ with gr.Blocks() as robot:
241
  gr.Column()
242
 
243
 
244
- go_btn.click(detect,[im,target_lang,target_lang2],[out_im,out_txt,data_f])
245
  robot.queue(concurrency_count=10).launch()
 
198
 
199
 
200
 
201
+ def blur_im(img,bounds):
202
+ for bound in bounds:
203
+ p0, p1, p2, p3 = bound[0]
204
 
205
+ img[p0, p2] = cv2.blur(img[p0, p2] ,(23,23))
206
+ return img
207
+
208
  def draw_boxes(image, bounds, color='blue', width=1):
209
  draw = ImageDraw.Draw(image)
210
  for bound in bounds:
 
220
  else:
221
  lang=[f"{lang_id[target_lang]}"]
222
  pass
223
+ global bounds
224
  reader = easyocr.Reader(lang)
225
  bounds = reader.readtext(img)
226
  im = PIL.Image.open(img)
227
  im_out=draw_boxes(im, bounds)
228
+ blr_out=blur_im(im,bounds)
229
+ return im_out,blr_out,pd.DataFrame(bounds),pd.DataFrame(bounds).iloc[:,1:]
230
+
231
+
232
+
233
+
234
+
235
  with gr.Blocks() as robot:
236
  with gr.Row():
237
  with gr.Column():
 
253
  gr.Column()
254
 
255
 
256
+ go_btn.click(detect,[im,target_lang,target_lang2],[out_im,trans_im,out_txt,data_f])
257
  robot.queue(concurrency_count=10).launch()