Update app.py

app.py

@@ -349,15 +349,29 @@ def process_images(input_folder='./output/merged_imgs'):
 
 def process_line(img, outdir='./output'):
     img_path = "./input"
-    cv2.imwrite(os.path.join(img_path, "input_image.png"), img)  # 保存上传的图像
-    depth_anything(img_path, os.path.join(outdir, "depth_anything"))
-    teed_imgs(img_path, os.path.join(outdir, "teed_imgs"), [1, 7, 2])
-    teed_imgs(os.path.join(outdir, "depth_anything"), os.path.join(outdir, "dp_teed_imgs"), [0, 7, 2])
-    merge_images_in_2_folder(os.path.join(outdir, "teed_imgs"), os.path.join(outdir, "dp_teed_imgs"), os.path.join(outdir, "merged_imgs"), '_depth', 1, 'multiply', [[2, 0], [2, 1]], [1, 0])
     
-    # 返回处理后的最后一张图像
-    result_img_path = os.path.join(outdir, "merged_imgs", "your_final_image_name.png")  # 修改为你的输出图像路径
-    return Image.open(result_img_path)
+    # 保存上传的图像
+    cv2.imwrite(os.path.join(img_path, "input_image.png"), img)  
+    
+    # 确保输出目录存在
+    os.makedirs(outdir, exist_ok=True)
+
+    try:
+        depth_anything(img_path, os.path.join(outdir, "depth_anything"))
+        teed_imgs(img_path, os.path.join(outdir, "teed_imgs"), [1, 7, 2])
+        teed_imgs(os.path.join(outdir, "depth_anything"), os.path.join(outdir, "dp_teed_imgs"), [0, 7, 2])
+        merge_images_in_2_folder(os.path.join(outdir, "teed_imgs"), 
+                                  os.path.join(outdir, "dp_teed_imgs"), 
+                                  os.path.join(outdir, "merged_imgs"), 
+                                  '_depth', 1, 'multiply', [[2, 0], [2, 1]], [1, 0])
+        
+        # 返回处理后的最后一张图像
+        result_img_path = os.path.join(outdir, "merged_imgs", "your_final_image_name.png")  # 修改为你的输出图像路径
+        return Image.open(result_img_path)
+
+    except Exception as e:
+        print(f"Error processing the image: {e}")
+        return None
 
 
 if __name__ == '__main__':