Add Chrome cookies.txt upload and yt_dlp refactor

Enable uploading Chrome cookies.txt in the Gradio UI for improved YouTube access. Refactor yt_audio_get_tracks.py to always use yt_dlp, preferring cookies.txt if available and falling back to browser cookies. Remove Piped API fallback and update progress handling.

app.py

@@ -182,6 +182,7 @@ def process_video(video_id: str, progress=gr.Progress(track_tqdm=True)) -> str:
 def process_video_with_progress(
     video_id: str,
     uploaded_audio: str | None,
+    cookies_upload: str | None,
     progress=gr.Progress(track_tqdm=True),
 ):
     status_lines = []
@@ -190,6 +191,8 @@ def process_video_with_progress(
         status_lines.append(message)
 
     try:
+        if cookies_upload is not None:            
+            shutil.copy(cookies_upload, "modules/cookies.txt")
         if uploaded_audio:
             progress(0.05, desc="Preparing uploaded audio")
             yield "", "Preparing uploaded audio..."
@@ -252,6 +255,19 @@ with gr.Blocks(title="SeparateTracks") as demo:
             scale=4,
         )
         run_btn = gr.Button("Separate Tracks", variant="primary", scale=1)
+    with gr.Row():
+        cookies_upload = gr.File(
+            label="Upload Chrome cookies.txt (Netscape format)",
+            file_types=[".txt"],
+            type="filepath",
+        )
+        gr.Markdown("""
+            **How to get cookies.txt:**
+            1. Install [Get cookies.txt LOCALLY](https://chromewebstore.google.com/detail/get-cookiestxt-locally/cclelndahbckbenkjhflpdbgdldlbecc)
+            2. Log into YouTube in Chrome
+            3. Click extension → Export cookies
+            4. Upload the file here
+            """)
 
     upload_input = gr.File(
         label="Audio File Override (.wav or .mp3)",
@@ -265,7 +281,7 @@ with gr.Blocks(title="SeparateTracks") as demo:
 
     run_btn.click(
         fn=process_video_with_progress,
-        inputs=[video_id_input, upload_input],
+        inputs=[video_id_input, upload_input, cookies_upload],
         outputs=[audio_output, progress_output],
     )
 

modules/cookies.txt

@@ -7,9 +7,10 @@
 .youtube.com	TRUE	/	TRUE	1809665126	__Secure-1PSIDTS	sidts-CjQBhkeRd2zWGbcDaL6SNNxU-DnNCLJWYODQ2KzKAcsoR_FM29OpDU_NudXF0BGvxu3-4ilCEAA
 .youtube.com	TRUE	/	TRUE	1809665126	__Secure-3PSIDTS	sidts-CjQBhkeRd2zWGbcDaL6SNNxU-DnNCLJWYODQ2KzKAcsoR_FM29OpDU_NudXF0BGvxu3-4ilCEAA
 .youtube.com	TRUE	/	TRUE	1809665900	__Secure-3PSIDCC	AKEyXzWiyEqsbs6vWksfuhqkVix8gfb29gWxUBcn-A_zxFpN_jSGtqbMuekeAysH5txR5v_tqg
-.youtube.com	TRUE	/	TRUE	1793686088	VISITOR_INFO1_LIVE	0HOxtcrbVL8
-.youtube.com	TRUE	/	TRUE	1793686088	VISITOR_PRIVACY_METADATA	CgJVUxIEGgAgaw%3D%3D
+.youtube.com	TRUE	/	TRUE	1793688861	VISITOR_INFO1_LIVE	0HOxtcrbVL8
+.youtube.com	TRUE	/	TRUE	1793688861	VISITOR_PRIVACY_METADATA	CgJVUxIEGgAgaw%3D%3D
 .youtube.com	TRUE	/	TRUE	1793681123	__Secure-ROLLOUT_TOKEN	CObY3I_n2pG-tQEQ1bTnseyZkwMYx9-alq-mlAM%3D
 .youtube.com	TRUE	/	TRUE	1793681123	__Secure-YNID	18.YT=Tx0A40MPkqhXx1ssHRZzuknGzSYMu4HquWBxKZVhu_0va9vVL6-pj-hgIxnNj494GkolBTXqTIRUTUC6fOJCobVAekeq8te87cIsbGIO2g_jPtJnm9zE0kpjWOKceLvXzON4m1GZqXSjj5c-IjmVR2V8_43Iqnn5sEdQDd9cDor11A7gsHRNqJVIsOMuKB2n1Np9oHkp3xMIZWg38MUB13NSur0Rr3SgGIvHZBEZ1z_MTrqCYndzX5CPDbSdhT1QfUR6lC85Y1o29OlpOE3hsSUq8xROpXOX016gCc8PsSbpu54DQOw2FHydbuW4FcWEtODp1vJ3ob5QbdAN6oYeeQ
 .youtube.com	TRUE	/	TRUE	0	SOCS	CAI
-.youtube.com	TRUE	/	TRUE	1778135888	GPS	1
+.youtube.com	TRUE	/	TRUE	1778138661	GPS	1
+.youtube.com	TRUE	/	TRUE	0	YSC	nu8y7GrJi5c

modules/yt_audio_get_tracks.py

@@ -13,86 +13,41 @@ def _emit_progress(progress_callback, message):
 cookie_path = os.path.join(os.path.dirname(__file__), 'cookies.txt')
 
 def download_audio(url, video_id, progress_callback=None):
-    try:
-        temp_dir = 'separated'
-        os.makedirs(temp_dir, exist_ok=True)
-        _emit_progress(progress_callback, 'Downloading audio from YouTube...')
-        compat_opts = ['no-youtube-unavailable-videoplayback']
-
-        is_hf = bool(os.getenv("SPACE_ID") or os.getenv("HF_SPACE") or os.path.exists("/.dockerenv"))
-
-        ydl_opts = {
-            'format': 'bestaudio/best',
-            'outtmpl': os.path.join(temp_dir, f'{video_id}.%(ext)s'),
-            'postprocessors': [{'key': 'FFmpegExtractAudio', 'preferredcodec': 'wav'}],
-            'keepvideo': True,
-            'quiet': False,
-            'no_warnings': False,
-            'user_agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36',
-            'http_headers': {'Referer': 'https://www.youtube.com/'},
-            'socket_timeout': 30,
-            'retry_sleep': [1, 2, 5, 10],
-            'max_retries': 3,
-            'compat_opts': compat_opts,
-        }
-
-        if is_hf or os.path.exists(cookie_path):
-            ydl_opts['cookiefile'] = cookie_path
-            ydl_opts['nocheckcertificate'] = True
-        else:
-            ydl_opts['cookiesfrombrowser'] = ('chrome', None, None, None)
-
-        with yt_dlp.YoutubeDL(ydl_opts) as ydl:
-            ydl.download([url])
-        _emit_progress(progress_callback, 'Converting downloaded audio to WAV...')
-        return os.path.join(temp_dir, f'{video_id}.wav')
-    except Exception:
-        return download_audio_piped(url, video_id, progress_callback)
-
-def download_audio_piped(url, video_id, progress_callback=None):
     temp_dir = 'separated'
     os.makedirs(temp_dir, exist_ok=True)
     _emit_progress(progress_callback, 'Downloading audio from YouTube...')
-
-    # Try multiple Piped instances
-    instances = [
-        "https://pipedapi.tokhmi.xyz",
-        "https://piped-api.lunar.icu",
-        "https://pipedapi.moomoo.me",
-        "https://pipedapi.leptons.xyz"
-    ]
-
-    for piped_api in instances:
-        try:
-            resp = requests.get(f"{piped_api}/streams/{video_id}", timeout=20)
-            resp.raise_for_status()
-            data = resp.json()
-            break
-        except Exception:
-            continue
+    compat_opts = ['no-youtube-unavailable-videoplayback']
+    # if shutil.which('deno') is None:
+    #     print("⚠️ Deno not found.")
+    #     compat_opts.append('no-youtube-js')
+
+    is_hf = bool(os.getenv("SPACE_ID") or os.getenv("HF_SPACE") or os.path.exists("/.dockerenv"))
+
+    ydl_opts = {
+        'format': 'bestaudio/best',
+        'outtmpl': os.path.join(temp_dir, f'{video_id}.%(ext)s'),
+        'postprocessors': [{'key': 'FFmpegExtractAudio', 'preferredcodec': 'wav'}],
+        'keepvideo': True,
+        'quiet': False,
+        'no_warnings': False,
+        'user_agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36',
+        'http_headers': {'Referer': 'https://www.youtube.com/'},
+        'socket_timeout': 30,
+        'retry_sleep': [1, 2, 5, 10],
+        'max_retries': 3,
+        'compat_opts': compat_opts,
+    }
+
+    if is_hf or os.path.exists(cookie_path):
+        ydl_opts['cookiefile'] = cookie_path
+        ydl_opts['nocheckcertificate'] = True
     else:
-        raise Exception("All Piped instances failed")
-
-    # Best audio
-    audio_streams = [s for s in data.get("audioStreams", []) if not s.get("videoOnly")]
-    if not audio_streams:
-        raise Exception("No audio stream")
-    best = max(audio_streams, key=lambda x: int(x.get("bitrate", 0)))
-
-    # Download
-    m4a_path = os.path.join(temp_dir, f'{video_id}.m4a')
-    with requests.get(best["url"], stream=True, timeout=60) as r:
-        r.raise_for_status()
-        with open(m4a_path, 'wb') as f:
-            for chunk in r.iter_content(8192):
-                f.write(chunk)
+        ydl_opts['cookiesfrombrowser'] = ('chrome', None, None, None)
 
-    # Convert
-    _emit_progress(progress_callback, 'Converting to WAV...')
-    wav_path = os.path.join(temp_dir, f'{video_id}.wav')
-    subprocess.run(['ffmpeg', '-y', '-i', m4a_path, '-acodec', 'pcm_s16le', '-ar', '44100', wav_path],
-                    stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL, check=True)
-    return wav_path
+    with yt_dlp.YoutubeDL(ydl_opts) as ydl:
+        ydl.download([url])
+    _emit_progress(progress_callback, 'Converting downloaded audio to WAV...')
+    return os.path.join(temp_dir, f'{video_id}.wav')
 
 def separate_tracks(input_wav, video_id, progress_callback=None):
     if not os.path.exists(input_wav):