feat: implement mp3 quality selection and fix uuid routes

app.py

@@ -20,6 +20,8 @@ def run_download(job_id, url, format_choice, format_id):
     cmd = ["yt-dlp", "--no-playlist", "-o", out_template]
 
     if format_choice == "audio":
+        if format_id:
+            cmd += ["-f", format_id]
         cmd += ["-x", "--audio-format", "mp3"]
     elif format_id:
         cmd += ["-f", f"{format_id}+bestaudio/best", "--merge-output-format", "mp4"]
@@ -95,7 +97,16 @@ def get_info():
 
         # Build quality options — keep best format per resolution
         best_by_height = {}
+        audio_by_abr = {}
         for f in info.get("formats", []):
+            if f.get("vcodec") == "none":
+                abr = f.get("abr")
+                if abr:
+                    abr = int(abr)
+                    if abr not in audio_by_abr or (f.get("asr", 0) > audio_by_abr[abr].get("asr", 0)):
+                        audio_by_abr[abr] = f
+                continue
+
             height = f.get("height")
             if height and f.get("vcodec", "none") != "none":
                 tbr = f.get("tbr") or 0
@@ -111,12 +122,22 @@ def get_info():
             })
         formats.sort(key=lambda x: x["height"], reverse=True)
 
+        audio_formats = []
+        for abr, f in audio_by_abr.items():
+            audio_formats.append({
+                "id": f["format_id"],
+                "label": f"{abr}kbps",
+                "abr": abr,
+            })
+        audio_formats.sort(key=lambda x: x["abr"], reverse=True)
+
         return jsonify({
             "title": info.get("title", ""),
             "thumbnail": info.get("thumbnail", ""),
             "duration": info.get("duration"),
             "uploader": info.get("uploader", ""),
             "formats": formats,
+            "audioFormats": audio_formats,
         })
     except subprocess.TimeoutExpired:
         return jsonify({"error": "Timed out fetching video info"}), 400
@@ -158,7 +179,8 @@ def check_status(job_id):
 
 
 @app.route("/api/file/<job_id>")
-def download_file(job_id):
+@app.route("/api/file/<job_id>/<path:filename>")
+def download_file(job_id, filename=None):
     job = jobs.get(job_id)
     if not job or job["status"] != "done":
         return jsonify({"error": "File not ready"}), 404

templates/index.html

@@ -593,6 +593,8 @@
               uploader: data.uploader || '',
               formats: data.formats || [],
               selectedFormatId: data.formats?.[0]?.id || null,
+              audioFormats: data.audioFormats || [],
+              selectedAudioFormatId: data.audioFormats?.[0]?.id || null,
             };
           }
         } catch (err) {
@@ -664,7 +666,11 @@
       }
 
       let qualityChips = '';
-      if (!isAudio && c.formats && c.formats.length > 1) {
+      if (isAudio && c.audioFormats && c.audioFormats.length > 1) {
+        qualityChips = c.audioFormats.map(f =>
+          `<button class="q-chip${f.id === c.selectedAudioFormatId ? ' active' : ''}" onclick="pickFormat(${idx}, '${f.id}')">${f.label}</button>`
+        ).join('');
+      } else if (!isAudio && c.formats && c.formats.length > 1) {
         qualityChips = c.formats.map(f =>
           `<button class="q-chip${f.id === c.selectedFormatId ? ' active' : ''}" onclick="pickFormat(${idx}, '${f.id}')">${f.label}</button>`
         ).join('');
@@ -705,7 +711,11 @@
     }
 
     function pickFormat(idx, formatId) {
-      cardData[idx].selectedFormatId = formatId;
+      if (currentFormat === 'audio') {
+        cardData[idx].selectedAudioFormatId = formatId;
+      } else {
+        cardData[idx].selectedFormatId = formatId;
+      }
       renderCard(idx);
     }
 
@@ -716,13 +726,14 @@
       renderCard(idx);
 
       try {
+        const payloadFormatId = currentFormat === 'audio' ? c.selectedAudioFormatId : c.selectedFormatId;
         const res = await fetch('/api/download', {
           method: 'POST',
           headers: { 'Content-Type': 'application/json' },
           body: JSON.stringify({
             url: c.url,
             format: currentFormat,
-            format_id: c.selectedFormatId,
+            format_id: payloadFormatId,
             title: c.title || '',
           }),
         });
@@ -773,7 +784,7 @@
       const c = cardData[idx];
       if (!c.jobId) return;
       const a = document.createElement('a');
-      a.href = `/api/file/${c.jobId}`;
+      a.href = `/api/file/${c.jobId}/${encodeURIComponent(c.filename || 'download.mp4')}`;
       a.download = c.filename || '';
       a.click();
     }