Public Access
159 lines
7.3 KiB
Python
159 lines
7.3 KiB
Python
import os
|
|
import requests
|
|
import time
|
|
import concurrent.futures
|
|
import base64
|
|
from modules.logger import log_event
|
|
|
|
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
|
|
|
|
# LISTE DE REPLI HF : Modèles garantis sur le tier 100% gratuit
|
|
modeles_gratuits = [
|
|
"runwayml/stable-diffusion-v1-5", # L'indestructible par défaut
|
|
"prompthero/openjourney", # Excellent pour un style Midjourney
|
|
"stabilityai/stable-diffusion-2-1"
|
|
]
|
|
|
|
for modele_cible in modeles_gratuits:
|
|
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=60)
|
|
if resp.status_code == 200:
|
|
return resp.content, "Succès", modele_cible
|
|
else:
|
|
log_event("WARNING", "MEDIA", f"Échec HF ({modele_cible}) : {resp.status_code} - {resp.text}")
|
|
continue # Le modèle a été rejeté, on tente immédiatement le suivant
|
|
except Exception as e:
|
|
log_event("WARNING", "MEDIA", f"Exception HF ({modele_cible}) : {str(e)}")
|
|
continue
|
|
|
|
# Si la boucle se termine sans aucun succès
|
|
return None, "Tous les modèles gratuits ont été rejetés (410) ou sont indisponibles", "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\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\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\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 |