Update app.py

app.py

@@ -203,8 +203,9 @@ def pano_depth_to_world_points(depth):
     z = radius * np.sin(lon) * np.sin(lat)
 
     pts3d = np.stack([x, y, z], axis=1)
+    uv = np.stack([lon, lat], axis=1)
 
-    return pts3d
+    return [pts3d, uv]
 
 def rgb2gray(rgb):
     return np.dot(rgb[...,:3], [0.333, 0.333, 0.333])
@@ -212,7 +213,9 @@ def rgb2gray(rgb):
 def get_mesh(image, depth):
     gdepth = rgb2gray(depth)
     print('depth to gray - ok')
-    pts3d = pano_depth_to_world_points(gdepth)
+    points = pano_depth_to_world_points(gdepth)
+    pts3d = points[0]
+    uv = points[1]
     print('radius from depth - ok')
 
     # Create a trimesh mesh from the points
@@ -224,7 +227,7 @@ def get_mesh(image, depth):
     print('triangles - ok')
     #colors = image.reshape(-1, 3)
     mesh = trimesh.Trimesh(vertices=verts, faces=triangles)#, vertex_colors=colors)
-    uv = mesh.visual.uv
+    print(mesh.vertex_normals[0])
     material = trimesh.visual.texture.SimpleMaterial(image=image)
     texture = trimesh.visual.TextureVisuals(uv=uv, image=image, material=material)
     mesh.visual = texture