Update app.py

app.py

@@ -322,44 +322,40 @@ button:hover {
 }
 """
 
-def download_audio(url, cookies_file=None, out_dir="ytdl"):
-    """Download audio from a URL using yt-dlp with cookies support and format debugging."""
-    if not url or not validators.url(url):
-        raise ValueError("Invalid or missing URL.")
-    
-    if os.path.exists(out_dir):
-        shutil.rmtree(out_dir)
-    os.makedirs(out_dir, exist_ok=True)
-    
+import yt_dlp
+
+def download_audio(url, download_type, cookie_file):
     ydl_opts = {
-        'format': 'bestaudio/best/best',
-        'postprocessors': [{'key': 'FFmpegExtractAudio', 'preferredcodec': 'wav', 'preferredquality': '192'}],
-        'outtmpl': os.path.join(out_dir, '%(title)s.%(ext)s'),
+        'format': 'bestaudio/best',
+        'postprocessors': [{
+            'key': 'FFmpegExtractAudio',
+            'preferredcodec': 'wav',
+            'preferredquality': '192',
+        }],
+        'outtmpl': 'ytdl/%(title)s.%(ext)s',
         'user_agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.124 Safari/537.36',
         'geo_bypass': True,
         'force_ipv4': True,
         'referer': 'https://www.youtube.com/',
         'noplaylist': True,
-        'cookiefile': cookies_file if cookies_file else None,
+        'cookiefile': cookie_file.name if cookie_file else None,
         'extractor_retries': 5,
         'ignoreerrors': False,
         'no_check_certificate': True,
         'verbose': True,
     }
-    try:
-        with YoutubeDL(ydl_opts) as ydl:
-            try:
-                info_dict = ydl.extract_info(url, download=True)
-                return ydl.prepare_filename(info_dict).rsplit('.', 1)[0] + '.wav'
-            except yt_dlp.utils.DownloadError as e:
-                logger.error(f"Download failed, checking available formats: {e}")
-                formats_info = ydl.extract_info(url, download=False)
-                if formats_info:
-                    logger.info(f"Available formats: {formats_info.get('formats', [])}")
-                raise
-    except Exception as e:
-        logger.error(f"Download failed: {e}")
-        raise RuntimeError(f"Download failed: {e}")
+
+    with yt_dlp.YoutubeDL(ydl_opts) as ydl:
+        try:
+            info_dict = ydl.extract_info(url, download=True)
+            file_path = ydl.prepare_filename(info_dict)
+            return file_path, "Download successful", None, None, None, None
+        except yt_dlp.utils.ExtractorError as e:
+            if "Sign in to confirm you’re not a bot" in str(e):
+                return None, "Authentication failed. Please upload updated cookies from a logged-in browser session.", None, None, None, None
+            return None, f"Download failed: {str(e)}", None, None, None, None
+        except Exception as e:
+            return None, f"Unexpected error: {str(e)}", None, None, None, 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)):