Update app.py

app.py

@@ -144,6 +144,28 @@ def salamandra_chat_endpoint(prompt: str) -> Dict[str, Any]:
 
     return {"text": text}
 
+def resumir_frases(frase, num_palabras):
+    num_palabras = int(num_palabras)
+    prompt = f"Instrució: Resumeix la següent frase en {num_palabras} paraules. Input: {frase}"
+    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
+
+def identity_manager (frase, persona):
+    prompt = f"""Instrucció: Substitueix el subjecte de la frase per la persona indicada, mantenint la resta igual.
+        Frase: {frase}
+        Substitució: {persona}
+        Resposta:"""
+    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.
@@ -169,6 +191,7 @@ with gr.Blocks(title="Salamandra 7B Instruct · ZeroGPU") as demo:
     out_engine = gr.Textbox(label="Respuesta (ENGINE)")
     gr.Button("Probar /predict").click(predict_for_engine, [in_prompt_engine], out_engine, api_name="predict", concurrency_limit=1)
 
+    gr.Markdown('<h2 style="text-align:center">Sortida del model Salamandra a partir d'una petició</h2>')
     with gr.Row():
         prompt = gr.Textbox(label="prompt", lines=10)
     with gr.Row():
@@ -178,4 +201,38 @@ with gr.Blocks(title="Salamandra 7B Instruct · ZeroGPU") as demo:
 
     btn2.click(salamandra_chat_endpoint, [prompt], out2, api_name="generate_out_from_prompt", concurrency_limit=1)
 
+    gr.Markdown('<h2 style="text-align:center">Resumir frases</h2>')
+    with gr.Row():
+        with gr.Column(scale=1):
+            frase = gr.Textbox(label="Frase a resumir", value="", lines=3)
+            num_paraules = gr.Textbox(label="Nombre de paraules del resum", value="4")
+            btn_resumir = gr.Button("Resumir", variant="primary")
+        with gr.Column(scale=1):
+            out_resumir = gr.Textbox(label="Resposta", lines=18)
+
+    btn_resumir.click(
+        resumir_frases,
+        inputs=[frase, num_paraules],
+        outputs=out_resumir,
+        api_name="resumir",
+        concurrency_limit=1
+    )
+
+    gr.Markdown('<h2 style="text-align:center">Inclusió d’identitats</h2>')
+    with gr.Row():
+        with gr.Column(scale=1):
+            frase = gr.Textbox(label="Frase a modificar", value="", lines=3)
+            persona = gr.Textbox(label="Persones reconegudes", value='"Mireia Martí": 4, "Xavier Busquets": 5')
+            btn_modificar = gr.Button("Modificar frase", variant="primary")
+        with gr.Column(scale=1):
+            out_modificat = gr.Textbox(label="Resposta", lines=18)
+
+    btn_modificar.click(
+        identity_manager,
+        inputs=[frase, persona],
+        outputs=out_modificat,
+        api_name="modificar",
+        concurrency_limit=1
+    )
+
 demo.queue(max_size=16).launch()