outro check looks at song ending only

app.py

@@ -158,7 +158,9 @@ def generate(style, lyrics, cot, seed, cfg_scale, duration, ending):
     if not exact:
         # Natural: let the model finish on its own; close with a real [Outro]
         # tag (official YuE2 vocabulary) so it composes a proper ending.
-        if not re.search(r"\[(outro|end|ending)\]", lyrics or "", re.I):
+        # Only the song's ENDING counts — a mid-song [outro] doesn't fool us.
+        tail = "\n".join([ln for ln in (lyrics or "").strip().splitlines() if ln.strip()][-3:])
+        if not re.search(r"\[(outro|end|ending)\]", tail, re.I):
             lyrics = (lyrics or "").rstrip() + "\n[Outro]"
     gen_kwargs = dict(style=style, lyrics=lyrics, cot=cot, seed=seed)
     if cfg_scale and float(cfg_scale) != 1.0: