app

app.py

@@ -133,7 +133,6 @@ def do_inference(input_3d, sample_seed=0, do_sampling=False, do_marching_cubes=F
         mesh.visual.vertex_colors = np.tile(orange_color, (mesh.vertices.shape[0], 1))
     input_save_name = f"processed_input_{int(time.time())}.obj"
     mesh.export(input_save_name)
-    input_render_res = wireframe_render(mesh)
 
     pc_coor = pc_coor / np.abs(pc_coor).max() * 0.9995 # input should be from -1 to 1
     assert (np.linalg.norm(normals, axis=-1) > 0.99).all(), "normals should be unit vectors, something wrong"
@@ -176,8 +175,7 @@ def do_inference(input_3d, sample_seed=0, do_sampling=False, do_marching_cubes=F
     # add time stamp to avoid cache
     save_name = f"output_{int(time.time())}.obj"
     artist_mesh.export(save_name)
-    output_render = wireframe_render(artist_mesh)
-    return input_save_name, input_render_res, save_name, output_render
+    return input_save_name, input_save_name, save_name, save_name
 
 
 _HEADER_ = '''
@@ -215,11 +213,15 @@ preprocess_model_obj = gr.Model3D(
     label="Processed Input Mesh (OBJ Format)",
     clear_color=[1, 1, 1, 1],
 )
-input_image_render = gr.Image(
-    label="Wireframe Render of Processed Input Mesh",
-)
-output_image_render = gr.Image(
+input_wireframe_render = gr.Model3D(
+                    label="Wireframe Render of Processed Input Mesh",
+                    clear_color=[1,1,1,1],
+                    display_mode="wireframe"
+                )
+output_wireframe_render = gr.Model3D(
     label="Wireframe Render of Generated Mesh",
+    clear_color=[1, 1, 1, 1],
+    display_mode="wireframe"
 )
 with (gr.Blocks() as demo):
     gr.Markdown(_HEADER_)
@@ -246,19 +248,19 @@ with (gr.Blocks() as demo):
                         os.path.join("examples", img_name) for img_name in sorted(os.listdir("examples"))
                     ],
                     inputs=input_3d,
-                    outputs=[preprocess_model_obj, input_image_render, output_model_obj, output_image_render],
+                    outputs=[preprocess_model_obj, input_wireframe_render, output_model_obj, output_wireframe_render],
                     fn=do_inference,
                     cache_examples = "lazy",
                     examples_per_page=10
                 )
         with gr.Column():
             with gr.Row():
-                input_image_render.render()
+                input_wireframe_render.render()
             with gr.Row():
                 with gr.Tab("OBJ"):
                     preprocess_model_obj.render()
             with gr.Row():
-                output_image_render.render()
+                output_wireframe_render.render()
             with gr.Row():
                 with gr.Tab("OBJ"):
                     output_model_obj.render()
@@ -272,7 +274,7 @@ with (gr.Blocks() as demo):
     submit.click(
         fn=do_inference,
         inputs=[input_3d, sample_seed, do_sampling, do_marching_cubes],
-        outputs=[preprocess_model_obj, input_image_render, output_model_obj, output_image_render],
+        outputs=[preprocess_model_obj, input_wireframe_render, output_model_obj, output_wireframe_render],
     )
 
 demo.launch(share=True)