Update app.py

app.py

@@ -15,14 +15,15 @@ import yt_dlp
 ROOTPATH = os.path.dirname(os.path.abspath(__file__))
 MODEL_PATH = os.path.join(ROOTPATH, "weights_v3")
 FINISH_PATH = os.path.join(ROOTPATH, "finish")
-TEMP_AUDIO_PATH = os.path.join(ROOTPATH, "tmp", "audio_temp.wav")
+TEMP_PATH = os.path.join(ROOTPATH, "tmp")
+TEMP_AUDIO_PATH = os.path.join(TEMP_PATH, "audio_temp.wav")
 VIDEO_OUTPUT_PATH = os.path.join(FINISH_PATH, "upscaled_video_temp.mp4")
 FINAL_OUTPUT_PATH = os.path.join(FINISH_PATH, "upscaled_video.mp4")
 
 # Create directories
 try:
     os.makedirs(FINISH_PATH, exist_ok=True)
-    os.makedirs(os.path.dirname(TEMP_AUDIO_PATH), exist_ok=True)
+    os.makedirs(TEMP_PATH, exist_ok=True)
 except PermissionError as e:
     raise PermissionError(f"Failed to create directories for anime upscaling: {e}")
 
@@ -34,6 +35,16 @@ try:
 except FileNotFoundError:
     raise FileNotFoundError(f"Anime model directory not found at {MODEL_PATH}.")
 
+# Clean previous media files
+def clean_previous_media():
+    for folder in [TEMP_PATH, FINISH_PATH]:
+        for file in os.listdir(folder):
+            if file.endswith(('.mp4', '.png', '.wav')):
+                try:
+                    os.remove(os.path.join(folder, file))
+                except Exception as e:
+                    print(f"Error deleting previous file {file}: {str(e)}")
+
 # Check if video has audio
 def has_audio(video_path):
     try:
@@ -70,7 +81,7 @@ def add_audio_to_video(video_path, audio_path, output_path):
 
 # Download media using yt-dlp
 def download_media(url, media_type):
-    output_path = os.path.join(ROOTPATH, "tmp", f"anime_media_to_upscale.{'mp4' if media_type == 'Video' else 'png'}")
+    output_path = os.path.join(TEMP_PATH, f"anime_media_to_upscale.{'mp4' if media_type == 'Video' else 'png'}")
     os.makedirs(os.path.dirname(output_path), exist_ok=True)
     
     ydl_opts = {
@@ -207,6 +218,8 @@ def upscale_video(video_path, scale_factor, selected_model, progress=gr.Progress
 
 # Process anime video or image
 def process_media(drive_link, uploaded_file, scale_factor, selected_model, media_type, progress=gr.Progress()):
+    if drive_link:
+        clean_previous_media()  # Clean previous media when a new link is provided
     file_path = None
     if uploaded_file:
         file_path = uploaded_file
@@ -216,7 +229,7 @@ def process_media(drive_link, uploaded_file, scale_factor, selected_model, media
             try:
                 file_id = drive_link.split('/d/')[1].split('/')[0]
                 extension = ".mp4" if media_type == "Video" else ".png"
-                file_path = os.path.join(ROOTPATH, "tmp", f"anime_media_to_upscale{extension}")
+                file_path = os.path.join(TEMP_PATH, f"anime_media_to_upscale{extension}")
                 os.makedirs(os.path.dirname(file_path), exist_ok=True)
                 gdown.download(f"https://drive.google.com/uc?id={file_id}", file_path, quiet=False)
             except Exception as e:
@@ -364,6 +377,7 @@ with gr.Blocks(theme=gr.themes.Soft(), css=custom_css) as demo:
         - Use low-quality anime sources for optimal upscaling.
         - 2x is great for subtle enhancements, 4x maximizes detail.
         - Ensure links from Google Drive, YouTube, Instagram, or TikTok are accessible.
+        - Previous media files are deleted when a new link is provided.
         - Best suited for 2D anime styles with vibrant colors and sharp lines!
         """
     )