Update app.py

app.py

@@ -344,18 +344,22 @@ def process_images(input_folder='./output/merged_imgs'):
 
 def process_line(img, outdir='./output'):
     print(f"Received img type: {type(img)}")
-    
-    # 确保 img 是一个有效的 NumPy 数组
-    if isinstance(img, np.ndarray):
+
+    # 检查输入是否为字符串路径
+    if isinstance(img, str):
+        img = cv2.imread(img)  # 读取图像文件
+        if img is None:
+            raise ValueError("Failed to read the image. Check the file path.")
+    elif isinstance(img, np.ndarray):
         img = img.astype(np.uint8)  # 转换为 uint8
-        
-        # 创建输入目录
-        os.makedirs("./input", exist_ok=True)
-        # 保存上传的图像
-        cv2.imwrite(os.path.join("./input", "input_image.png"), img)
     else:
         raise ValueError("Input img is not a valid NumPy array")
 
+    # 创建输入目录
+    os.makedirs("./input", exist_ok=True)
+    # 保存上传的图像
+    cv2.imwrite(os.path.join("./input", "input_image.png"), img)
+
     # 继续处理
     depth_anything("./input", os.path.join(outdir, "depth_anything"))
     teed_imgs("./input", os.path.join(outdir, "teed_imgs"), [1, 7, 2])