Replace Comfy YuE with standalone YuEGP profile 1

server/api/generate/music.post.ts

@@ -1,3 +1,4 @@
+import { yueGpConfigured } from '~/server/utils/yueGp'
 import { addStudioJob, kickStudioQueue, listStudioJobs, videoJobsBusy } from '~/server/utils/studioQueue'
 import { comfyConfigured } from '~/server/utils/comfy'
 import {
@@ -28,12 +29,15 @@ export default defineEventHandler(async (event) => {
     instrumental?: unknown
     folderId?: unknown
     name?: unknown
+    yueProfile?: unknown
     engine?: unknown
     sampler_name?: unknown
     scheduler?: unknown
   }>(event).catch(() => ({} as Record<string, unknown>))
 
-  if (!comfyConfigured()) {
+  const engine = parseMusicEngine(body.engine)
+  if (engine === 'yue' && !yueGpConfigured()) throw createError({ statusCode: 503, statusMessage: 'YuEGP host is not configured. Set COMFY_CONTROL_URL.' })
+  if (engine !== 'yue' && !comfyConfigured()) {
     throw createError({
       statusCode: 503,
       statusMessage: 'Beast ComfyUI is not configured. Set COMFY_HOST.'
@@ -62,12 +66,15 @@ export default defineEventHandler(async (event) => {
   const destFolder = library.folders.find(folder => folder.id === folderId)
   const folderLocked = Boolean(destFolder?.protected && !destFolder.unlocked)
 
-  const engine = parseMusicEngine(body.engine)
+  if (engine === 'yue' && instrumental) throw createError({ statusCode: 400, statusMessage: 'YuEGP currently requires lyrics. Use ACE for instrumental music.' })
+  const yueProfile = body.yueProfile ?? 1
+  if (engine === 'yue' && yueProfile !== 1 && yueProfile !== 3) throw createError({ statusCode: 400, statusMessage: 'Choose YuEGP profile 1 or manual fallback 3.' })
   if (engine === 'yue' && !instrumental) {
     const problem = yueLyricsProblem(lyrics)
     if (problem) throw createError({ statusCode: 400, statusMessage: problem })
   }
   const duration = clampMusicDuration(body.duration)
+  if (engine === 'yue' && duration > 150) throw createError({ statusCode: 400, statusMessage: 'YuEGP supports up to 150 seconds per section.' })
   const steps = clampMusicSteps(
     body.steps,
     engine === 'ace-step-1.5' ? MUSIC_STEPS_DEFAULT_15 : MUSIC_STEPS_DEFAULT
@@ -115,7 +122,8 @@ export default defineEventHandler(async (event) => {
       lyrics,
       instrumental,
       lyricsStrength,
-      musicEngine: engine
+      musicEngine: engine,
+      ...(engine === 'yue' ? { yueProfile: yueProfile as 1 | 3 } : {})
     }
   })
   await kickStudioQueue()

server/api/generate/recover.post.ts

@@ -82,6 +82,7 @@ export default defineEventHandler(async (event) => {
     if (live.status === 'complete' || live.status === 'error' || live.trackId || live.clipId) {
       return jobSnapshot(live)
     }
+    if (live.yueGp) 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