Update app.py

app.py

@@ -9,29 +9,34 @@ from tempfile import TemporaryDirectory
 cachedir = ".cache"
 
 
-def clip(yt_url, start_timestamp, end_timestamp):
+def clip(yt_url, video_file_path, start_timestamp, end_timestamp):
     start_timestamp = int(start_timestamp)
     duration = int(end_timestamp) - int(start_timestamp)
     
     with TemporaryDirectory() as tmpdir:
-        video_output_path = os.path.join(tmpdir, "video.mp4")
-        ydl_opts = {
-            "quiet": True,
-            "outtmpl": video_output_path,
-            "postprocessors": [{"key": "FFmpegVideoConvertor", "preferedformat": "mp4"}],
-            "cachedir": cachedir
-        }
-        if not os.path.exists(video_output_path):
-            try:
-                ydl_opts["format"] = "22"
-                with yt_dlp.YoutubeDL(ydl_opts) as ydl:
-                    ydl.download([yt_url])
-            except Exception as e:
-                print(e)
-                ydl_opts["format"] = "bestvideo[height<=480][ext=mp4]+bestaudio[ext=m4a]/best[height<=480][ext=mp4]/best"
-                sleep(5)
-                with yt_dlp.YoutubeDL(ydl_opts) as ydl:
-                    ydl.download([yt_url])
+        if yt_url is not None:
+            video_output_path = os.path.join(tmpdir, "video.mp4")
+            ydl_opts = {
+                "quiet": True,
+                "outtmpl": video_output_path,
+                "postprocessors": [{"key": "FFmpegVideoConvertor", "preferedformat": "mp4"}],
+                "cachedir": cachedir
+            }
+            if not os.path.exists(video_output_path):
+                try:
+                    ydl_opts["format"] = "22"
+                    with yt_dlp.YoutubeDL(ydl_opts) as ydl:
+                        ydl.download([yt_url])
+                except Exception as e:
+                    print(e)
+                    ydl_opts["format"] = "bestvideo[height<=480][ext=mp4]+bestaudio[ext=m4a]/best[height<=480][ext=mp4]/best"
+                    sleep(5)
+                    with yt_dlp.YoutubeDL(ydl_opts) as ydl:
+                        ydl.download([yt_url])
+        elif os.path.exists(video_file_path):
+            video_output_path = video_file_path
+        else:
+            raise Exception("Select valid input.")
         clip_path = f"{uuid4()}.mp3"
         print(f'ffmpeg -ss {start_timestamp} -t {duration} -i "{video_output_path}" "{clip_path}"')
         subprocess.run(f'ffmpeg -ss {start_timestamp} -t {duration} -i "{video_output_path}" "{clip_path}"', shell=True)
@@ -39,10 +44,11 @@ def clip(yt_url, start_timestamp, end_timestamp):
 
 
 url_link = gr.Textbox(label="YouTube URL")
+video_file = gr.Video(label="Upload Video")
 start_timestamp = gr.Textbox(label="Clip start timestamp (in seconds)")
 end_timestamp = gr.Textbox(label="Clip end timestamp (in seconds)")
 
-interface = gr.Interface(clip, [url_link, start_timestamp, end_timestamp], "audio", title="YouTube Clipper", description="After downloading the clips, please go to https://demo.deepsync.co/11-voice-cloning to clone the voice.")
+interface = gr.Interface(clip, [url_link, video_file, start_timestamp, end_timestamp], "audio", title="YouTube / Video Clipper", description="After downloading the clips, please go to https://demo.deepsync.co/11-voice-cloning to clone the voice.")
 
 if __name__=="__main__":
     interface.queue().launch(auth=(os.environ.get("GRADIO_USERNAME"), os.environ.get("GRADIO_PASSWORD")))