Create main.py

main.py

@@ -0,0 +1,121 @@
+import os, json, glob, shutil, subprocess, tempfile, time, re, requests
+from pathlib import Path
+import gradio as gr
+from ytmusicapi import YTMusic
+import spotipy
+from spotipy.oauth2 import SpotifyClientCredentials
+from mutagen.id3 import ID3, TIT2, TPE1, TALB, TCON, TYER, TRCK, TPE2, USLT, APIC, COMM
+from mutagen.mp3 import MP3
+
+# ------------- CONFIG -------------
+# Secrets must be set in HF Space → Settings → Secrets
+SPOTIFY_CLIENT_ID = os.getenv("SPOTIFY_CLIENT_ID")
+SPOTIFY_CLIENT_SECRET = os.getenv("SPOTIFY_CLIENT_SECRET")
+# ----------------------------------
+
+TMP = Path("/tmp")              # HF allows full r/w here
+STATIC = Path(__file__).parent  # read-only, keep code / assets only
+
+ytm = YTMusic()
+sp = None
+if SPOTIFY_CLIENT_ID and SPOTIFY_CLIENT_SECRET:
+    sp = spotipy.Spotify(
+        auth_manager=SpotifyClientCredentials(
+            client_id=SPOTIFY_CLIENT_ID,
+            client_secret=SPOTIFY_CLIENT_SECRET,
+        )
+    )
+
+# ------------- HELPERS -------------
+def safe(s: str) -> str:
+    return re.sub(r'[\\/:"*?<>|]+', "", s).strip()
+
+def ytdlp_fast(url: str, out: str) -> None:
+    """Download best-audio → 320 k MP3 using aria2c."""
+    subprocess.run(
+        [
+            "yt-dlp",
+            "-f", "bestaudio/best",
+            "--extract-audio",
+            "--audio-format", "mp3",
+            "--audio-quality", "320K",
+            "--concurrent-fragments", "16",
+            "--external-downloader", "aria2c",
+            "--external-downloader-args",
+            "aria2c:-x16 -s16 -j16 -k1M --file-allocation=none",
+            "--fragment-retries", "infinite",
+            "--skip-unavailable-fragments",
+            "--no-part",
+            "--no-cache-dir",
+            "--force-ipv4",
+            "--sponsorblock-remove", "all",
+            "-o", out,
+            url,
+        ],
+        check=True,
+    )
+
+def write_tags(
+    fn: str,
+    title: str,
+    artist: str,
+    album: str,
+    year: str,
+    cover: str,
+    lyrics: str,
+    genre: str = "",
+    track: str = "",
+    album_artist: str = "",
+    description: str = "",
+):
+    audio = MP3(fn, ID3=ID3)
+    if audio.tags is None:
+        audio.add_tags()
+    tags = audio.tags
+    tags.add(TIT2(encoding=3, text=title))
+    tags.add(TPE1(encoding=3, text=artist))
+    tags.add(TALB(encoding=3, text=album))
+    if genre:
+        tags.add(TCON(encoding=3, text=genre))
+    if year:
+        tags.add(TYER(encoding=3, text=year))
+    if track:
+        tags.add(TRCK(encoding=3, text=track))
+    if album_artist:
+        tags.add(TPE2(encoding=3, text=album_artist))
+    if lyrics:
+        tags.add(USLT(encoding=3, lang="eng", desc="", text=lyrics))
+    if description:
+        tags.add(COMM(encoding=3, lang="eng", desc="desc", text=description))
+    if cover:
+        try:
+            img = requests.get(cover, timeout=15).content
+            tags.add(
+                APIC(
+                    encoding=3,
+                    mime="image/jpeg",
+                    type=3,
+                    desc="Cover",
+                    data=img,
+                )
+            )
+        except Exception:
+            pass
+    audio.save(v2_version=3)
+
+# ------------- CORE -------------
+def download_spotify(url: str) -> Path | None:
+    if not sp:
+        raise RuntimeError("Spotipy credentials not set")
+    track = sp.track(url)
+    title = track["name"]
+    artist = ", ".join(a["name"] for a in track["artists"])
+    album = track["album"]["name"]
+    date = track["album"].get("release_date", "")
+    year = date.split("-")[0] if date else ""
+    cover = track["album"]["images"][0]["url"]
+    query = f"{title} {artist}"
+
+    # search YTM
+    res = ytm.search(query, filter="songs", limit
+