From b32871cf2818505e7ab75490478ff8936d8d1e55 Mon Sep 17 00:00:00 2001
From: Broque Thomas <batmanesignal@gmail.com>
Date: Tue, 6 Jan 2026 14:27:14 -0800
Subject: [PATCH] Improve album classification and modal data handling

Refines album/single classification in get_wishlist_stats by considering explicit 'album' type for short multi-track releases. Also updates openDownloadModalForRecentAlbum to include total_tracks and album_type in album data for more accurate processing.
---
 web_server.py          | 3 +++
 webui/static/script.js | 4 +++-
 2 files changed, 6 insertions(+), 1 deletion(-)

diff --git a/web_server.py b/web_server.py
index 0b20301e..2e8f7a73 100644
--- a/web_server.py
+++ b/web_server.py
@@ -9331,6 +9331,9 @@ def get_wishlist_stats():
                 # Use track count (most accurate)
                 if total_tracks >= 6:
                     albums_count += 1
+                elif album_type == 'album' and total_tracks > 1:
+                    # Respect explicit 'album' type for multi-track releases even if short
+                    albums_count += 1
                 else:
                     singles_count += 1
             else:
diff --git a/webui/static/script.js b/webui/static/script.js
index 29a22e91..cb07d0ae 100644
--- a/webui/static/script.js
+++ b/webui/static/script.js
@@ -32853,7 +32853,9 @@ async function openDownloadModalForRecentAlbum(albumIndex) {
                 artists: artists,
                 album: {
                     name: album.album_name,
-                    images: album.album_cover_url ? [{ url: album.album_cover_url }] : []
+                    images: album.album_cover_url ? [{ url: album.album_cover_url }] : [],
+                    total_tracks: albumData.total_tracks || albumData.tracks.length,
+                    album_type: albumData.album_type || 'album'
                 },
                 duration_ms: track.duration_ms || 0
             };