Update app.py

[fix]more assertion

app.py

@@ -32,28 +32,46 @@ def generate_mask(img, coord):
     #x, y = map(int, coord.split(','))
     
     #
-    mask = sepia(img,img,img,img,img, stack_image=False) 
+    mask = sepia(img,img*0.999999,img*0.999999,img*0.999999,img*0.999999, stack_image=False) 
     return overlay_mask(img, mask)
 
 def overlay_mask(img, mask):
-    # 尺寸对齐
-    mask_resized = cv2.resize(mask, (img.shape[1], img.shape[0]))  # 匹配原图尺寸
+    # 前置校验（参考网页1数据验证方案）
+    assert mask is not None, "掩码数据为空"
+    assert img.size > 0 and mask.size > 0, "输入图像或掩码尺寸异常"
     
-    # 通道数转换
+    # 数据类型强制转换（根据网页2的浮点型要求）
+    mask_float = mask.astype(np.float32) if mask.dtype != np.float32 else mask
+    
+    # 维度压缩（应对4D数组问题）
+    if mask_float.ndim == 4:
+        mask_float = np.squeeze(mask_float, axis=(0,1))  # 去除批次和通道维度
+    
+    # 尺寸调整（参考网页3的最佳实践）
+    try:
+        mask_resized = cv2.resize(
+            mask_float, 
+            (img.shape[1], img.shape[0]), 
+            interpolation=cv2.INTER_CUBIC  # 使用高阶插值保持缺陷边缘
+        )
+    except cv2.error as e:
+        print(f"Resize参数异常: 原图尺寸{img.shape} 掩码尺寸{mask.shape}")
+        raise
+    
+    # 通道统一（三通道对齐）
     if len(mask_resized.shape) == 2:
         mask_rgb = cv2.cvtColor(mask_resized, cv2.COLOR_GRAY2BGR)
     else:
         mask_rgb = mask_resized
         
-    # 归一化处理（参考网页5的医学图像处理方案）
-    mask_normalized = mask_rgb.astype(np.float32) / 255.0
-    img_normalized = img.astype(np.float32) / 255.0
-    
-    # 带透明度叠加（根据网页3的工业标准）
-    alpha = 0.7  # 原图可见度
-    blended = cv2.addWeighted(img_normalized, alpha, 
-                             mask_normalized, 1-alpha, 0)
-    return (blended * 255).astype(np.uint8)
+    # 归一化处理
+    return cv2.addWeighted(
+        img.astype(np.float32)/255, 
+        0.7,
+        mask_rgb.astype(np.float32)/255,
+        0.3,
+        0
+    ).astype(np.uint8)
 
 def create_mode2_interface():
     with gr.Blocks() as mode2: