Update app_rvc.py

download and fix updated!

app_rvc.py

@@ -105,6 +105,9 @@ import argparse
 import time
 import hashlib
 import sys
+import yt_dlp
+from moviepy.editor import VideoFileClip
+
 
 directories = [
     "downloads",
@@ -276,6 +279,42 @@ def check_openai_api_key():
             "translation process in Advanced settings."
         )
 
+def download_and_adjust_youtube_video(url, speed_factor, start_time=None, end_time=None):
+    # Create the 'downloaded' folder if it doesn't exist
+    os.makedirs("downloaded", exist_ok=True)
+
+    # Configure yt-dlp options
+    ydl_opts = {
+        'format': 'bestvideo[ext=mp4]+bestaudio[ext=m4a]/best[ext=mp4]/best',
+        'outtmpl': 'downloaded/%(title)s.%(ext)s',
+    }
+
+    # Download the video
+    with yt_dlp.YoutubeDL(ydl_opts) as ydl:
+        info = ydl.extract_info(url, download=True)
+        filename = ydl.prepare_filename(info)
+
+    # Load the video and adjust speed
+    video = VideoFileClip(filename)
+
+    # Trim the video if start_time and end_time are provided
+    if start_time is not None and end_time is not None:
+        video = video.subclip(start_time, end_time)
+
+    # Adjust speed
+    adjusted_video = video.speedx(speed_factor)
+
+    # Generate output filename
+    output_filename = f"downloaded/{os.path.splitext(os.path.basename(filename))[0]}_speed{speed_factor}.mp4"
+
+    # Write the adjusted video
+    adjusted_video.write_videofile(output_filename)
+
+    # Close the video objects
+    video.close()
+    adjusted_video.close()
+
+    return output_filename
 
 class SoniTranslate(SoniTrCache):
     def __init__(self, cpu_mode=False):
@@ -2111,6 +2150,22 @@ def create_gui(theme, logs_in_gui=False):
                         outputs=[video_output],
                         cache_examples=False,
                     )
+        
+
+        with gr.Tab("YouTube Download"):
+            with gr.Column():
+                youtube_url = gr.Textbox(label="YouTube URL")
+                speed_factor = gr.Slider(minimum=0.5, maximum=2.0, value=1.0, step=0.1, label="Speed Factor")
+                start_time = gr.Number(label="Start Time (seconds)", value=None)
+                end_time = gr.Number(label="End Time (seconds)", value=None)
+                download_button = gr.Button("Download and Adjust")
+                download_output = gr.File(label="Downloaded Video")
+
+            download_button.click(
+                download_and_adjust_youtube_video,
+                inputs=[youtube_url, speed_factor, start_time, end_time],
+                outputs=download_output
+            )
 
         with gr.Tab(lg_conf["tab_docs"]):
             with gr.Column():
@@ -2921,4 +2976,4 @@ if __name__ == "__main__":
         show_error=True,
         quiet=False,
         debug=(True if logger.isEnabledFor(logging.DEBUG) else False),
-    )
+    )