added convertor

app.py

@@ -1,110 +1,7 @@
-import glob
-import os
-import io
-import ffmpeg
-import requests
-from PIL import Image
 import gradio as gr
-import shutil
-import concurrent.futures
+from video_watermark_remover import *
+from video_convertor import *
 
-def process_image(mask_data, image_path):
-    image = Image.open(image_path)
-    image_data = io.BytesIO()
-    image.save(image_data, format=image.format)
-    image_data = image_data.getvalue()
-
-    # Prepare form data
-    form_data = {
-            'ldmSteps': 25,
-            'ldmSampler': 'plms',
-            'zitsWireframe': True,
-            'hdStrategy': 'Original',
-            'hdStrategyCropMargin': 196,
-            'hdStrategyCropTrigerSize': 1280,
-            'hdStrategyResizeLimit': 2048,
-            'prompt': '',
-            'negativePrompt': '',
-            'croperX': -24,
-            'croperY': -23,
-            'croperHeight': 512,
-            'croperWidth': 512,
-            'useCroper': False,
-            'sdMaskBlur': 5,
-            'sdStrength': 0.75,
-            'sdSteps': 50,
-            'sdGuidanceScale': 7.5,
-            'sdSampler': 'pndm',
-            'sdSeed': 42,
-            'sdMatchHistograms': False,
-            'sdScale': 1,
-            'cv2Radius': 5,
-            'cv2Flag': 'INPAINT_NS',
-            'paintByExampleSteps': 50,
-            'paintByExampleGuidanceScale': 7.5,
-            'paintByExampleSeed': 42,
-            'paintByExampleMaskBlur': 5,
-            'paintByExampleMatchHistograms': False,
-            'sizeLimit': 1024,
-        }
-
-    files_data = {
-        'image': (os.path.basename(image_path), image_data),
-        'mask': ('mask.png', mask_data)
-    }
-
-    response = requests.post('https://ahmedghani-lama-cleaner-lama.hf.space/inpaint', data=form_data, files=files_data)
-
-    if response.headers['Content-Type'] == 'image/jpeg' or response.headers['Content-Type'] == 'image/png':
-        output_image_path = os.path.join('output_images', os.path.splitext(os.path.basename(image_path))[0] + '_inpainted' + os.path.splitext(image_path)[1])
-        with open(output_image_path, 'wb') as output_image_file:
-            output_image_file.write(response.content)
-    else:
-        print(f"Error processing {image_path}: {response.text}")
-
-def remove_watermark(sketch, images_path='frames', output_path='output_images'):
-    if os.path.exists('output_images'):
-        shutil.rmtree('output_images')
-    os.makedirs('output_images')
-
-    mask_data = io.BytesIO()
-    sketch["mask"].save(mask_data, format=sketch["mask"].format)
-    mask_data = mask_data.getvalue()
-
-    image_paths = glob.glob(f'{images_path}/*.*')
-
-    with concurrent.futures.ThreadPoolExecutor() as executor:
-        executor.map(lambda image_path: process_image(mask_data, image_path), image_paths)
-
-    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)
 
 css = """
     #remove_btn {
@@ -116,33 +13,73 @@ css = """
     #remove_btn:hover {
         background: linear-gradient(#2bbbc3, #201d18);
     }
