Update app.py

app.py

@@ -242,7 +242,6 @@ def make_video(video_path, outdir='./vis_video_depth', encoder='vits', blur_data
             count += 1
 
         final_vid = create_video(comb_frames, frame_rate, "orig")
-        depth_vid = create_video(depth_frames, frame_rate, "depth")
             
         final_zip = zip_files(orig_frames, depth_frames)
         raw_video.release()
@@ -265,7 +264,7 @@ def make_video(video_path, outdir='./vis_video_depth', encoder='vits', blur_data
         else:
             gradient = cv2.imread('./gradient_small.png').astype(np.uint8)
         
-        return final_vid, final_zip, frames, masks[frame_selected], depths, depth_vid #output_path
+        return final_vid, final_zip, frames, masks[frame_selected], depths #output_path
 
 def depth_edges_mask(depth):
     """Returns a mask of edges in the depth map.
@@ -453,38 +452,8 @@ def draw_mask(o, b, v, d, evt: gr.EventData):
     switch_rows(False)
     return gr.ImageEditor(value=d)
 
-
-load_model="""
-async()=>{
-  var intv = setInterval(function(){
-    if (document.getElementById("output_video").getElementsByTagName("video")) {
-try {
-      const xhttp = new XMLHttpRequest();
-      xhttp.onreadystatechange = function() {
-        if (this.readyState == 4 && this.status == 200) {
-          
-          const blob = new Blob([this.response], {
-            type: "video/mp4",
-          });
-          
-          document.getElementById("model3d").src = document.getElementById("model3d").src.split("?")[0] + "?url=" + URL.createObjectURL(blob);
-          document.getElementById("newtab").href = document.getElementById("model3d").src;
-        }
-      };
-      xhttp.responseType = 'arraybuffer';
-      xhttp.open("GET", document.getElementById("output_video").getElementsByTagName("video")[0].src);
-      xhttp.send(); 
-    
-      clearInterval(intv);
-} catch(e) {alert(e)}
-    }
-  }, 40);
-}
-"""
-
 js = """
 async()=>{
-console.log('Hi');
 
 const chart = document.getElementById('chart');
 const blur_in = document.getElementById('blur_in').getElementsByTagName('textarea')[0];
@@ -681,7 +650,6 @@ with gr.Blocks(css=css, js=js, head=head) as demo:
           model_type = gr.Dropdown([("small", "vits"), ("base", "vitb"), ("large", "vitl"), ("giant", "vitg")], type="value", value="vits", label='Model Type')
           processed_video = gr.Video(label="Output Video", format="mp4", elem_id="output_video", interactive=False)
           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)
             
           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>
@@ -752,9 +720,8 @@ with gr.Blocks(css=css, js=js, head=head) as demo:
                   {"lat": 50.073823157821664, "lng": 14.437124189538856, "heading": 152.95769, "pitch": 4.233024999999998}
                 ]"""
               coords = gr.Textbox(elem_id="coords", value=example_coords, label="Coordinates", interactive=False)
-            render = gr.Button("Render")
-            model3d = gr.HTML(value="""<a id='newtab' href='https://freeali.se/freealise/transparent_video/' target='_blank'>Open in new tab</a><br/>
-            <iframe id='model3d' src='https://freeali.se/freealise/transparent_video/' width='100%' height='320'></iframe>
+            model3d = gr.HTML(value="""
+              <a id='newtab' href='https://freeali.se/freealise/transparent_video/' target='_blank'>Open renderer in new tab and upload your video there</a>
             """)
             input_json.input(show_json, inputs=[input_json], outputs=[processed_video, processed_zip, output_frame, output_mask, output_depth, coords])
             
@@ -800,11 +767,10 @@ with gr.Blocks(css=css, js=js, head=head) as demo:
 
         return output_video_path + (json.dumps(locations),)
 
-    render.click(None, inputs=None, outputs=None, js=load_model)
-    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])
+    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, coords])
 
     example_files = [["./examples/streetview.mp4", "vits", blurin, 1, 32, example_coords]]
-    examples = gr.Examples(examples=example_files, fn=on_submit, cache_examples=True, inputs=[input_video, model_type, blur_in, boffset, bsize, coords], outputs=[processed_video, processed_zip, output_frame, output_mask, output_depth, depth_video, coords])
+    examples = gr.Examples(examples=example_files, fn=on_submit, cache_examples=True, inputs=[input_video, model_type, blur_in, boffset, bsize, coords], outputs=[processed_video, processed_zip, output_frame, output_mask, output_depth, coords])
     
 
 if __name__ == '__main__':