Update agents/specialized.py

agents/specialized.py

@@ -8,6 +8,7 @@ import re
 import json
 import asyncio
 import requests
+import base64
 from pathlib import Path
 from datetime import datetime
 
@@ -104,8 +105,42 @@ async def generate_with_gemini(queries, model_name):
     return []
 
 async def generate_with_hf(queries):
-    # ⚠️ tu lógica HF
-    return []
+    urls = []
+
+    for i, prompt in enumerate(queries[:2]):
+        response = requests.post(
+            "https://router.huggingface.co/v1/images/generations",
+            headers={
+                "Authorization": f"Bearer {os.getenv('HF_API_TOKEN')}"
+            },
+            json={
+                "model": "black-forest-labs/FLUX.1-schnell",
+                "prompt": prompt,
+                "size": "1024x1024"
+            }
+        )
+
+        if response.status_code == 200:
+            data = response.json()
+
+            image_base64 = data["data"][0]["b64_json"]
+            image_bytes = base64.b64decode(image_base64)
+
+            DOCS_DIR = Path("data/docs")
+            DOCS_DIR.mkdir(exist_ok=True)
+
+            file_name = f"hf_image_{datetime.now().timestamp()}.png"
+            file_path = DOCS_DIR / file_name
+
+            with open(file_path, "wb") as f:
+                f.write(image_bytes)
+
+            urls.append(f"/docs/{file_name}")
+
+        else:
+            print("HF error:", response.text)
+
+    return urls
 
 # 🟢 FUNCIÓN PRINCIPAL