Update app.py

app.py

@@ -115,8 +115,12 @@ def main(share=True):
             cl, ind = pcd.remove_statistical_outlier(nb_neighbors=20, std_ratio=2.0)
             pcd = pcd.select_by_index(ind)
 
-        tempdir = tempfile.TemporaryDirectory()
-        out_dir = tempdir.name
+        tempdir = Path(tempfile.gettempdir(), 'ppd')
+        tempdir.mkdir(exist_ok=True)
+        output_path = Path(tempdir, request.session_hash)
+        shutil.rmtree(output_path, ignore_errors=True)
+        output_path.mkdir(exist_ok=True, parents=True)
+        
         pointcloud_path = os.path.join(out_dir, 'pointcloud.ply')
 
         # save pcd to temporary .ply
@@ -128,14 +132,14 @@ def main(share=True):
         vertices[:, 2] = -vertices[:, 2]
         vertices[:, 1] = -vertices[:, 1]
         mesh = trimesh.PointCloud(vertices=vertices, colors=vertex_colors)
-        glb_path = os.path.join(out_dir, 'pointcloud.glb')
+        glb_path = os.path.join(output_path, 'pointcloud.glb')
         # tmp_glb = tempfile.NamedTemporaryFile(suffix='.glb', delete=False)
         mesh.export(glb_path)
 
 
         # save raw depth (npy)
         depth = cv2.resize(ppd_depth, (W, H), interpolation=cv2.INTER_LINEAR)
-        raw_depth_path = os.path.join(out_dir, 'raw_depth.npy')
+        raw_depth_path = os.path.join(output_path, 'raw_depth.npy')
         np.save(raw_depth_path, depth)
     
         # tmp_raw_depth = tempfile.NamedTemporaryFile(suffix='.npy', delete=False)
@@ -148,14 +152,20 @@ def main(share=True):
         split_region = np.ones((image.shape[0], 50, 3), dtype=np.uint8) * 255
         combined_result = cv2.hconcat([image[:, :, ::-1], split_region, colored_depth[:, :, ::-1]])
 
-        vis_path = os.path.join(out_dir, 'image_depth_vis.png')
+        vis_path = os.path.join(output_path, 'image_depth_vis.png')
         cv2.imwrite(vis_path, combined_result)
     
         # tmp_concat = tempfile.NamedTemporaryFile(suffix='.png', delete=False)
         # cv2.imwrite(tmp_concat.name, combined_result)
 
         # download_files = [tmp_concat.name, tmp_raw_depth.name, tmp_ply.name]
-        download_files = [vis_path, raw_depth_path, pointcloud_path]
+        file_names = ["image_depth_vis.png", "raw_depth.npy", "pointcloud.ply"]
+        download_files = [
+            (output_path / name).as_posix()
+            for name in file_names
+            if (output_path / name).exists()
+        ]
+        # download_files = [vis_path, raw_depth_path, pointcloud_path]
 
         return [(image, colored_depth), glb_path, download_files]