Update app.py

app.py

@@ -171,7 +171,20 @@ def identity_manager (frase, persona):
 # pero con Gradio Client es suficiente para engine/local.
 
 # ------------------- UI -------------------
-with gr.Blocks(title="Salamandra 7B Instruct · ZeroGPU") as demo:
+custom_css = """
+h2 {
+    background: #e3e4e6 !important;
+    padding: 14px 22px !important;
+    border-radius: 14px !important;
+    box-shadow: 0 4px 12px rgba(0,0,0,0.08) !important;
+    display: block !important;       /* ocupa tot l'ample */
+    width: 100% !important;          /* assegura 100% */
+    margin: 20px auto !important;
+    text-align:center;
+}
+"""
+
+with gr.Blocks(title="Salamandra 7B Instruct · ZeroGPU",css=custom_css,theme=gr.themes.Soft()) as demo:
     gr.Markdown("## Salamandra-7B-Instruct · ZeroGPU\nTexto → respuesta instruccional.")
     with gr.Row():
         with gr.Column(scale=1):
@@ -185,12 +198,13 @@ with gr.Blocks(title="Salamandra 7B Instruct · ZeroGPU") as demo:
             out = gr.Textbox(label="Respuesta", lines=18)
 
     btn.click(generate_advanced, [in_prompt, in_system, max_new, temp, top_p], out, api_name="generate", concurrency_limit=1)
-
+    gr.Markdown("---")
     # Endpoint minimalista compatible con el ENGINE (/predict: solo prompt)
     in_prompt_engine = gr.Textbox(label="Prompt (ENGINE)", value="Di hola en una frase.")
     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("---")
+    
     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)
@@ -200,15 +214,17 @@ with gr.Blocks(title="Salamandra 7B Instruct · ZeroGPU") as demo:
         out2 = gr.JSON(label="Salida")
 
     btn2.click(salamandra_chat_endpoint, [prompt], out2, api_name="generate_out_from_prompt", concurrency_limit=1)
-
+    gr.Markdown("---")
+    
     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)
+    with gr.Row():
+        btn_resumir = gr.Button("Resumir", variant="primary")
 
     btn_resumir.click(
         resumir_frases,
@@ -217,16 +233,18 @@ with gr.Blocks(title="Salamandra 7B Instruct · ZeroGPU") as demo:
         api_name="resumir",
         concurrency_limit=1
     )
-
+    gr.Markdown("---")
+    
     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)
-
+    with gr.Row():
+        btn_modificar = gr.Button("Modificar frase", variant="primary")
+        
     btn_modificar.click(
         identity_manager,
         inputs=[frase, persona],