Update app_rvc.py

app_rvc.py

@@ -1,5 +1,6 @@
 import gradio as gr
 import os
+import subprocess
 import logging
 os.system("pip install -q piper-tts==1.2.0")
 os.system("pip install -q -r requirements_xtts.txt")
@@ -128,7 +129,6 @@ directories = [
     if not os.path.exists(directory)
 ]
 
-logging.basicConfig(level=logging.DEBUG)
 logger = logging.getLogger(__name__)
 
 class TTS_Info:
@@ -284,7 +284,7 @@ def check_openai_api_key():
             "translation process in Advanced settings."
         )
 
-def download_and_adjust_youtube_video(url: str, speed_factor: float, start_time: float = None, end_time: float = None) -> str:
+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)
 
@@ -299,35 +299,33 @@ def download_and_adjust_youtube_video(url: str, speed_factor: float, start_time:
         info = ydl.extract_info(url, download=True)
         filename = ydl.prepare_filename(info)
 
-    # Process the video
-    try:
-        with VideoFileClip(filename) as video:
-            logger.debug(f"Original video FPS: {video.fps}")
-            logger.debug(f"Original video duration: {video.duration}")
+    logger.info(f"Downloaded video: {filename}")
 
-            # Set a default fps if it's None
-            if video.fps is None:
-                video = video.set_fps(30.0)
-                logger.warning(f"Original video FPS is None. Setting to default: {video.fps}")
+    # Generate output filename
+    output_filename = f"downloaded/{os.path.splitext(os.path.basename(filename))[0]}_speed{speed_factor}.mp4"
 
-            # 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)
+    # Prepare FFmpeg command
+    ffmpeg_cmd = ['ffmpeg', '-i', filename]
 
-            # Adjust speed
-            adjusted_video = video.speedx(speed_factor)
-            adjusted_video = adjusted_video.set_fps(video.fps)  # Ensure the adjusted video has the same fps
-            logger.debug(f"Adjusted video FPS: {adjusted_video.fps}")
-            logger.debug(f"Adjusted video duration: {adjusted_video.duration}")
+    # Add trim options if start_time and end_time are provided
+    if start_time is not None and end_time is not None:
+        ffmpeg_cmd.extend(['-ss', str(start_time), '-to', str(end_time)])
 
-            # Generate output filename
-            output_filename = f"downloaded/{os.path.splitext(os.path.basename(filename))[0]}_speed{speed_factor}.mp4"
+    # Add speed adjustment
+    filter_complex = f"[0:v]setpts={1/speed_factor}*PTS[v];[0:a]atempo={speed_factor}[a]"
+    ffmpeg_cmd.extend(['-filter_complex', filter_complex, '-map', '[v]', '-map', '[a]'])
 
-            # Write the adjusted video
-            adjusted_video.write_videofile(output_filename, fps=adjusted_video.fps)
+    # Output options
+    ffmpeg_cmd.extend(['-c:v', 'libx264', '-c:a', 'aac', '-y', output_filename])
 
-    except Exception as e:
-        logger.error(f"Error processing video file: {str(e)}")
+    # Run FFmpeg command
+    try:
+        logger.info(f"Running FFmpeg command: {' '.join(ffmpeg_cmd)}")
+        subprocess.run(ffmpeg_cmd, check=True, capture_output=True, text=True)
+        logger.info(f"Video processing completed: {output_filename}")
+    except subprocess.CalledProcessError as e:
+        logger.error(f"Error processing video: {e}")
+        logger.error(f"FFmpeg stderr: {e.stderr}")
         raise
 
     return output_filename