Update app.py

app.py

@@ -749,9 +749,9 @@ async(c, o, p, d, n, m)=>{
         function requestMap() {
           ctx.drawImage(video, 0, 0, video.videoWidth, video.videoHeight);
           if (buffer) {
+            counter = parseInt(video.currentTime);
+            if (!coords[counter]) {counter = coords.length-1;}
             applyDisplacementMapFromBuffer(sceneToRender.meshes[sceneToRender.meshes.length-1], buffer, video.videoWidth, video.videoHeight, 0, -1, null, null, true);
-            counter++;
-            if (counter == coords.length) {counter = 0;}
           }
           buffer = ctx.getImageData(0, 0, video.videoWidth, video.videoHeight).data;
           applyDisplacementMapFromBuffer(BABYLON.Engine.LastCreatedScene.meshes[BABYLON.Engine.LastCreatedScene.meshes.length-1], buffer, video.videoWidth, video.videoHeight, 0, 1, null, null, true);
@@ -774,8 +774,7 @@ async(c, o, p, d, n, m)=>{
         });
 
         videoDome.videoTexture.video.oncanplaythrough = function () {
-          videoDome.videoTexture.video.play();
-          video.play();
+          play();
           videoDome.videoTexture.video.oncanplaythrough = null;
           video.oncanplaythrough = null;
         };
@@ -945,6 +944,21 @@ function stop_recording() {
 }
 window.stop_recording = stop_recording;
 
+function seek() {
+  videoDome.videoTexture.video.pause();
+  video.pause();
+  var t = Math.random();
+  videoDome.videoTexture.video.currentTime = t * videoDome.videoTexture.video.duration;
+  video.currentTime = t * video.duration;
+}
+window.seek = seek;
+
+function play() {
+  videoDome.videoTexture.video.play();
+  video.play();
+}
+window.play = play;
+
 
     function applyDisplacementMapFromBuffer(
         mesh,
@@ -1270,7 +1284,9 @@ with gr.Blocks(css=css, js=js, head=head) as demo:
             snapshot = gr.HTML(value="""
               <a href='#' onclick='snapshot();'>Screenshot</a>
               <a href='#' onclick='record_video();'>Record</a>
-              <a href='#' onclick='stop_recording();'>Stop</a><br/>
+              <a href='#' onclick='stop_recording();'>Stop rec.</a>
+              <a href='#' onclick='seek();'>Seek</a>
+              <a href='#' onclick='play();'>Play</a><br/>
               <img src='' id='img_out' onload='var ctxt = document.getElementById(\"cnv_out\").getContext(\"2d\");ctxt.drawImage(this, 0, 0);'/><br/>
               <canvas id='cnv_out'></canvas>
               <div id='compass_box'><div id='compass'><a id='fullscreen' onclick='