Update app.py

app.py

@@ -34,22 +34,18 @@ def inference(filepath, lang):
     bounds = reader.readtext(filepath)
     new_bounds=[]
     for (bbox, text, prob) in bounds:
-        x1,y1 = bbox[0]
-        x2,y2 = bbox[1]
-        x3,y3 = bbox[2]
-        x4,y4 = bbox[3]
-        
-        # crop the region of interest (ROI)
-        img = Image.open(filepath)
-        #img = img[y0:y1, x0:x1]
-        width, height =img.size
-        cropped_image = img.crop((max(0,x1-5), max(0,y1-5), min(x3+5,width), min(y3+5, height))) # crop the image
-        try:
-            cropped_image = Image.fromarray(cropped_image)
-        except:
-            print('cropping error')
-            continue
+        (tl, tr, br, bl) = bbox
+        tl = (int(tl[0]), int(tl[1]))
+        tr = (int(tr[0]), int(tr[1]))
+        br = (int(br[0]), int(br[1]))
+        bl = (int(bl[0]), int(bl[1]))
 
+        min_x = min(tl[0], tr[0], br[0], bl[0])
+        max_x = max(tl[0], tr[0], br[0], bl[0])
+        min_y = min(tl[1], tr[1], br[1], bl[1])
+        max_y = max(tl[1], tr[1], br[1], bl[1])
+        # crop the region of interest (ROI)
+        cropped_image = Image.fromarray(img[min_y:max_y,min_x:max_x]) # crop the image
         out = recognitor.predict(cropped_image)
         print(out)
         new_bounds.append((bbox,text, out, prob))