Update app.py

app.py

@@ -107,6 +107,9 @@ def parse_pmns(md):
             elif low.startswith("percusion:"):
                 p = line.split(":", 1)[1].strip().lower()
                 sec["percusion"] = p if p in PERCUSSION_STYLES else "ninguno"
+            elif low.startswith("repetir:"):
+                m = re.search(r"(\d+)", line)
+                sec["repeat"] = max(1, min(8, int(m.group(1)))) if m else 1
 
         if not sec["notes"]: errors.append("falta notas")
         for tok in sec["chords"].split():
@@ -215,7 +218,9 @@ def midi_to_mp3(midi_path, mp3_path):
     if os.path.exists(wav_path): os.unlink(wav_path)
     return mp3_path
 
-SYSTEM = """Escribe SOLO el texto de la partitura. Nada de charla. Formato exacto:
+SYSTEM = """Escribe SOLO el texto de la partitura. Nada de charla. Formato exacto.
+
+Para que la canción dure 1 minuto, escribe OBLIGATORIAMENTE 8 secciones (Intro, Verso 1, Pre-Estribillo, Estribillo, Verso 2, Puente, Climax, Outro). Escribe al menos 12 notas por sección. Usa el campo "repetir: 2" en los estribillos.
 
 # Titulo
 ## Tempo
@@ -223,18 +228,24 @@ SYSTEM = """Escribe SOLO el texto de la partitura. Nada de charla. Formato exact
 ## Estructura
 ### Intro
 instrumento: piano
-notas: C4q E4q G4q C5h
-acordes: C4
+notas: C4q E4q G4q C5h E5q D5q C5h
+acordes: C4 G4
 
-### Tema
+### Verso 1
 instrumento: guitar
-notas: A3q C4q E4q A4h G4q F4q E4q
-acordes: Am4 F2
+notas: A3q C4q E4q A4h G4q F4q E4q D4q C4q D4h
+acordes: Am4 F2 C4 G4
 
 ### Estribillo
 instrumento: strings
-notas: F4q G4q A4q C5h
-acordes: F4 C4
+repetir: 2
+notas: F4q G4q A4q C5h D5q C5q A4q G4h F4q E4q D4q C4h
+acordes: F4 C4 G4 Am4
+
+### Puente
+instrumento: flute
+notas: D5q E5q F5q G5h E5q D5q C5q A4q G4q F4q E4q D4h
+acordes: Dm4 G4 C4 Am4
 
 ### Outro
 instrumento: piano
@@ -253,7 +264,7 @@ def clean_score(text):
     if "notas:" not in md.lower(): return None
     return md.strip()
 
-def generate_text(model, tokenizer, messages, max_new=800):
+def generate_text(model, tokenizer, messages, max_new=1500):
     text = tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
     inputs = tokenizer(text, return_tensors="pt").to(model.device)
     with torch.no_grad():
@@ -355,19 +366,19 @@ footer, .built-with, [class*="footer"] { display: none !important; }
 HEADER_HTML = f"""<div id="aether-topbar"><div id="aether-logo">🎼</div><div id="aether-brand-name">{STUDIO_NAME}</div><span id="aether-brand-model">{MODEL_DISPLAY_NAME}</span></div>"""
 SELECT_JS = """<script>function selectSong(id) { const el = document.querySelector('#song-selector input, #song-selector textarea'); if (el) { el.value = id; el.dispatchEvent(new Event('input', { bubbles: true })); el.dispatchEvent(new Event('change', { bubbles: true })); } }</script>"""
 
-@spaces.GPU(duration=60, size="xlarge")
+@spaces.GPU(duration=30, size="xlarge")
 def generate_song(prompt):
     model, tokenizer = load_model()
     model.to("cuda")
     messages = [{"role": "system", "content": SYSTEM}, {"role": "user", "content": f"Compón: {prompt}"}]
-    return generate_text(model, tokenizer, messages, max_new=800)
+    return generate_text(model, tokenizer, messages, max_new=1500)
 
 def process_request(prompt):
     if not prompt or not prompt.strip():
         yield "Escribe qué canción quieres.", "", render_library(), gr.update(), gr.update()
         return
 
-    yield "🎼 *Componiendo...*", "", render_library(), gr.update(), gr.update()
+    yield "🎼 *Componiendo (esto tomará más tiempo para generar 1 minuto de música)...*", "", render_library(), gr.update(), gr.update()
     raw_output = generate_song(prompt)
     md = clean_score(raw_output)