Update agents/specialized.py

agents/specialized.py

@@ -4,13 +4,17 @@ from utils.prompts import get_today_prefix, get_chat_style
 from agents.registry import AGENT_REGISTRY
 from core.file_builder import build_docx, build_excel
 import re
+import json
+from pathlib import Path
+from datetime import datetime
 
 AGENT_ROLES = {
     "manager": """
-Eres el gerente principal. Cuando no haya delegación, responde de forma natural, amigable y útil al usuario.
+Eres el gerente principal. Cuando no haya delegación o la tarea sea simple, responde de forma natural, amigable y útil al usuario.
 No necesitas generar JSON aquí, solo contesta directamente.
+Mantén un tono profesional pero cercano.
 """,
-    
+
     "image_agent": """
 Eres el agente de imágenes.
 Cuando se te pida generar o buscar imágenes, responde SOLO con:
@@ -61,13 +65,10 @@ async def generate_image_from_prompt(prompt: str, model: str = "black-forest-lab
         image = await asyncio.to_thread(
             client.text_to_image,
             prompt,
-            num_inference_steps=20,  # ajusta para más calidad (más lento)
+            num_inference_steps=20,
             guidance_scale=7.5
         )
         
-        # Guardar temporalmente
-        from pathlib import Path
-        from datetime import datetime
         DOCS_DIR = Path("data/docs")
         DOCS_DIR.mkdir(exist_ok=True)
         ts = datetime.now().strftime("%Y%m%d_%H%M%S")
@@ -91,7 +92,6 @@ async def run_specialized_agent(agent_key: str, task: str, context: dict = None)
 
     full_prompt = today + role + get_chat_style() + f"\n\nTarea asignada: {task}"
 
-    # Agregar contexto si viene de otros agentes
     if context:
         full_prompt += "\n\nContexto de otros agentes:\n" + "\n".join([f"[{k.upper()}] {v[:300]}..." for k, v in context.items()])
 
@@ -105,7 +105,8 @@ async def run_specialized_agent(agent_key: str, task: str, context: dict = None)
             "response": response,
             "success": True,
             "file_path": None,
-            "file_type": None
+            "file_type": None,
+            "image_url": None
         }
 
         # Generar archivo según agente
@@ -120,40 +121,35 @@ async def run_specialized_agent(agent_key: str, task: str, context: dict = None)
                 result["file_error"] = str(e)
 
         elif agent_key == "backend_dev":
-            # Detectar EXCEL_TEMPLATE
             match = re.search(r'EXCEL_TEMPLATE:\s*(\{.*\})', response, re.DOTALL | re.IGNORECASE)
             if match:
                 try:
-                    import json
                     excel_data = json.loads(match.group(1))
                     file_path = build_excel(task, excel_data)
                     if file_path:
                         result["file_path"] = file_path.name
                         result["file_type"] = "xlsx"
-                        # Limpiar respuesta para no mostrar el template crudo
-                        result["response"] = response.split("EXCEL_TEMPLATE:")[0].strip()
+                        result["response"] = response.split("EXCEL_TEMPLATE:")[0].strip()  # limpiar
                 except Exception as e:
                     result["file_error"] = str(e)
 
         elif agent_key == "image_agent":
-            # Generar queries (como antes)
             match = re.search(r'\{.*"image_queries".*\}', response, re.DOTALL)
             if match:
                 try:
                     queries = json.loads(match.group(0))["image_queries"]
                     result["queries"] = queries
                     
-                    # Si la tarea pide generar imagen real
+                    # Generar imagen real si el usuario lo pide explícitamente
                     if any(word in task.lower() for word in ["genera la imagen", "crea la imagen", "muestra la imagen", "imagen real"]):
-                        # Usa la primera query (o combina todas)
                         best_prompt = ", ".join(queries) + ", highly detailed, 8k, masterpiece"
                         img_result = await generate_image_from_prompt(best_prompt)
                         if img_result["success"]:
                             result["image_url"] = img_result["image_url"]
                         else:
                             result["image_error"] = img_result["error"]
-                except:
-                    pass
+                except Exception as e:
+                    result["image_error"] = str(e)
 
         return result