Update app.py

app.py

@@ -662,12 +662,15 @@ async(p, d, n)=>{
         }
         BABYLON.Engine.LastCreatedScene.getMeshByName("dome_mesh").applyDisplacementMap(d.background.url, 0, 255, function(m){try{alert(BABYLON.Engine.Version);}catch(e){alert(e);}}, null, null, true, function(e){alert(e);});
 
-        clearInterval(intv);
         try {
         BABYLON.GLTF2Export.GLBAsync(BABYLON.Engine.LastCreatedScene, "fileName").then((glb) => {
-          return glb.glTFFiles.fileName;
+          document.getElementById("file3D").getElementsByTagName("a")[0].download = Object.keys(glb.glTFFiles)[0];
+          document.getElementById("file3D").getElementsByTagName("a")[0].innerText = Object.keys(glb.glTFFiles)[0];
+          document.getElementById("file3D").getElementsByTagName("a")[0].href = glb.glTFFiles[Object.keys(glb.glTFFiles)[0]];
         });
         } catch (e){alert(e);}
+        
+        clearInterval(intv);
       }
     } else if (BABYLON) {
       BABYLON = null;
@@ -1041,7 +1044,7 @@ with gr.Blocks(css=css, js=js) as demo:
                 coords = gr.Textbox(elem_id="coords", value=example_coords, label="Coordinates", interactive=False)
                 mesh_order = gr.Textbox(elem_id="order", value="", label="Order", interactive=False)
                 
-            result_file = gr.File(label="3D file (.glb)")
+            result_file = gr.File(elem_id="file3D", label="3D file (.glb)")
             result_file.change(fn=None, inputs=[result_file, coords, mesh_order, bgcolor, output_frame, output_mask, selected], outputs=None, js=load_model)
             html = gr.HTML(value="""<label for='zoom'>Zoom</label><input id='zoom' type='range' style='width:256px;height:1em;' value='0.8' min='0.157' max='1.57' step='0.001' oninput='
               if (!BABYLON.Engine.LastCreatedScene.activeCamera.metadata) {
@@ -1196,7 +1199,7 @@ with gr.Blocks(css=css, js=js) as demo:
         return output_video_path + (json.dumps(locations),)
 
     submit.click(on_submit, inputs=[input_video, model_type, coords], outputs=[processed_video, processed_zip, output_frame, output_mask, coords])
-    render.click(None, inputs=[output_frame, output_mask, selected], outputs=[result_file], js=on_click)
+    render.click(None, inputs=[output_frame, output_mask, selected], outputs=None, js=on_click)
     render.click(partial(get_mesh), inputs=[output_frame, output_mask, blur_in, load_all], outputs=[result, result_file, mesh_order])
 
     example_files = os.listdir('examples')