Update app.py

app.py

@@ -35,37 +35,38 @@ def sample_info(sample_id_str, fieldn):
     str__ = f"loading sample {sample_id_str}"
     
     
-    # generate mesh
-    sphere = trimesh.creation.icosphere(subdivisions=4, radius=0.8)
-    sphere.vertices+=1e-2*np.random.randn(*sphere.vertices.shape)
-    mesh = pyrender.Mesh.from_trimesh(sphere, smooth=False)
+    # # generate mesh
+    # sphere = trimesh.creation.icosphere(subdivisions=4, radius=0.8)
+    # sphere.vertices+=1e-2*np.random.randn(*sphere.vertices.shape)
+    # mesh = pyrender.Mesh.from_trimesh(sphere, smooth=False)
     
-    # compose scene
-    scene = pyrender.Scene(ambient_light=[.1, .1, .3], bg_color=[0, 0, 0])
-    camera = pyrender.PerspectiveCamera( yfov=np.pi / 3.0)
-    light = pyrender.DirectionalLight(color=[1,1,1], intensity=2e3)
+    # # compose scene
+    # scene = pyrender.Scene(ambient_light=[.1, .1, .3], bg_color=[0, 0, 0])
+    # camera = pyrender.PerspectiveCamera( yfov=np.pi / 3.0)
+    # light = pyrender.DirectionalLight(color=[1,1,1], intensity=2e3)
     
-    scene.add(mesh, pose=  np.eye(4))
-    scene.add(light, pose=  np.eye(4))
+    # scene.add(mesh, pose=  np.eye(4))
+    # scene.add(light, pose=  np.eye(4))
     
-    c = 2**-0.5
-    scene.add(camera, pose=[[ 1,  0,  0,  0],
-                            [ 0,  c, -c, -2],
-                            [ 0,  c,  c,  2],
-                            [ 0,  0,  0,  1]])
+    # c = 2**-0.5
+    # scene.add(camera, pose=[[ 1,  0,  0,  0],
+    #                         [ 0,  c, -c, -2],
+    #                         [ 0,  c,  c,  2],
+    #                         [ 0,  0,  0,  1]])
     
-    # render scene
-    r = pyrender.OffscreenRenderer(512, 512)
-    color, _ = r.render(scene)
+    # # render scene
+    # r = pyrender.OffscreenRenderer(512, 512)
+    # color, _ = r.render(scene)
     
-    plt.figure(figsize=(8,8))
-    plt.imshow(color)
+    # plt.figure(figsize=(8,8))
+    # plt.imshow(color)
     
-    plt.savefig("test.png")
+    # plt.savefig("test.png")
 
 
 
-    return str__, "test.png"
+    # return str__, "test.png"
+    return str__, str__
 
 
 if __name__ == "__main__":
@@ -76,7 +77,8 @@ if __name__ == "__main__":
 
 
         output1 = gr.Text(label="Training sample info")
-        output2 = gr.Image(label="Training sample visualization")
+        output2 = gr.Text(label="Training sample visualization")
+        # output2 = gr.Image(label="Training sample visualization")
 
         # d1.input(update_second, d1, d2)