import os import requests import time import concurrent.futures import base64 from modules.logger import log_event def obtenir_modele_image_hf_dynamique(api_key: str) -> str: """Interroge le Hub HF pour trouver le modèle image gratuit le plus populaire ET actif (warm).""" url = "https://huggingface.co/api/models" params = { "pipeline_tag": "text-to-image", "sort": "downloads", "direction": -1, "inference": "warm", # GARANTIT que le modèle est gratuit et en ligne "limit": 3 } headers = {"Authorization": f"Bearer {api_key}"} if api_key else {} try: resp = requests.get(url, headers=headers, params=params, timeout=5) if resp.status_code == 200 and len(resp.json()) > 0: return resp.json()[0]["id"] except Exception: pass # Secours absolu si l'API de liste ne répond pas return "runwayml/stable-diffusion-v1-5" def generer_image_cloudflare(prompt: str) -> bytes: account_id = os.getenv("CLOUDFLARE_ACCOUNT_ID") api_token = os.getenv("CLOUDFLARE_API_TOKEN") if not account_id or not api_token: return None url = f"https://api.cloudflare.com/client/v4/accounts/{account_id}/ai/run/@cf/stabilityai/stable-diffusion-xl-base-1.0" headers = {"Authorization": f"Bearer {api_token}"} try: resp = requests.post(url, headers=headers, json={"prompt": prompt}, timeout=120) if resp.status_code == 200: if "application/json" in resp.headers.get("Content-Type", ""): data = resp.json() if data.get("success") and "result" in data: return base64.b64decode(data["result"]) else: return resp.content except Exception as e: log_event("WARNING", "MEDIA", f"Échec Cloudflare : {e}") return None def generer_image_openrouter(prompt: str) -> tuple: api_key = os.getenv("OPENROUTER_API_KEY") if not api_key: return None, None url = "https://openrouter.ai/api/v1/images" headers = { "Authorization": f"Bearer {api_key}", "HTTP-Referer": "https://aethas38.duckdns.org", "X-Title": "Aethas38" } # LISTE DE REPLI (Fallback) : Maîtrise des coûts tout en évitant les crashs modeles_payants = ["bytedance-seed/seedream-4.5", "openai/dall-e-3"] for modele in modeles_payants: payload = {"prompt": prompt, "model": modele} try: resp = requests.post(url, headers=headers, json=payload, timeout=120) if resp.status_code == 200: data = resp.json() if "data" in data and len(data["data"]) > 0 and "b64_json" in data["data"][0]: img_bytes = base64.b64decode(data["data"][0]["b64_json"]) return img_bytes, modele except Exception as e: log_event("WARNING", "MEDIA", f"Échec OpenRouter ({modele}) : {e}") continue # Passe au modèle de secours suivant return None, None def generer_image_huggingface(prompt: str) -> tuple: api_key = os.getenv("HUGGINGFACE_API_KEY") if not api_key: return None, "Clé API manquante", None # Sélection dynamique HF + Utilisation de la passerelle de secours réseau (router) modele_cible = obtenir_modele_image_hf_dynamique(api_key) url = f"https://router.huggingface.co/hf-inference/models/{modele_cible}" headers = {"Authorization": f"Bearer {api_key}"} try: resp = requests.post(url, headers=headers, json={"inputs": prompt}, timeout=120) if resp.status_code == 200: return resp.content, "Succès", modele_cible else: return None, f"Erreur HTTP {resp.status_code} : {resp.text}", modele_cible except Exception as e: log_event("WARNING", "MEDIA", f"Échec HF : {e}") return None, f"Exception système : {str(e)}", "Inconnu" def generer_les_images(prompt: str, fichiers: list, premium: bool) -> tuple: trace = [] if fichiers: trace.append("🟢 Mode Image-to-Image") return f"*(BETA)* Le mode Image-to-Image est désactivé temporairement. \n\nFichiers : {', '.join(fichiers)}", "#ff9800", "Système", "En dev", trace trace.append(f"🟢 Lancement Text-to-Image (Premium: {premium})") with concurrent.futures.ThreadPoolExecutor() as executor: future_or = executor.submit(generer_image_openrouter, prompt) if premium else None future_cf = executor.submit(generer_image_cloudflare, prompt) future_hf = executor.submit(generer_image_huggingface, prompt) img_or, nom_modele_or = future_or.result() if future_or else (None, None) img_cf = future_cf.result() img_hf, msg_hf, nom_modele_hf = future_hf.result() if future_hf else (None, None, None) dossier_media = "/DATA/AppData/MULTI-IA-AETHAS38/RAG/Media" os.makedirs(dossier_media, exist_ok=True) reponse_md = f"**Prompt analysé :** {prompt}\n\n" fournisseurs_actifs = [] # --- Rendu OpenRouter --- if future_or is None: trace.append("⚪ OR : Ignoré (Mode Gratuit)") elif img_or: nom_or = f"img_or_{int(time.time())}.png" with open(os.path.join(dossier_media, nom_or), 'wb') as f: f.write(img_or) reponse_md += f"### 🌐 OpenRouter ({nom_modele_or})\n![OpenRouter](/Media/{nom_or})\n" reponse_md += f"**[📥 Télécharger l'image OR](/Media/{nom_or})**\n\n---\n\n" fournisseurs_actifs.append("OpenRouter") trace.append(f"🟢 OR ({nom_modele_or}) : Succès") else: trace.append("🔴 OR : Échec") # --- Rendu Cloudflare --- if img_cf: nom_cf = f"img_cf_{int(time.time())}.png" with open(os.path.join(dossier_media, nom_cf), 'wb') as f: f.write(img_cf) reponse_md += f"### ☁️ Cloudflare Workers AI\n![Cloudflare](/Media/{nom_cf})\n" reponse_md += f"**[📥 Télécharger l'image CF](/Media/{nom_cf})**\n\n---\n\n" fournisseurs_actifs.append("Cloudflare") trace.append("🟢 CF : Succès") else: trace.append("🔴 CF : Échec") # --- Rendu HuggingFace Dynamique --- if img_hf: nom_hf = f"img_hf_{int(time.time())}.png" with open(os.path.join(dossier_media, nom_hf), 'wb') as f: f.write(img_hf) nom_propre = nom_modele_hf.split('/')[-1] if nom_modele_hf else "Générateur" reponse_md += f"### 🤗 HuggingFace ({nom_propre})\n![HuggingFace](/Media/{nom_hf})\n" reponse_md += f"**[📥 Télécharger l'image HF](/Media/{nom_hf})**\n\n" fournisseurs_actifs.append("HuggingFace") trace.append(f"🟢 HF ({nom_propre}) : Succès") else: erreur_courte = str(msg_hf)[:50] + "..." if len(str(msg_hf)) > 50 else str(msg_hf) trace.append(f"🔴 HF : Échec ({erreur_courte})") reponse_md += f"### 🤗 HuggingFace\n*⚠️ Échec de la génération : {msg_hf}*\n\n" if not img_or and not img_cf and not img_hf: return "*Échec critique : Aucune API n'a pu générer d'image.*", "#ff4444", "Système", "Erreur", trace return reponse_md, "#00e5ff", " + ".join(fournisseurs_actifs) if fournisseurs_actifs else "Aucun", "Moteur Média", trace def router_media(prompt: str, type_media: str, premium: bool, fichiers: list) -> tuple: trace = [f"🟢 Routage Média : {type_media.upper()}"] if type_media == "image": return generer_les_images(prompt, fichiers, premium) elif type_media == "video": return "*La génération vidéo nécessite une API payante.* \n\nPrompt : " + prompt, "#ff4444", "Système", "En attente API", trace return "Type de média non reconnu.", "#333", "Erreur", "Aucun", trace