+    #convert_btn {
+        background: linear-gradient(#201d18, #2bbbc3);
+        font-weight: bold;
+        font-size: 18px;
+        color:white;
+    }
+    #convert_btn:hover {
+        background: linear-gradient(#2bbbc3, #201d18);
+    }
     footer {
         display: none !important;
     }
 """
+
 demo = gr.Blocks(css=css, title="Video Watermark Remover")
 with demo:
-    gr.Markdown("""
-    # <center>🎥 Video Watermark Remover</center>
-    """)
-    with gr.Row():
-        with gr.Column():
-            input_video = gr.Video(label="Upload a Video")
-        with gr.Column():
-            mask = gr.Image(label="Create a mask for the image", tool="sketch", type="pil", interactive=False)
-    with gr.Row():
-        with gr.Column():
-            pass
-        with gr.Column():
-            remove_btn = gr.Button("Remove Watermark", interactive=False, elem_id="remove_btn")
-        with gr.Column():
-            pass
-    
-    output_video = gr.Video(label="Output Video", interactive=False)
-    input_video.change(convert_video_to_frames, inputs=[input_video], outputs=[mask, remove_btn])
-    remove_btn.click(remove_watermark, inputs=[mask], outputs=[output_video, remove_btn])
-
-    #position:fixed;bottom:0;left:0;right:0;
+    with gr.Tab("Video Watermark Remover"):
+        gr.Markdown("""
+        # <center>🎥 Video Watermark Remover</center>
+        """)
+        with gr.Row():
+            with gr.Column():
+                input_video = gr.Video(label="Upload a Video")
+            with gr.Column():
+                mask = gr.Image(label="Create a mask for the image", tool="sketch", type="pil", interactive=False)
+        with gr.Row():
+            with gr.Column():
+                pass
+            with gr.Column():
+                remove_btn = gr.Button("Remove Watermark", interactive=False, elem_id="remove_btn")
+            with gr.Column():
+                pass
+        
+        output_video = gr.Video(label="Output Video", interactive=False)
+        input_video.change(convert_video_to_frames, inputs=[input_video], outputs=[mask, remove_btn])
+        remove_btn.click(remove_watermark, inputs=[mask], outputs=[output_video, remove_btn])
+
+    with gr.Tab("Video Convertor"):
+        gr.Markdown("""
+        # <center>🎥 Video Convertor</center>
+        """)
+        video_format = ['webm', 'wmv', 'mkv', 'mp4', 'avi', 'mpeg', 'vob', 'flv']
+        audio_format = ['mp3', 'wav', 'ogg', 'flac', 'aac']
+        with gr.Row():
+            with gr.Column():
+                input_video = gr.Video(label="Upload a Video")
+            with gr.Column():
+                with gr.Row():
+                    format_select = gr.Radio(["Video", "Audio"], label="Select Format", default="Video")
+                with gr.Row():
+                    format = gr.Radio(video_format, label="Select Format", interactive=False)
+        with gr.Row():
+            with gr.Column():
+                pass
+            with gr.Column():
+                convert_btn = gr.Button("Convert Video", interactive=False, elem_id="convert_btn")
+            with gr.Column():
+                pass
+        with gr.Row():
+            with gr.Column():
+                output_video = gr.Video(label="Output Video", interactive=False)
+            with gr.Column():
+                output_audio = gr.Audio(label="Output Audio", interactive=False)
+        status = gr.Textbox(label="Status", interactive=False)
+        format_select.change(lambda x: gr.Radio.update(choices=video_format if x == "Video" else audio_format, interactive=True), inputs=[format_select], outputs=[format])
+        format.change(lambda x: gr.Button.update(interactive=True), None, outputs=[convert_btn])
+        convert_btn.click(convert_video, inputs=[input_video, format], outputs=[output_audio, output_video, status])
     gr.Markdown("""## <center style="margin:20px;">Developed by Muhammad Ahmed<img src="https://avatars.githubusercontent.com/u/63394104?v=4" style="height:50px;width:50px;border-radius:50%;margin:5px;"></img></center>
     """)
 demo.launch(show_api=False)

requirements.txt

@@ -1,2 +1,4 @@
 gradio==3.22.1
-ffmpeg-python
+ffmpeg-python
+moviepy
+pydub

video_convertor.py

@@ -0,0 +1,57 @@
+import os
+from moviepy.editor import *
+from pydub import AudioSegment
+
+class VideoConverter:
+    def __init__(self, input_file):
+        self.input_file = input_file
+        self.video = None
+        self.audio = None
+
+        if not os.path.exists(self.input_file):
+            raise FileNotFoundError(f"File not found: {self.input_file}")
+
+        self.load_video()
+
+    def load_video(self):
+        try:
+            self.video = VideoFileClip(self.input_file)
+            self.audio = self.video.audio
+        except Exception as e:
+            raise Exception(f"Error loading video: {e}")
+
+    def convert_video(self, output_file, format):
+        if format not in ['webm', 'wmv', 'mkv', 'mp4', 'avi', 'mpeg', 'vob', 'flv']:
+            raise ValueError(f"Unsupported format: {format}")
+
+        try:
+            self.video.write_videofile(output_file, codec=format.lower())
+            print(f"Video converted to {format} format successfully!")
+            return output_file
+        except Exception as e:
+            raise Exception(f"Error converting video: {e}")
+
+
+    def convert_audio(self, output_file, format):
+        if format not in ['mp3', 'wav', 'ogg', 'flac', 'aac']:
+            raise ValueError(f"Unsupported format: {format}")
+
+        try:
+            audio_segment = AudioSegment.from_file(self.input_file, self.audio.codec)
+            audio_segment.export(output_file, format=format.lower())
+            print(f"Audio converted to {format} format successfully!")
+            return output_file
+        except Exception as e:
+            raise Exception(f"Error converting audio: {e}")
+
+def convert_video(input_file, format):
+    try:
+        converter = VideoConverter(input_file)
+        if format in ['webm', 'wmv', 'mkv', 'mp4', 'avi', 'mpeg', 'vob', 'flv']:
+            return None, converter.convert_video(f"output.{format}", format), "Converted video successfully!"
+        elif format in ['mp3', 'wav', 'ogg', 'flac', 'aac']:
+            return converter.convert_audio(f"output.{format}", format), None, "Converted audio successfully!"
+        else:
+            return None, None, "Unsupported format!"
+    except Exception as e:
+        return None, None, str(e)

video_watermark_remover.py

@@ -0,0 +1,157 @@
+import glob
+import os
+import io
+import ffmpeg
+import requests
+from PIL import Image
+import shutil
+import concurrent.futures
+import gradio as gr
+import cv2
+import re
+
+def process_image(mask_data, image_path):
+    image = Image.open(image_path)
+    image_data = io.BytesIO()
+    image.save(image_data, format=image.format)
+    image_data = image_data.getvalue()
+
+    # Prepare form data
+    form_data = {
+            'ldmSteps': 25,
+            'ldmSampler': 'plms',
+            'zitsWireframe': True,
+            'hdStrategy': 'Original',
+            'hdStrategyCropMargin': 196,
+            'hdStrategyCropTrigerSize': 1280,
+            'hdStrategyResizeLimit': 2048,
+            'prompt': '',
+            'negativePrompt': '',
+            'croperX': -24,
+            'croperY': -23,
+            'croperHeight': 512,
+            'croperWidth': 512,
+            'useCroper': False,
+            'sdMaskBlur': 5,
+            'sdStrength': 0.75,
+            'sdSteps': 50,
+            'sdGuidanceScale': 7.5,
+            'sdSampler': 'pndm',
+            'sdSeed': 42,
+            'sdMatchHistograms': False,
+            'sdScale': 1,
+            'cv2Radius': 5,
+            'cv2Flag': 'INPAINT_NS',
+            'paintByExampleSteps': 50,
+            'paintByExampleGuidanceScale': 7.5,
+            'paintByExampleSeed': 42,
+            'paintByExampleMaskBlur': 5,
+            'paintByExampleMatchHistograms': False,
+            'sizeLimit': 1024,
+        }
+
+    files_data = {
+        'image': (os.path.basename(image_path), image_data),
+        'mask': ('mask.png', mask_data)
+    }
+
+    response = requests.post('https://ahmedghani-lama-cleaner-lama.hf.space/inpaint', data=form_data, files=files_data)
+
+    if response.headers['Content-Type'] == 'image/jpeg' or response.headers['Content-Type'] == 'image/png':
+        output_image_path = os.path.join('output_images', os.path.splitext(os.path.basename(image_path))[0] + '_inpainted' + os.path.splitext(image_path)[1])
+        with open(output_image_path, 'wb') as output_image_file:
+            output_image_file.write(response.content)
+    else:
+        print(f"Error processing {image_path}: {response.text}")
+
+def remove_watermark(sketch, images_path='frames', output_path='output_images'):
+    if os.path.exists('output_images'):
+        shutil.rmtree('output_images')
+    os.makedirs('output_images')
+
+    mask_data = io.BytesIO()
+    sketch["mask"].save(mask_data, format=sketch["mask"].format)
+    mask_data = mask_data.getvalue()
+
+    image_paths = glob.glob(f'{images_path}/*.*')
+
+    with concurrent.futures.ThreadPoolExecutor() as executor:
+        executor.map(lambda image_path: process_image(mask_data, image_path), image_paths)
+
+    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')
+
+    os.system(f"ffmpeg -i {video} input_video.mp4")
+    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]
+    vidcap = cv2.VideoCapture(video_path)
+    success, image = vidcap.read()
+    count = 1
+    while success:
+        cv2.imwrite(f"frames/{video_name}_{count}.jpg", image)
+        success, image = vidcap.read()
+        count += 1
+
+    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')
+
+    img_array = []
+    filelist = glob.glob(f"{frames_path}/*.jpg")
+
+    # Sort frames by number
+    frame_numbers = [int(re.findall(r'\d+', os.path.basename(frame))[0]) for frame in filelist]
+    sorted_frames = [frame for _, frame in sorted(zip(frame_numbers, filelist), key=lambda pair: pair[0])]
+
+    for filename in sorted_frames:
+        img = cv2.imread(filename)
+        height, width, layers = img.shape
+        size = (width, height)
+        img_array.append(img)
+
+    out = cv2.VideoWriter('output_video.mp4', cv2.VideoWriter_fourcc(*'mp4v'), 25, size)
+
+    for i in range(len(img_array)):
+        out.write(img_array[i])
+    out.release()
+
+    return gr.Video.update(value='output_video.mp4', visible=True, interactive=True), gr.Button.update(interactive=False)