Update app.py

app.py

@@ -329,14 +329,7 @@ def get_mesh(image, depth, blur_data, loadall, dcolor):
     glb_file = tempfile.NamedTemporaryFile(suffix='.glb', delete=False)
     scene.export(glb_file.name)
     print('file - ok')
-    
-    bg = json.loads(dcolor)
-    j = 0
-    while j < len(bg):
-        bg[j] = bg[j] / 255
-        j=j+1
-    print(bg)
-    return gr.Model3D(value=glb_file.name, clear_color=bg), glb_file.name, ",".join(mesh_n)
+    return glb_file.name, glb_file.name, ",".join(mesh_n)
 
 def blur_image(image, depth, blur_data):
     blur_a = blur_data.split()
@@ -553,11 +546,16 @@ async(f, c, o, b)=>{
 """
 
 on_click="""
-async()=>{
+async(of, om, bi, la, bg)=>{
   var intv = setInterval(function(){
     if (document.getElementById("iframe3D")===null || typeof document.getElementById("iframe3D")==="undefined") {
       if (BABYLON && BABYLON.Engine.LastCreatedScene) {
         BABYLON.Engine.LastCreatedScene.onAfterRenderObservable.add(function() { //onDataLoadedObservable
+          var bg = JSON.parse(bg);
+          for (var i=0; i<bg.length; i++) {
+            bg[i] /= 255;
+          }
+          BABYLON.Engine.LastCreatedScene.clearColor = new BABYLON.Color4(bg[0], bg[1], bg[2], bg[3]);
           for (var i=0; i<BABYLON.Engine.LastCreatedScene.getNodes().length; i++) {
             if (BABYLON.Engine.LastCreatedScene.getNodes()[i].material) {
               BABYLON.Engine.LastCreatedScene.getNodes()[i].material.pointSize = Math.ceil(Math.log2(Math.PI/document.getElementById("zoom").value));
@@ -1070,7 +1068,6 @@ with gr.Blocks(css=css, js=js) as demo:
             <canvas id='cnv_out'/>""")
             load_all = gr.Checkbox(label="Load all")
             render = gr.Button("Render")
-            render.click(None, inputs=None, outputs=None, js=on_click)
             input_json.input(show_json, inputs=[input_json], outputs=[processed_video, processed_zip, output_frame, output_mask, coords])
     
     def on_submit(uploaded_video,model_type,coordinates):
@@ -1110,7 +1107,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(partial(get_mesh), inputs=[output_frame, output_mask, blur_in, load_all, bgcolor], outputs=[result, result_file, mesh_order])
+    render.click(partial(get_mesh), inputs=[output_frame, output_mask, blur_in, load_all, bgcolor], outputs=[result, result_file, mesh_order], js=on_click)
 
     example_files = os.listdir('examples')
     example_files.sort()