Update app.py

app.py

@@ -391,6 +391,14 @@ def get_mesh(image, depth, blur_data, loadall):
     #print('file - ok')
     return "./TriangleWithoutIndices.gltf", ",".join(mesh_n)
 
+
+def scene_file(data):
+    f = open("./scene.babylon", "w")
+    f.write(data)
+    f.close()
+    return "./scene.babylon"
+    
+
 def blur_image(image, depth, blur_data):
     blur_a = blur_data.split()
     #print(f'blur data {blur_data}')
@@ -790,6 +798,12 @@ async(c, o, p, d, n, m, s)=>{
           toggleDisplay("model");
         }
         clearInterval(intv);
+
+
+        var serializedScene = BABYLON.SceneSerializer.Serialize(BABYLON.Engine.LastCreatedScene);
+        var strScene = JSON.stringify(serializedScene);
+
+        return strScene;
       }
       } catch(e) {alert(e);}
     }
@@ -1148,6 +1162,14 @@ with gr.Blocks(css=css, js=js, head=head) as demo:
           processed_zip = gr.File(label="Output Archive", interactive=False)
           depth_video = gr.Video(label="Depth Video", format="mp4", elem_id="depth_video", interactive=False, visible=True)
           result = gr.Model3D(label="3D Mesh", clear_color=[0.5, 0.5, 0.5, 0.0], camera_position=[0, 90, 0], zoom_speed=2.0, pan_speed=2.0, interactive=True, elem_id="model3D")
+          with gr.Accordion(label="Embed in website", open=False):
+            embed_model = gr.Textbox(elem_id="embed_model", label="Include this wherever the model is to appear on the page", interactive=False, value="""
+              <script src='https://preview.babylonjs.com/viewer/babylon.viewer.js'></script>
+              <babylon model='./scene.babylon'></babylon>
+            """)
+            embed_json = gr.JSON(label="Scene", interactive=False)
+            embed_zip = gr.File(label="Scene File", interactive=False)
+            embed_json.change(fn=scene_file, inputs=[embed_json], outputs=[embed_zip])
           with gr.Tab("Blur"):
             chart_c = gr.HTML(elem_id="chart_c", value="""<div id='chart' onpointermove='window.drawLine(event.clientX, event.clientY);' onpointerdown='window.pointerDown(event.clientX, event.clientY);' onpointerup='window.pointerUp();' onpointerleave='window.pointerUp();' onpointercancel='window.pointerUp();' onclick='window.resetLine();'></div>
             <style>
@@ -1371,7 +1393,7 @@ with gr.Blocks(css=css, js=js, head=head) as demo:
         return output_video_path + (json.dumps(locations),)
 
     submit.click(on_submit, inputs=[input_video, model_type, blur_in, boffset, bsize, coords], outputs=[processed_video, processed_zip, output_frame, output_mask, output_depth, depth_video, coords])
-    render.click(None, inputs=[coords, mesh_order, output_frame, output_mask, selected, output_depth, output_switch], outputs=None, js=load_model)
+    render.click(None, inputs=[coords, mesh_order, output_frame, output_mask, selected, output_depth, output_switch], outputs=[embed_json], js=load_model)
     render.click(partial(get_mesh), inputs=[output_frame, output_mask, blur_in, load_all], outputs=[result, mesh_order])
 
     example_files = [["./examples/streetview.mp4", "vits", blurin, 1, 32, example_coords]]