Update app.py

app.py

@@ -166,6 +166,18 @@ def identity_manager (frase, persona):
         clean_output = frase
     return clean_output
 
+def free_narration (srt_final):
+    prompt = f"""Instrucció: Converteix aquesta audiodescripció en una narració lliure breu, natural i coherent.,
+        input: {srt_final}
+        output:
+    """
+    result = generate_advanced(prompt=prompt, system="", max_new_tokens=512, temperature=0.7, top_p=0.95)
+    if "assistant" in result:
+        clean_output = result.split("assistant", 1)[1].strip().split("\n")[0]
+    else:
+        clean_output = frase
+    return clean_output
+
 # ------------------- HTTP (opcional, clientes puros) -------------------
 # Si quieres, puedes añadir un endpoint HTTP POST /generate (FastAPI),
 # pero con Gradio Client es suficiente para engine/local.
@@ -253,4 +265,20 @@ with gr.Blocks(title="Salamandra 7B Instruct · ZeroGPU",css=custom_css,theme=gr
         concurrency_limit=1
     )
 
+    gr.Markdown('<h2 style="text-align:center">Narració lliure</h2>')
+    with gr.Row():
+        with gr.Column(scale=1):
+            srt = gr.Textbox(label="Audiodescripció", value="(AD)\nTOTS CANTANT: avui celebrem la nostra festa major\nAINA: som hi tots a ballar", lines=3)
+            btn_modificar = gr.Button("Generar audiodescripció", variant="primary")
+        with gr.Column(scale=1):
+            narració_lliure = gr.Textbox(label="Narració lliure", lines=18)
+
+    btn_modificar.click(
+        free_narration,
+        inputs=[srt],
+        outputs=narració_lliure,
+        api_name="narració",
+        concurrency_limit=1
+    )
+
 demo.queue(max_size=16).launch()