Add ACE-Step 1.5 and YuE on Music, and put Video/Image/Music cards back.

Recover finished Comfy audio when queueing hangs so a completed track is not stuck on Queueing.

Co-authored-by: Cursor <cursoragent@cursor.com>

server/api/generate/recover.post.ts

@@ -16,11 +16,78 @@ function recentLibraryClip(event: Parameters<typeof assertLibraryOwner>[0]) {
   }
 }
 
+function recentLibraryTrack(event: Parameters<typeof assertLibraryOwner>[0]) {
+  const { catalog } = assertLibraryOwner(event)
+  const newest = catalog.tracks?.[0]
+  if (!newest || Date.now() - newest.createdAt > RECENT_CLIP_MS) return null
+  return {
+    type: 'complete' as const,
+    status: 'complete' as const,
+    message: 'Track ready',
+    progress: 100,
+    trackId: newest.id,
+    filename: newest.comfyFilename,
+    folderLocked: false
+  }
+}
+
+async function recoverFinishedMedia(event: Parameters<typeof assertLibraryOwner>[0], opts?: { tags?: string }) {
+  const { owner } = assertLibraryOwner(event)
+  const library = publicLibrary(event)
+  const folderId = library.folders.find(folder => folder.unlocked)?.id
+  const tracks = folderId ? await importMissingComfyAudio(owner, folderId) : []
+  const want = String(opts?.tags || '').trim()
+  const { catalog } = assertLibraryOwner(event)
+  const match = [...tracks].reverse().find((track) => {
+    if (!want) return true
+    const tags = String(track.tags || '').trim()
+    return tags === want || tags.includes(want) || want.includes(tags)
+  }) || (want ? catalog.tracks.find((track) => {
+    const tags = String(track.tags || '').trim()
+    return Date.now() - track.createdAt < RECENT_CLIP_MS && (tags === want || tags.includes(want) || want.includes(tags))
+  }) : null)
+  if (match) {
+    return {
+      type: 'complete' as const,
+      status: 'complete' as const,
+      message: 'Recovered track from ComfyUI',
+      progress: 100,
+      trackId: match.id,
+      filename: match.comfyFilename,
+      folderLocked: false
+    }
+  }
+  if (opts?.tags) return null
+  const videos = folderId ? await importMissingComfyVideos(owner, folderId) : []
+  const newestVideo = videos[videos.length - 1]
+  if (newestVideo) {
+    return {
+      type: 'complete' as const,
+      status: 'complete' as const,
+      message: 'Recovered video from ComfyUI',
+      progress: 100,
+      clipId: newestVideo.id,
+      filename: newestVideo.comfyFilename,
+      hideThumbnail: newestVideo.hideThumbnail
+    }
+  }
+  return recentLibraryTrack(event) || recentLibraryClip(event)
+}
+
 export default defineEventHandler(async (event) => {
   const body = await readBody<{ jobId?: string }>(event).catch(() => ({} as { jobId?: string }))
   const jobId = String(body?.jobId || '')
   const live = jobId ? getJob(jobId) : undefined
-  if (live) return jobSnapshot(live)
+  if (live) {
+    if (live.status === 'complete' || live.status === 'error' || live.trackId || live.clipId) {
+      return jobSnapshot(live)
+    }
+    if (live.kind === 'music' && Date.now() - live.startedAt > 15_000) {
+      const recovered = await recoverFinishedMedia(event, { tags: live.library?.tags })
+      if (recovered?.trackId) return recovered
+    }
+    return jobSnapshot(live)
+  }
 
   if (jobId) {
     const pending = readPendingJob(jobId)
@@ -29,28 +96,12 @@ export default defineEventHandler(async (event) => {
       if (done) return done
       return jobSnapshot(ensurePendingWatch(pending))
     }
-    const saved = recentLibraryClip(event)
-    if (saved) return saved
+    const recovered = await recoverFinishedMedia(event)
+    if (recovered) return recovered
     throw createError({ statusCode: 404, statusMessage: 'Job not found' })
   }
 
-  const { owner } = assertLibraryOwner(event)
-  const library = publicLibrary(event)
-  const folderId = library.folders.find(folder => folder.unlocked)?.id
-  const imported = folderId ? await importMissingComfyVideos(owner, folderId) : []
-  const newest = imported[imported.length - 1]
-  if (newest) {
-    return {
-      type: 'complete',
-      status: 'complete',
-      message: 'Recovered video from ComfyUI',
-      progress: 100,
-      clipId: newest.id,
-      filename: newest.comfyFilename,
-      hideThumbnail: newest.hideThumbnail
-    }
-  }
-  const saved = recentLibraryClip(event)
-  if (saved) return saved
-  throw createError({ statusCode: 404, statusMessage: 'No finished ComfyUI video to recover' })
+  const recovered = await recoverFinishedMedia(event)
+  if (recovered) return recovered
+  throw createError({ statusCode: 404, statusMessage: 'No finished ComfyUI output to recover' })
 })