Update app_rvc.py

app_rvc.py

@@ -1,5 +1,6 @@
 import gradio as gr
 import os
+import logging
 os.system("pip install -q piper-tts==1.2.0")
 os.system("pip install -q -r requirements_xtts.txt")
 os.system("pip install -q TTS==0.21.1  --no-deps")
@@ -127,6 +128,8 @@ directories = [
     if not os.path.exists(directory)
 ]
 
+logging.basicConfig(level=logging.DEBUG)
+logger = logging.getLogger(__name__)
 
 class TTS_Info:
     def __init__(self, piper_enabled, xtts_enabled):
@@ -281,7 +284,7 @@ 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):
+def download_and_adjust_youtube_video(url: str, speed_factor: float, start_time: float = None, end_time: float = None) -> str:
     # Create the 'downloaded' folder if it doesn't exist
     os.makedirs("downloaded", exist_ok=True)
 
@@ -296,40 +299,37 @@ def download_and_adjust_youtube_video(url, speed_factor, start_time=None, end_ti
         info = ydl.extract_info(url, download=True)
         filename = ydl.prepare_filename(info)
 
-    # Load the video
-    video = VideoFileClip(filename)
-    logger.debug(f"Original video FPS: {video.fps}")
-    logger.debug(f"Original video duration: {video.duration}")
+    # 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}")
 
-    # Set a default fps if it's None
-    if video.fps is None:
-        video.fps = 30.0
-        logger.warning(f"Original video FPS is None. Setting to default: {video.fps}")
+            # 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}")
 
-    # 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)
+            # 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)
-    adjusted_video.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}")
+            # 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}")
 
-    # Generate output filename
-    output_filename = f"downloaded/{os.path.splitext(os.path.basename(filename))[0]}_speed{speed_factor}.mp4"
+            # 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, fps=adjusted_video.fps)
 
-    # Write the adjusted video
-    try:
-        adjusted_video.write_videofile(output_filename, fps=adjusted_video.fps)
     except Exception as e:
-        logger.error(f"Error writing video file: {str(e)}")
+        logger.error(f"Error processing video file: {str(e)}")
         raise
 
-    # Close the video objects
-    video.close()
-    adjusted_video.close()
-
     return output_filename
 
 class SoniTranslate(SoniTrCache):