Update app_rvc.py

app_rvc.py

@@ -2,6 +2,7 @@ import gradio as gr
 import os
 import subprocess
 import logging
+import re
 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")
@@ -285,12 +286,6 @@ def check_openai_api_key():
             "translation process in Advanced settings."
         )
 
-import os
-import subprocess
-import yt_dlp
-import logging
-
-logger = logging.getLogger(__name__)
 
 def download_and_adjust_youtube_video(url, speed_factor, start_time=None, end_time=None):
     # Create the 'downloaded' folder if it doesn't exist
@@ -309,14 +304,15 @@ def download_and_adjust_youtube_video(url, speed_factor, start_time=None, end_ti
 
     logger.info(f"Downloaded video: {filename}")
 
-    # Generate output filename
-    output_filename = f"downloaded/{os.path.splitext(os.path.basename(filename))[0]}_speed{speed_factor}.mp4"
+    # Sanitize filename
+    safe_filename = re.sub(r'[^\w\-_\. ]', '_', os.path.splitext(os.path.basename(filename))[0])
+    output_filename = f"downloaded/{safe_filename}_speed{speed_factor}.mp4"
 
     # Prepare FFmpeg command
     ffmpeg_cmd = ['ffmpeg', '-i', filename]
 
-    # Add trim options if start_time and end_time are provided
-    if start_time is not None and end_time is not None:
+    # Add trim options if start_time and end_time are provided and valid
+    if start_time is not None and end_time is not None and start_time < end_time:
         ffmpeg_cmd.extend(['-ss', str(start_time), '-to', str(end_time)])
 
     # Add speed adjustment
@@ -335,7 +331,6 @@ def download_and_adjust_youtube_video(url, speed_factor, start_time=None, end_ti
     except subprocess.CalledProcessError as e:
         logger.error(f"Error processing video: {e}")
         logger.error(f"FFmpeg stderr: {e.stderr}")
-        logger.error(f"FFmpeg stdout: {e.stdout}")
         raise
 
     return output_filename