Update app.py

app.py

@@ -116,13 +116,23 @@ def calculate_delaunay_triangles(rect, points):
 def warp_triangle(img1,img2,t1,t2):
     if len(t1)!=3 or len(t2)!=3:
         return
-    r1=cv2.boundingRect(np.float32([t1])); r2=cv2.boundingRect(np.float32([t2]))
-    if r1[2]==0 or r1[3]==0 or r2[2]==0 or r2[3]==0:
+    r1=cv2.boundingRect(np.float32([t1]))
+    r2=cv2.boundingRect(np.float32([t2]))
+    
+    if r1[2] <= 0 or r1[3] <= 0 or r2[2] <= 0 or r2[3] <= 0:
         return
+
+    img1_rect = img1[r1[1]:r1[1]+r1[3], r1[0]:r1[0]+r1[2]]
+    if img1_rect.size == 0: 
+        return
+
     t1r=[(t1[i][0]-r1[0],t1[i][1]-r1[1]) for i in range(3)]
     t2r=[(t2[i][0]-r2[0],t2[i][1]-r2[1]) for i in range(3)]
+    
     mask=np.zeros((r2[3],r2[2],3),dtype=np.float32)
+    
     cv2.fillConvexPoly(mask,np.int32(t2r),(1,1,1),16)
+    
     src=img1[r1[1]:r1[1]+r1[3],r1[0]:r1[0]+r1[2]]
     M=cv2.getAffineTransform(np.float32(t1r),np.float32(t2r))
     warped=cv2.warpAffine(src,M,(r2[2],r2[3]),flags=cv2.INTER_LINEAR,borderMode=cv2.BORDER_REFLECT_101)