Update index.html

index.html

@@ -729,8 +729,7 @@
                 
                 const finalLyrics = isInstrumental ? "" : lyrics;
 
-                // ** LOGIC CORRECTION: Determine Generation Mode **
-                // If audio is uploaded, we must switch the task from "text2music" to "prompt" (or equivalent for conditioning)
+                // ** LOGIC FIX: Switch task mode based on audio input **
                 const taskType = uploadedAudioObj ? "prompt" : "text2music";
 
                 const payload = [
@@ -777,6 +776,7 @@
             let hasResult = false;
             
             function addAudio(url) {
+                if (!url) return;
                 const fullUrl = url.startsWith('http') ? url : ACE_SPACE_URL.replace(/\/$/, '') + url;
                 if (processedUrls.has(fullUrl)) return;
                 processedUrls.add(fullUrl);
@@ -790,14 +790,30 @@
                 playerWrapper.innerHTML += `<div class="audio-item"><audio controls autoplay src="${fullUrl}"></audio></div>`;
             }
 
+            // Enhanced Traversal to find nested audio files
             function traverse(obj) {
-                if (typeof obj === 'string' && obj.includes('/file=') && obj.endsWith('.mp3')) addAudio(obj);
-                else if (obj && typeof obj === 'object') {
-                    if (obj.url && obj.url.endsWith('.mp3')) addAudio(obj.url);
+                if (typeof obj === 'string') {
+                    if (obj.includes('/file=') && (obj.endsWith('.mp3') || obj.endsWith('.wav'))) {
+                        addAudio(obj);
+                    }
+                } else if (obj && typeof obj === 'object') {
+                    if (obj.url && (obj.url.endsWith('.mp3') || obj.url.endsWith('.wav'))) {
+                        addAudio(obj.url);
+                    }
+                    if (obj.path && (obj.path.endsWith('.mp3') || obj.path.endsWith('.wav'))) {
+                        // Sometimes 'path' is local, need to prepend gradio_api/file=
+                        addAudio(`/gradio_api/file=${obj.path}`);
+                    }
                     Object.values(obj).forEach(traverse);
                 }
             }
-            traverse(data);
+
+            // Check data structure carefully
+            if (data && data.data) {
+                data.data.forEach(item => traverse(item));
+            } else {
+                traverse(data);
+            }
 
             if (hasResult) {
                 const headerText = document.getElementById('resultHeaderText');
@@ -807,7 +823,7 @@
                 finalLyricsBox.innerHTML = formatLyrics(lyrics);
                 window.scrollTo({ top: 0, behavior: 'smooth' });
             } else {
-                alert("فایل صوتی یافت نشد!");
+                alert("فایل صوتی یافت نشد! (خروجی سرور خالی بود)");
                 step1.style.display = 'block';
                 historySection.style.display = 'block';
             }