Update app.py

app.py

@@ -74,7 +74,7 @@ def predict_depth(image, model):
 #def predict_depth(model, image):
 #    return model(image)["depth"]
 
-def make_video(video_path, outdir='./vis_video_depth', encoder='vits'):
+def make_video(video_path, outdir='./vis_video_depth', encoder='vits', blur_data):
     if encoder not in ["vitl","vitb","vits","vitg"]:
         encoder = "vits"
 
@@ -218,7 +218,7 @@ def make_video(video_path, outdir='./vis_video_depth', encoder='vits'):
                       continue
               thumbnail_old = thumbnail
             
-            cv2.imwrite(f"f{count}.png", raw_frame)
+            cv2.imwrite(f"f{count}.png", blur_image(raw_frame, depth_color, blur_data))
             orig_frames.append(f"f{count}.png")
             
             cv2.imwrite(f"f{count}_dmap.png", depth_color)
@@ -228,8 +228,8 @@ def make_video(video_path, outdir='./vis_video_depth', encoder='vits'):
             masks.append(f"f{count}_mask.png")
             count += 1
 
-        #final_vid = create_video(orig_frames, frame_rate, "orig")
-        final_vid = create_video(depth_frames, frame_rate, "depth")
+        final_vid = create_video(orig_frames, frame_rate, "orig")
+        #final_vid = create_video(depth_frames, frame_rate, "depth")
             
         final_zip = zip_files(orig_frames, depth_frames)
         raw_video.release()
@@ -1140,7 +1140,7 @@ with gr.Blocks(css=css, js=js) as demo:
             render = gr.Button("Render")
             input_json.input(show_json, inputs=[input_json], outputs=[processed_video, processed_zip, output_frame, output_mask, output_depth, coords])
     
-    def on_submit(uploaded_video,model_type,coordinates):
+    def on_submit(uploaded_video,model_type,blur_in,coordinates):
         global locations
         locations = []
         avg = [0, 0]
@@ -1174,16 +1174,16 @@ with gr.Blocks(css=css, js=js) as demo:
         print(locations)
             
         # Process the video and get the path of the output video
-        output_video_path = make_video(uploaded_video,encoder=model_type)
+        output_video_path = make_video(uploaded_video,encoder=model_type,blur_in=blur_in)
 
         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, output_depth, coords])
+    submit.click(on_submit, inputs=[input_video, model_type, blur_in, coords], outputs=[processed_video, processed_zip, output_frame, output_mask, output_depth, coords])
     render.click(None, inputs=[coords, mesh_order, bgcolor, output_frame, output_mask, selected, output_depth], outputs=None, js=load_model)
     render.click(partial(get_mesh), inputs=[output_frame, output_mask, blur_in, load_all], outputs=[result, result_file, mesh_order])
 
-    example_files = [["./examples/streetview.mp4", "vits", example_coords]]
-    examples = gr.Examples(examples=example_files, fn=on_submit, cache_examples=True, inputs=[input_video, model_type, coords], outputs=[processed_video, processed_zip, output_frame, output_mask, output_depth, coords])
+    example_files = [["./examples/streetview.mp4", "vits", blur_in, example_coords]]
+    examples = gr.Examples(examples=example_files, fn=on_submit, cache_examples=True, inputs=[input_video, model_type, blur_in, coords], outputs=[processed_video, processed_zip, output_frame, output_mask, output_depth, coords])
     
 
 if __name__ == '__main__':