Update app.py

app.py

@@ -561,7 +561,7 @@ def draw_mask(o, b, v, d, evt: gr.EventData):
 load_model="""
 async(c, o, p, d, n, m)=>{
   var intv = setInterval(function(){
-    if (document.getElementById("iframe3D")===null || typeof document.getElementById("iframe3D")==="undefined") {
+    if (document.getElementById("model3D").getElementsByTagName("canvas")[0]) {
       try {
       if (typeof BABYLON !== "undefined" && BABYLON.Engine && BABYLON.Engine.LastCreatedScene) {
         BABYLON.Engine.LastCreatedScene.onAfterRenderObservable.add(function() { //onDataLoadedObservable
@@ -639,38 +639,13 @@ async(c, o, p, d, n, m)=>{
           BABYLON.Engine.LastCreatedScene.meshes[BABYLON.Engine.LastCreatedScene.meshes.length-1].material.useAlphaFromDiffuseTexture = true;
           BABYLON.Engine.LastCreatedScene.meshes[BABYLON.Engine.LastCreatedScene.meshes.length-1].applyDisplacementMap(m[o_[j]].url, 0, 255, function(m){try{alert(BABYLON.Engine.Version);}catch(e){alert(e);}}, null, null, true, function(e){alert(e);});
         }
+        if (document.getElementById("model")) {
+          document.getElementById("model").appendChild(document.getElementById("model3D"));
+          toggleDisplay("model");
+        }
         clearInterval(intv);
       }
       } catch(e) {alert(e);}
-    } else if (BABYLON || BABYLON == null) {
-    try {
-      BABYLON = null;
-      if (document.getElementById("model3D").getElementsByTagName("canvas")[0]) {
-        document.getElementById("model3D").getElementsByTagName("canvas")[0].remove();
-      }
-      document.getElementById("iframe3D").src = "index.htm";
-      document.getElementById("iframe3D").onload = function() {
-        if (o.indexOf(""+n) < 0) {
-          if (o != "") { o += ","; }
-          o += n;
-        }
-        //alert(o);
-        var o_ = o.split(",");
-        document.getElementById("iframe3D").contentDocument.getElementById("coords").value = c;
-        document.getElementById("iframe3D").contentDocument.getElementById("order").value = o;
-        document.getElementById("iframe3D").contentDocument.getElementById("bgimage").value = "";
-        document.getElementById("iframe3D").contentDocument.getElementById("bgdepth").value = "";
-        for (var j=0; j<o_.length; j++) {
-          o_[j] = parseInt(o_[j]);
-          //alert(o_[j]);
-          document.getElementById("iframe3D").contentDocument.getElementById("bgimage").value += p[o_[j]].image.url + ",";
-          document.getElementById("iframe3D").contentDocument.getElementById("bgdepth").value += m[o_[j]].url + ",";
-        }
-      }
-      toggleDisplay("model");
-      
-      clearInterval(intv);
-    } catch(e) {alert(e)}
     }
   }, 40);
 }