Update app.py

app.py

@@ -131,6 +131,18 @@ def predict_for_engine(prompt: str) -> str:
 def generate_advanced(prompt: str, system: str, max_new_tokens: int, temperature: float, top_p: float) -> str:
     return _generate(prompt=prompt, system=system, max_new_tokens=max_new_tokens, temperature=temperature, top_p=top_p)
 
+def salamandra_chat_endpoint(prompt: str) -> Dict[str, Any]:
+    global _salamandra
+    if _salamandra is None:
+        _salamandra = SalamandraClient()   # usa tu clase
+
+    try:
+        text = _salamandra.chat(prompt)
+    except Exception as e:
+        text = f"Error ejecutando SalamandraClient: {str(e)}"
+
+    return {"text": text}
+
 # ------------------- HTTP (opcional, clientes puros) -------------------
 # Si quieres, puedes añadir un endpoint HTTP POST /generate (FastAPI),
 # pero con Gradio Client es suficiente para engine/local.