Update app.py

app.py

@@ -27,6 +27,7 @@ from src.misc.image_io import save_interpolated_video
 from src.model.model.anysplat import AnySplat
 from src.model.ply_export import export_ply
 from src.utils.image import process_image
+import open3d as o3d
 
 os.environ["ANYSPLAT_PROCESSED"] = f"{os.getcwd()}/proprocess_results"
 
@@ -75,9 +76,22 @@ def get_reconstructed_scene(outdir, model, device):
         save_sh_dc_only=True,
     )
 
-
-    from instant_texture import Converter
-    Converter().convert(plyfile, glbfile)  # outputs scene.glb with baked albedo
+    import trimesh
+    import numpy as np
+    
+    # 1. Load PLY and preserve attributes
+    mesh = trimesh.load(plyfile, process=False)
+    
+    # 2. Check or assign vertex colors
+    if mesh.visual.vertex_colors is None or mesh.visual.vertex_colors.shape[1] < 4:
+        # Example: assume mesh.metadata['vertex_color'] holds (N×3) array
+        rgb = np.array(mesh.metadata['vertex_color'], dtype=np.uint8)
+        alpha = np.full((rgb.shape[0], 1), 255, dtype=np.uint8)
+        mesh.visual.vertex_colors = np.concatenate([rgb, alpha], axis=1)
+    
+    # 3. Export GLB
+    mesh.export(glbfile, file_type='glb')
+    print("Export complete: scene_colored.glb")
 
     # Clean up
     torch.cuda.empty_cache()