Update main.py

main.py

@@ -104,9 +104,17 @@ DEFAULT_AGENTS = [
 # ── CHAT ROLES — conversational versions for direct chat (not mission mode) ──
 CHAT_ROLES = {
     "manager": (
-        "Eres el Manager de Mission Control AI, un gerente de proyectos experimentado y amigable. "
-        "En este chat puedes conversar libremente, dar opiniones, ayudar a planificar, responder preguntas generales. "
-        "Habla de forma natural y directa. Recuerda el historial de la conversación y refiérete a él cuando sea relevante."
+        "Eres el Manager de Mission Control AI, gerente de proyectos con acceso real a un equipo de agentes IA. "
+        "Tu equipo: backend_dev (Python/APIs), frontend_dev (HTML/CSS/JS), analyst (análisis), "
+        "writer (redacción), image_agent (imágenes/arte). "
+        "REGLA CRÍTICA: Si el usuario pide algo que requiere trabajo real de un agente "
+        "(generar imagen, escribir código, hacer Excel, crear informe, diseñar web, etc.), "
+        "responde ÚNICAMENTE con este JSON y nada más: "
+        '{"action":"delegate","task":"descripcion clara de lo que se necesita"} '
+        "Ejemplos que SÍ requieren delegar: "
+        "'genera una imagen', 'hazme un script', 'crea un formulario HTML', 'haz un informe', 'excel de ventas'. "
+        "Para conversación normal (saludos, preguntas, planificación, consejos) responde con texto normal. "
+        "Recuerda el historial y refiérete a él."
     ),
     "backend_dev": (
         "Eres Backend Dev, un programador senior especializado en Python, APIs, bases de datos y DevOps. "
@@ -871,13 +879,100 @@ async def chat_with_agent(request: Request):
     try:
         response = await call_llm_multiturn(agent, history)
 
-        # ImageAgent: intercept generate_image action
-        img_result = None
-        img_files  = []
+        # Intercept special actions from agents
+        img_result    = None
+        img_files     = []
+        delegate_result = None
         import json as _json2
         try:
             parsed = _json2.loads(response.strip())
-            if isinstance(parsed, dict) and parsed.get("action") == "generate_image":
+            # Manager delegates to team
+            if isinstance(parsed, dict) and parsed.get("action") == "delegate" and agent_key == "manager":
+                delegate_task = parsed.get("task", message)
+                # Run as a collab mission
+                sub_results = {}
+                sub_context = "TAREA: " + delegate_task + "\n"
+                sub_today   = datetime.now().strftime("%A %d de %B de %Y, %H:%M")
+                # Detect which agents to use
+                lo = delegate_task.lower()
+                sub_delegates = []
+                if any(w in lo for w in ["imagen","image","foto","dibujo","genera","crea una imagen"]):
+                    sub_delegates = ["image_agent"]
+                elif any(w in lo for w in ["excel","planilla","spreadsheet"]):
+                    sub_delegates = ["backend_dev"]
+                elif any(w in lo for w in ["informe","reporte","documento"]):
+                    sub_delegates = ["writer","analyst"]
+                elif any(w in lo for w in ["html","web","formulario","interfaz"]) and any(w in lo for w in ["api","backend","python"]):
+                    sub_delegates = ["backend_dev","frontend_dev"]
+                elif any(w in lo for w in ["html","web","formulario","interfaz"]):
+                    sub_delegates = ["frontend_dev"]
+                elif any(w in lo for w in ["python","api","script","backend","codigo","código"]):
+                    sub_delegates = ["backend_dev"]
+                else:
+                    sub_delegates = ["analyst"]
+                # Execute each sub-agent
+                sub_imgs = []
+                for sub_key in sub_delegates:
+                    if sub_key not in agent_registry: continue
+                    sub_agent = dict(agent_registry[sub_key])
+                    sub_role  = get_chat_role(sub_key, sub_agent)
+                    sub_agent["role"] = "HOY ES: " + sub_today + ".\n" + sub_role + "\n\nContexto: " + sub_context
+                    try:
+                        if sub_key == "image_agent":
+                            ip = ("Find images for: " + delegate_task +
+                                  ' Respond ONLY: {"image_queries":["t1","t2","t3"]}')
+                            sub_raw = await call_llm(sub_agent, ip)
+                            m3 = re.search(r'"image_queries"\s*:\s*\[([^\]]*)\]', sub_raw)
+                            qs  = re.findall(r'"([^"]+)"', m3.group(1)) if m3 else [delegate_task[:40]]
+                            sub_imgs_r = await asyncio.gather(*[get_image(q, force_generate=True) for q in qs[:3]])
+                            _safe2 = re.sub(r"[^\w]","_",delegate_task[:28])
+                            _ts2   = datetime.now().strftime("%H%M%S")
+                            _base2 = "mgr_" + _safe2 + "_" + _ts2
+                            for idx2, im2 in enumerate(sub_imgs_r):
+                                if im2:
+                                    fn2 = _base2 + "_img" + str(idx2+1) + ".jpg"
+                                    (DOCS_DIR / fn2).write_bytes(im2)
+                                    sub_imgs.append(fn2)
+                            sub_results[sub_key] = {
+                                "status": "active",
+                                "message": str(len(sub_imgs)) + " imagen(es) para: " + ", ".join(qs[:3]),
+                                "img_base": _base2 if sub_imgs else None,
+                                "img_count": len(sub_imgs),
+                            }
+                            sub_context += "\n=== IMAGE AGENT ===\n" + str(len(sub_imgs)) + " images\n"
+                        else:
+                            sub_prompt = sub_context + "\nINSTRUCCION: " + delegate_task
+                            sub_resp   = await call_llm_multiturn(sub_agent, [{"role":"user","content":sub_prompt}])
+                            sub_results[sub_key] = {"status":"active","message":sub_resp,"model":sub_agent["models"][0]}
+                            sub_context += "\n=== " + sub_key.upper() + " ===\n" + sub_resp[:500] + "\n"
+                            # Save files
+                            _ts3  = datetime.now().strftime("%Y%m%d_%H%M%S")
+                            _safe3 = re.sub(r"[^\w\-]","_",delegate_task[:35])
+                            if sub_key == "frontend_dev" and not is_skip(sub_resp):
+                                html3 = re.sub(r"```\w*\n?","",sub_resp); html3 = re.sub(r"```","",html3).strip()
+                                if len(html3)>80:
+                                    fn3 = _safe3+"_frontend_"+_ts3+".html"
+                                    (DOCS_DIR/fn3).write_text(html3,encoding="utf-8")
+                                    sub_results[sub_key]["doc_file"]=fn3
+                            elif sub_key == "backend_dev" and not is_skip(sub_resp):
+                                code3 = re.sub(r"```\w*\n?","",sub_resp); code3 = re.sub(r"```","",code3).strip()
+                                if len(code3)>80:
+                                    fn3 = _safe3+"_backend_"+_ts3+".py"
+                                    (DOCS_DIR/fn3).write_text(code3,encoding="utf-8")
+                                    sub_results[sub_key]["doc_file"]=fn3
+                            mission_context_cache[sub_key] = "En tarea reciente: " + delegate_task[:60] + "\n" + sub_resp[:400]
+                    except Exception as sub_e:
+                        sub_results[sub_key] = {"status":"resting","message":str(sub_e)}
+                # Build manager response text
+                parts = ["Delegué la tarea a mi equipo:"]
+                for k,r in sub_results.items():
+                    parts.append("• " + k + ": " + (r.get("message","")[:120] or "done"))
+                response = "\n".join(parts)
+                delegate_result = {"sub_results": sub_results, "sub_imgs": sub_imgs,
+                                   "img_base": sub_results.get("image_agent",{}).get("img_base"),
+                                   "img_count": sub_results.get("image_agent",{}).get("img_count",0)}
+            # ImageAgent: generate image
+            elif isinstance(parsed, dict) and parsed.get("action") == "generate_image":
                 queries = parsed.get("queries", [message])[:3]
                 _safe   = re.sub(r"[^\w]", "_", message[:28])
                 _ts     = datetime.now().strftime("%H%M%S")
@@ -912,6 +1007,11 @@ async def chat_with_agent(request: Request):
         }
         if img_result:
             result.update(img_result)
+        if delegate_result:
+            result["delegate_result"] = delegate_result
+            if delegate_result.get("img_base"):
+                result["img_base"]  = delegate_result["img_base"]
+                result["img_count"] = delegate_result["img_count"]
         return JSONResponse(result)
 
     except Exception as e: