support direct mesh visualizer

app.py

@@ -113,17 +113,29 @@ def generate_mesh(image, source_size=512, render_size=384, mesh_size=512, export
             mesh.export(mesh_path, 'obj')
             return mesh_path
 
-# TODO: instead of outputting .obj file -> directly output a 3d model
-def gradio_interface(image):
-    mesh_file = generate_mesh(image)
-    print("Generated Mesh File Path:", mesh_file)
-    return mesh_file
+# we will convert image to mesh
+def step_1_generate_obj(image):
+    mesh_path = generate_mesh(image)
+    return mesh_path
 
+# we will convert mesh to 3d-image
+def step_2_display_3d_model(mesh_file):
+    return mesh_file
 
-gr.Interface(
-    fn=gradio_interface,
-    inputs=gr.Image(type="pil", label="Input Image"),
-    outputs=gr.File(label="Awesome 3D Mesh (.obj)"),
-    title="3D Mesh Generator by FacebookAI",
-    description="Upload an image and generate a 3D mesh (.obj) file using VFusion3D by FacebookAI"
-).launch()
+with gr.Blocks() as demo:
+    with gr.Row():
+        with gr.Column():
+            img_input = gr.Image(type="pil", label="Input Image")
+            generate_button = gr.Button("Generate and Visualize 3D Model")
+            obj_file_output = gr.File(label="Download .obj File")
+            
+        with gr.Column():
+            model_output = gr.Model3D(clear_color=[0.0, 0.0, 0.0, 0.0], label="3D Model Visualization")
+    
+    def generate_and_visualize(image):
+        mesh_path = step_1_generate_obj(image)
+        return mesh_path, mesh_path
+    
+    generate_button.click(generate_and_visualize, inputs=img_input, outputs=[obj_file_output, model_output])
+
+demo.launch()