Update app.py

app.py

@@ -48,7 +48,7 @@ def process_image(image_path):
     except Exception:
         gltf_path = create_3d_obj(np.array(image), depth_image, image_path, depth=8)
     
-    return [Image.fromarray(depth_image), gltf_path]
+    return Image.fromarray(depth_image), gltf_path, gltf_path
 
 def create_3d_obj(rgb_image, depth_image, image_path, depth=10):
     depth_o3d = o3d.geometry.Image(depth_image)
@@ -80,32 +80,32 @@ def create_3d_obj(rgb_image, depth_image, image_path, depth=10):
     
     return gltf_path
 
-title = "Zero-shot Depth Estimation with DPT + 3D Point Cloud"
-description = "DPT model predicts depth from an image, followed by 3D Point Cloud reconstruction."
+title = "Zero-shot Depth Estimation with DPT + 3D Model Preview"
+description = "Upload an image to generate a depth map and reconstruct a 3D model in .gltf format."
+
+with gr.Blocks() as demo:
+    gr.Markdown(f"## {title}")
+    gr.Markdown(description)
 
-with gr.Blocks() as iface:
-    gr.Markdown("# Zero-shot Depth Estimation with DPT + 3D Point Cloud")
     with gr.Row():
-        image_input = gr.Image(type="filepath", label="Input Image")
-        depth_output = gr.Image(label="Predicted Depth", type="pil")
-    gltf_output = gr.File(label="Download 3D gLTF")
+        with gr.Column():
+            image_input = gr.Image(type="filepath", label="Upload Image")
+            generate_button = gr.Button("Generate 3D Model")
+        
+        with gr.Column():
+            depth_output = gr.Image(label="Predicted Depth", type="pil")
 
-    # Embed an iframe for previewing the .gltf
     with gr.Row():
-        gr.HTML('<iframe id="gltf-viewer" width="100%" height="400px"></iframe>')
+        model_output = gr.Model3D(label="3D Model Preview (GLTF)")
+    
+    with gr.Row():
+        file_output = gr.File(label="Download 3D GLTF File")
 
-    def update_gltf_viewer(image_path):
-        gltf_path = process_image(image_path)[1]
-        iframe_html = f'''
-        <script>
-            document.getElementById('gltf-viewer').src = 'https://gltf-viewer.donmccurdy.com/?url=file://{gltf_path}';
-        </script>
-        '''
-        return process_image(image_path)[0], gltf_path, iframe_html
+    generate_button.click(fn=process_image, inputs=[image_input], outputs=[depth_output, model_output, file_output])
 
-    image_input.change(update_gltf_viewer, inputs=[image_input], outputs=[depth_output, gltf_output, gr.HTML()])
+if __name__ == "__main__":
+    demo.launch()
 
-iface.launch()