Update app.py

app.py

@@ -329,6 +329,7 @@ button:hover {
 
 import os
 import yt_dlp
+import gdown
 from scipy.io import wavfile
 
 def download_audio(url, cookie_file=None):
@@ -355,6 +356,9 @@ def download_audio(url, cookie_file=None):
     # Check if it's a Google Drive link
     if 'drive.google.com' in url or 'https://drive.google.com' in url:
         try:
+            # Create the 'ytdl' directory if it doesn't exist
+            os.makedirs('ytdl', exist_ok=True)
+            
             # Extract file ID from the URL
             file_id = url.split('/d/')[1].split('/')[0]
             download_url = f'https://drive.google.com/uc?id={file_id}'
@@ -370,6 +374,9 @@ def download_audio(url, cookie_file=None):
 
     # Handle YouTube link
     else:
+        # Create the 'ytdl' directory for YouTube downloads as well
+        os.makedirs('ytdl', exist_ok=True)
+        
         with yt_dlp.YoutubeDL(ydl_opts) as ydl:
             try:
                 info_dict = ydl.extract_info(url, download=True)
@@ -389,7 +396,7 @@ def download_audio(url, cookie_file=None):
                     return None, "Download failed: HTTP Error 403. This format requires a GVS PO Token, or the cookies are invalid. Please upload fresh cookies. See https://github.com/yt-dlp/yt-dlp/wiki/PO-Token-Guide for advanced troubleshooting.", None
                 return None, f"Download failed: {str(e)}", None
             except Exception as e:
-                return None, f"Unexpected error: {str(e)}", None 
+                return None, f"Unexpected error: {str(e)}", None
                 
 @spaces.GPU
 def roformer_separator(audio, model_key, seg_size, override_seg_size, overlap, pitch_shift, model_dir, output_dir, out_format, norm_thresh, amp_thresh, batch_size, exclude_stems="", progress=gr.Progress(track_tqdm=True)):