vfven commited on
Commit
d5f8a1c
·
verified ·
1 Parent(s): 1bb46da

Update agents/specialized.py

Browse files
Files changed (1) hide show
  1. agents/specialized.py +2 -2
agents/specialized.py CHANGED
@@ -87,7 +87,7 @@ async def generate_real_image(prompts: list, model: str = "black-forest-labs/FLU
87
 
88
  results.append({
89
  "success": True,
90
- "image_url": f"/data/docs/{file_name}",
91
  "prompt_used": enhanced_prompt
92
  })
93
  except Exception as e:
@@ -157,7 +157,7 @@ async def run_specialized_agent(agent_key: str, task: str, context: dict = None)
157
  result["response"] = "Ideas de imágenes generadas:\n" + "\n".join(queries)
158
 
159
  # Generar imágenes reales si el usuario lo pide explícitamente
160
- trigger_words = ["genera la imagen", "crea la imagen", "muestra la imagen", "imagen real", "quiero ver", "haz la imagen"]
161
  if any(word in task.lower() for word in trigger_words):
162
  img_results = await generate_real_image(queries)
163
  successful_images = [r["image_url"] for r in img_results if r["success"]]
 
87
 
88
  results.append({
89
  "success": True,
90
+ "image_url": f"/docs/{file_name}",
91
  "prompt_used": enhanced_prompt
92
  })
93
  except Exception as e:
 
157
  result["response"] = "Ideas de imágenes generadas:\n" + "\n".join(queries)
158
 
159
  # Generar imágenes reales si el usuario lo pide explícitamente
160
+ trigger_words = ["imagen", "image", "genera", "crear", "muestra", "ver"]
161
  if any(word in task.lower() for word in trigger_words):
162
  img_results = await generate_real_image(queries)
163
  successful_images = [r["image_url"] for r in img_results if r["success"]]