Update app.py

app.py

@@ -344,23 +344,21 @@ def process_images(input_folder='./output/merged_imgs'):
 
 def process_line(img, outdir='./output'):
     print(f"Received img type: {type(img)}")
-    img_path = "./input"
     
     # 确保 img 是一个有效的 NumPy 数组
     if isinstance(img, np.ndarray):
-        print(f"Image type: {type(img)}, shape: {img.shape}, dtype: {img.dtype}")
-        
-        # 确保图像是 uint8 类型
-        img = img.astype(np.uint8)
+        img = img.astype(np.uint8)  # 转换为 uint8
         
+        # 创建输入目录
+        os.makedirs("./input", exist_ok=True)
         # 保存上传的图像
-        cv2.imwrite(os.path.join(img_path, "input_image.png"), img)
+        cv2.imwrite(os.path.join("./input", "input_image.png"), img)
     else:
         raise ValueError("Input img is not a valid NumPy array")
 
     # 继续处理
-    depth_anything(img_path, os.path.join(outdir, "depth_anything"))
-    teed_imgs(img_path, os.path.join(outdir, "teed_imgs"), [1, 7, 2])
+    depth_anything("./input", os.path.join(outdir, "depth_anything"))
+    teed_imgs("./input", 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])
     
@@ -370,6 +368,7 @@ def process_line(img, outdir='./output'):
 
 
 
+
 if __name__ == '__main__':
     # depth_anything()
     # teed_imgs('./input', './output/teed_imgs', [1,7,2])
@@ -393,7 +392,6 @@ if __name__ == '__main__':
                              title="Image Processing",
                              description="Upload an image to process it using depth-anything and TEED.")
     
-
-
     # 启动 Gradio 接口
     interface.launch()
+