Update app.py

app.py

@@ -38,10 +38,31 @@ class LlamaRefiner:
             return text
         try:
             result = self.translator(text, max_length=250, clean_up_tokenization_spaces=True)
-            return result[0]['translation_text'].strip()
+            raw_translation = result[0]['translation_text'].strip()
         except Exception as e:
             logger.warning(f"Traducción local fallida: {e}. Usando texto original.")
-            return text
+            raw_translation = text
+
+        user_text_lower = text.lower()
+        output = raw_translation
+
+        if any(kw in user_text_lower for kw in ["llamas", "ardiendo", "quem", "incendi", "fuego"]):
+            output = output.replace("fiery", "on fire")
+            if not any(term in output.lower() for term in ["on fire", "burning", "in flames", "ablaze", "aflame"]):
+                output = output + " on fire"
+
+        if any(kw in user_text_lower for kw in ["oro", "dorado"]):
+            if "golden" not in output.lower() and "gold" not in output.lower():
+                if any(w in output.lower() for w in ["statue", "sculpture", "figure"]):
+                    output = output + " made of gold"
+                else:
+                    output = output + " golden"
+
+        if any(kw in user_text_lower for kw in ["congelado", "hielo", "helado", "ice"]):
+            if not any(term in output.lower() for term in ["frozen", "ice", "icy"]):
+                output = output + " frozen"
+
+        return output.strip()
 
     def retrieve_similar_examples(self, user_prompt_en: str, category: str = "auto", k: int = 6) -> list:
         if not self.agent.is_ready:
@@ -85,9 +106,9 @@ class LlamaRefiner:
         user_prompt_en = self.translate_to_english(user_prompt)
         examples = self.retrieve_similar_examples(user_prompt_en, category=category, k=6)
 
-        # ✅ Usar solo enriquecimiento local (más fiable y coherente)
+        # ✅ Usar SOLO enriquecimiento local (como en la versión que funcionaba)
         enhanced_prompt, _ = self.agent.enhance_prompt(user_prompt_en, category=category)
-        return enhanced_prompt.strip(), "✨ Enriquecimiento semántico por frases coherentes", examples
+        return enhanced_prompt.strip(), "✨ Enriquecimiento semántico local", examples
 
 class SDXLGenerator:
     def __init__(self):
@@ -124,7 +145,7 @@ def create_interface():
             return "", "", "Prompt vacío."
         if refiner is None:
             return "", "", "Servicios no disponibles."
-        progress(0.2, desc="🌍 Traduciendo y enriqueciendo...")
+        progress(0.2, desc="🌍 Traduciendo...")
         category_map = {
             "Automática": "auto",
             "Entidad": "entity",
@@ -195,7 +216,7 @@ def create_interface():
                     label="Prompt refinado (inglés)",
                     interactive=False,
                     lines=3,
-                    elem_classes=["gr-copyable"]  # ← activa el botón de copia en Gradio
+                    elem_classes=["gr-copyable"]
                 )
                 image_out = gr.Image(label="Imagen", type="filepath", height=450)
                 examples_out = gr.Textbox(label="Ejemplos del dataset (para análisis)", interactive=False, lines=6)