big fixed

video_watermark_remover.py

@@ -80,35 +80,6 @@ def remove_watermark(sketch, images_path='frames', output_path='output_images'):
 
     return gr.Video.update(value=convert_frames_to_video('output_images'), visible=True), gr.Button.update(value='Done!')
 
-# def convert_video_to_frames(video):
-#     print(f" input video is : {video}")
-#     if os.path.exists('input_video.mp4'):
-#         os.remove('input_video.mp4')
-
-#     ffmpeg.input(video).output('input_video.mp4').run()
-#     video_path = 'input_video.mp4'
-    
-#     if os.path.exists('frames'):
-#         shutil.rmtree('frames')
-#     os.makedirs('frames')
-
-#     video_name = os.path.splitext(os.path.basename(video_path))[0]  
-#     ffmpeg.input(video_path).output(f'frames/{video_name}_%d.jpg', qscale=2).run()
-#     return gr.Image.update(value=f"{os.getcwd()}/frames/{video_name}_1.jpg", interactive=True), gr.Button.update(interactive=True)
-
-# def convert_frames_to_video(frames_path):
-#     if os.path.exists('output_video.mp4'):
-#         os.remove('output_video.mp4')
-    
-#     (
-#         ffmpeg
-#         .input(f'{frames_path}/*.jpg', pattern_type='glob', framerate=25)
-#         .output('output_video.mp4')
-#         .run()
-#     )
-#     return gr.Video.update(value='output_video.mp4', visible=True, interactive=True), gr.Button.update(interactive=False)
-
-
 def convert_video_to_frames(video):
     if os.path.exists('input_video.mp4'):
         os.remove('input_video.mp4')
@@ -154,4 +125,4 @@ def convert_frames_to_video(frames_path):
         out.write(img_array[i])
     out.release()
 
-    return gr.Video.update(value='output_video.mp4', visible=True, interactive=True), gr.Button.update(interactive=False)
+    return 'output_video.mp4'