Version 2.1 amélioration sécurité et corrections bugs majeures.
This commit is contained in:
xavier committed
1 parent
0259427cf2
commit
e301b931c6
4 files changed
+828
-339
No files matched your search
@@ -6,215 +6,10 @@ import uuid
|
||||
import asyncio
|
||||
from datetime import datetime
|
||||
from dotenv import load_dotenv
|
||||
from auth import validate_password_strength, hash_password, verify_password, generate_totp_secret, verify_totp
|
||||
|
||||
load_dotenv()
|
||||
|
||||
# 2. INITIALISATION DE FASTHTML EN PREMIER (C'est ici que 'rt' est créé)
|
||||
app, rt = fast_app()
|
||||
|
||||
from auth import validate_password_strength, hash_password, verify_password, generate_totp_secret, verify_totp
|
||||
|
||||
# Exemple de structure pour les routes de connexion (Login / Setup initial)
|
||||
@rt('/login')
|
||||
def get():
|
||||
return Title("Connexion - AETHAS38"), Body(
|
||||
Div(
|
||||
H2("AETHAS 38 - Sécurité", style="color:#00e5ff; margin-bottom:20px;"),
|
||||
Form(
|
||||
Input(type="email", name="email", placeholder="E-mail", cls="input-box mb-3", required=True),
|
||||
Input(type="password", name="password", placeholder="Mot de passe", cls="input-box mb-3", required=True),
|
||||
Input(type="text", name="totp_code", placeholder="Code 2FA (6 chiffres)", cls="input-box mb-3", required=True),
|
||||
Button("Se connecter", type="submit", cls="send-btn w-full"),
|
||||
hx_post="/auth/login", hx_target="#auth-response"
|
||||
),
|
||||
Div(id="auth-response", style="color:red; margin-top:10px;"),
|
||||
cls="sidebar", style="margin: auto; width: 400px; height: auto; border-radius: 10px; margin-top: 15vh;"
|
||||
), style="background-color: #0f172a; height: 100vh; display: flex;"
|
||||
)
|
||||
|
||||
# --- 0. INITIALISATION DES DOSSIERS (BDD & LOGS) ---
|
||||
os.makedirs("sessions", exist_ok=True)
|
||||
os.makedirs("logs", exist_ok=True)
|
||||
|
||||
# Un seul fichier de log par lancement d'application
|
||||
APP_LAUNCH_TIME = datetime.now().strftime("%Y%m%d_%H%M%S")
|
||||
LOG_FILE = f"logs/AETHAS38_Run_{APP_LAUNCH_TIME}.log"
|
||||
|
||||
def log_event(session_id, category, action):
|
||||
"""Enregistre chaque action dans le fichier log global avec catégorisation."""
|
||||
sid_short = session_id[:8] if session_id and session_id != "system" else "SYSTEM"
|
||||
timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
|
||||
with open(LOG_FILE, "a", encoding="utf-8") as f:
|
||||
f.write(f"[{timestamp}] [Session:{sid_short}] [{category}] {action}\n")
|
||||
|
||||
# --- 1. GESTION DE L'HISTORIQUE (INDEX & SESSIONS) ---
|
||||
HISTORY_FILE = "history.json"
|
||||
|
||||
def load_index():
|
||||
if not os.path.exists(HISTORY_FILE): return {}
|
||||
with open(HISTORY_FILE, "r", encoding="utf-8") as f:
|
||||
try:
|
||||
idx = json.load(f)
|
||||
for sid, data in list(idx.items()):
|
||||
if "messages" in data:
|
||||
save_session_data(sid, data["messages"])
|
||||
del data["messages"]
|
||||
return idx
|
||||
except: return {}
|
||||
|
||||
def save_index(idx):
|
||||
with open(HISTORY_FILE, "w", encoding="utf-8") as f:
|
||||
json.dump(idx, f, indent=4, ensure_ascii=False)
|
||||
|
||||
def load_session_data(session_id):
|
||||
path = f"sessions/{session_id}.json"
|
||||
if not os.path.exists(path): return []
|
||||
with open(path, "r", encoding="utf-8") as f:
|
||||
try: return json.load(f)
|
||||
except: return []
|
||||
|
||||
def save_session_data(session_id, messages):
|
||||
with open(f"sessions/{session_id}.json", "w", encoding="utf-8") as f:
|
||||
json.dump(messages, f, indent=4, ensure_ascii=False)
|
||||
|
||||
def create_new_session():
|
||||
session_id = str(uuid.uuid4())
|
||||
idx = load_index()
|
||||
idx[session_id] = {"title": f"Discussion du {datetime.now().strftime('%d/%m %H:%M')}", "pinned": False}
|
||||
save_index(idx)
|
||||
save_session_data(session_id, [])
|
||||
log_event(session_id, "USER_ACTION", "Nouvelle session initialisée.")
|
||||
return session_id
|
||||
|
||||
def render_history_list():
|
||||
idx = load_index()
|
||||
sorted_sessions = sorted(idx.items(), key=lambda x: (not x[1].get('pinned', False), x[1]['title']), reverse=False)
|
||||
|
||||
items = [
|
||||
Button("➕ Nouvelle Discussion", cls="history-btn w-full mb-3 text-center", style="background:#00e5ff; color:#0f172a; font-weight:bold;", hx_get="/session/new", hx_target="#chat-history")
|
||||
]
|
||||
|
||||
for sid, data in sorted_sessions:
|
||||
pin_icon = "📍" if data.get('pinned') else "📌"
|
||||
pin_color = "#10b981" if data.get('pinned') else "#94a3b8"
|
||||
|
||||
item = Div(
|
||||
Div(data['title'], cls="truncate flex-grow cursor-pointer hover:text-cyan-400", hx_get=f"/session/load/{sid}", hx_target="#chat-history"),
|
||||
Div(
|
||||
Button("✏️", cls="text-xs mx-1 hover:text-white", onclick=f"let name = prompt('Nouveau nom:'); if(name) {{ htmx.ajax('POST', '/history/rename/{sid}', {{values: {{title: name}}, target: '#history-list-container'}}); }}"),
|
||||
Button(pin_icon, style=f"color:{pin_color};", cls="text-xs mx-1 hover:text-white", hx_post=f"/history/pin/{sid}", hx_target="#history-list-container"),
|
||||
Button("🗑️", cls="text-xs hover:text-red-500", onclick=f"if(confirm('Supprimer définitivement cette discussion ?')) {{ htmx.ajax('POST', '/history/delete/{sid}', {{target: '#history-list-container'}}); }}"),
|
||||
cls="flex-shrink-0"
|
||||
),
|
||||
cls="flex justify-between items-center history-btn w-full mb-1"
|
||||
)
|
||||
items.append(item)
|
||||
return Div(*items, id="history-list-container", hx_swap_oob="true")
|
||||
|
||||
|
||||
# --- 2. GESTION DU CATALOGUE (GENERIQUE & OPENROUTER) ---
|
||||
MODELS_DATA = []
|
||||
def init_models():
|
||||
global MODELS_DATA
|
||||
MODELS_DATA = [
|
||||
{"id": "groq|llama3-8b-8192", "name": "Groq - Llama 3 (8B)", "is_free": True, "description": "Modèle ultra-rapide hébergé par Groq. Idéal pour les tâches générales, la structuration de données et le code basique.", "architecture": {"modality": "text"}},
|
||||
{"id": "gemini|gemini-1.5-flash", "name": "Google - Gemini 1.5 Flash", "is_free": True, "description": "Modèle multimodal très rapide et léger de Google. Parfait pour le raisonnement rapide, le code et l'analyse de longs contextes.", "architecture": {"modality": "text/vision"}},
|
||||
{"id": "deepseek|deepseek-coder", "name": "DeepSeek - Coder", "is_free": False, "description": "Modèle d'élite spécialisé dans l'ingénierie logicielle, la création d'algorithmes et la génération de code complexe.", "architecture": {"modality": "text/code"}}
|
||||
]
|
||||
try:
|
||||
res = requests.get("https://openrouter.ai/api/v1/models", timeout=5)
|
||||
if res.status_code == 200:
|
||||
for m in res.json().get("data", []):
|
||||
m["id"] = f"openrouter|{m['id']}"
|
||||
m["name"] = f"OR - {m['name']}"
|
||||
MODELS_DATA.append(m)
|
||||
except Exception as e:
|
||||
log_event("system", "ERROR", f"Erreur chargement catalogue OpenRouter : {e}")
|
||||
init_models()
|
||||
|
||||
def get_model_options(filter_type="all"):
|
||||
options = []
|
||||
for m in sorted(MODELS_DATA, key=lambda x: x['name']):
|
||||
is_free = m.get("is_free", "free" in m.get("id", "").lower())
|
||||
if filter_type == "free" and not is_free: continue
|
||||
if filter_type == "paid" and is_free: continue
|
||||
tag = "Gratuit" if is_free else "Payant"
|
||||
options.append(Option(f"{m['name']} ({tag})", value=m['id']))
|
||||
return Select(*options, name="model_id", cls="model-select", style="width: 100%;")
|
||||
|
||||
|
||||
# --- 3. MOTEUR D'INTELLIGENCE & GESTION DES ERREURS ---
|
||||
def get_budget():
|
||||
mgmt_key = os.getenv("OPENROUTER_MANAGEMENT_KEY")
|
||||
if not mgmt_key: return "Clé manquante"
|
||||
try:
|
||||
data = requests.get("https://openrouter.ai/api/v1/credits", headers={"Authorization": f"Bearer {mgmt_key}"}, timeout=5).json().get("data", {})
|
||||
return f"{data.get('total_credits', 0) - data.get('total_usage', 0):.4f} $"
|
||||
except: return "Erreur lecture"
|
||||
|
||||
def handle_api_error(session_id, provider, raw_error):
|
||||
log_event(session_id, "ERROR", f"Erreur API {provider} : {raw_error}")
|
||||
sys_prompt = "Tu es un assistant technique. Une API distante a renvoyé l'erreur suivante. Traduis-la en français simplement et propose une explication claire pour aider l'utilisateur."
|
||||
try:
|
||||
qwen_res = requests.post("http://localhost:11434/api/generate", json={"model": "qwen2.5-coder:7b", "prompt": f"{sys_prompt}\n\nErreur ({provider}): {raw_error}", "stream": False}, timeout=15).json()["response"]
|
||||
return f"⚠️ **Alerte Serveur ({provider})**\n{qwen_res}"
|
||||
except:
|
||||
return f"⚠️ Erreur brute ({provider}) : {raw_error}"
|
||||
|
||||
async def async_ask_llm(session_id, provider, actual_model, msg):
|
||||
return await asyncio.to_thread(ask_llm, session_id, provider, actual_model, msg)
|
||||
|
||||
def ask_llm(session_id, provider, actual_model, msg):
|
||||
log_event(session_id, "API_REQ", f"Interrogation API distantes : {provider} | Modèle : {actual_model}")
|
||||
try:
|
||||
if provider == "openrouter":
|
||||
res = requests.post("https://openrouter.ai/api/v1/chat/completions", headers={"Authorization": f"Bearer {os.getenv('OPENROUTER_API_KEY')}"}, json={"model": actual_model, "messages": [{"role": "user", "content": msg}]})
|
||||
if res.status_code != 200: return handle_api_error(session_id, provider, res.text)
|
||||
return res.json()["choices"][0]["message"]["content"]
|
||||
|
||||
elif provider == "groq":
|
||||
res = requests.post("https://api.groq.com/openai/v1/chat/completions", headers={"Authorization": f"Bearer {os.getenv('GROQ_API_KEY')}"}, json={"model": actual_model, "messages": [{"role": "user", "content": msg}]})
|
||||
if res.status_code != 200: return handle_api_error(session_id, provider, res.text)
|
||||
return res.json()["choices"][0]["message"]["content"]
|
||||
|
||||
elif provider == "gemini":
|
||||
res = requests.post(f"https://generativelanguage.googleapis.com/v1beta/models/{actual_model}:generateContent?key={os.getenv('GEMINI_API_KEY')}", json={"contents": [{"parts": [{"text": msg}]}]})
|
||||
if res.status_code != 200: return handle_api_error(session_id, provider, res.text)
|
||||
return res.json()["candidates"][0]["content"]["parts"][0]["text"]
|
||||
|
||||
return handle_api_error(session_id, provider, "Fournisseur non implémenté.")
|
||||
except Exception as e:
|
||||
return handle_api_error(session_id, provider, str(e))
|
||||
|
||||
|
||||
# --- 4. PIPELINES DE TRAITEMENT ---
|
||||
def pipeline_minecraft(session_id, raw_msg):
|
||||
log_event(session_id, "PIPELINE", "Lancement Pipeline 100% Local Minecraft (Qwen -> Phuzzy -> Qwen)")
|
||||
try:
|
||||
en_prompt = requests.post("http://localhost:11434/api/generate", json={"model": "qwen2.5-coder:7b", "prompt": f"Translate this Minecraft modding request into technical English. Return ONLY English text.\n\n{raw_msg}", "stream": False}).json()["response"]
|
||||
phuzzy_res = requests.post("http://localhost:11434/api/generate", json={"model": "phuzzy:latest", "prompt": en_prompt, "stream": False}).json()["response"]
|
||||
final_fr = requests.post("http://localhost:11434/api/generate", json={"model": "qwen2.5-coder:7b", "prompt": f"Traduis les explications techniques en français. NE TRADUIS PAS le code.\n\n{phuzzy_res}", "stream": False}).json()["response"]
|
||||
return final_fr, "Qwen ➔ Phuzzy ➔ Qwen"
|
||||
except Exception as e:
|
||||
return handle_api_error(session_id, "Ollama Local", str(e)), "Erreur Locale"
|
||||
|
||||
async def pipeline_consolidated(session_id, raw_msg):
|
||||
log_event(session_id, "PIPELINE", "Lancement Pipeline Consolidation Multi-IA")
|
||||
try:
|
||||
opt_prompt = requests.post("http://localhost:11434/api/generate", json={"model": "qwen2.5-coder:7b", "prompt": f"Optimise cette demande pour des LLM codeurs. Sois ultra précis.\n\n{raw_msg}", "stream": False}).json()["response"]
|
||||
rep_groq, rep_gemini = await asyncio.gather(
|
||||
async_ask_llm(session_id, "groq", "llama3-8b-8192", opt_prompt),
|
||||
async_ask_llm(session_id, "gemini", "gemini-1.5-flash", opt_prompt)
|
||||
)
|
||||
sys_synth = "Tu es un Architecte Logiciel Senior. Voici la même demande traitée par deux IA différentes. Lis leurs propositions, corrige les erreurs potentielles, garde le meilleur des deux, et génère le code final absolu et parfait en français."
|
||||
final_prompt = f"{sys_synth}\n\n--- IA 1 (Groq) ---\n{rep_groq}\n\n--- IA 2 (Gemini) ---\n{rep_gemini}"
|
||||
final_res = requests.post("http://localhost:11434/api/generate", json={"model": "qwen2.5-coder:7b", "prompt": final_prompt, "stream": False}).json()["response"]
|
||||
return final_res, "Qwen ➔ [Groq + Gemini] ➔ Synthèse Qwen"
|
||||
except Exception as e:
|
||||
return handle_api_error(session_id, "Pipeline Multi-IA", str(e)), "Erreur Consolidation"
|
||||
|
||||
|
||||
# --- 5. INTERFACE UTILISATEUR & ROUTES ---
|
||||
theme_hdrs = [
|
||||
Script(src="https://cdn.tailwindcss.com"),
|
||||
Script(src="https://cdn.jsdelivr.net/npm/marked/marked.min.js"),
|
||||
@@ -250,7 +45,6 @@ theme_hdrs = [
|
||||
if(ctx === 'minecraft' || ctx === 'consolidated') { modBox.style.display = 'none'; }
|
||||
else { modBox.style.display = 'flex'; }
|
||||
}
|
||||
|
||||
function updateFileCount(input) {
|
||||
let label = document.getElementById('file-label-text');
|
||||
if(input && input.files && input.files.length > 0) {
|
||||
@@ -261,7 +55,6 @@ theme_hdrs = [
|
||||
label.style.color = '#94a3b8';
|
||||
}
|
||||
}
|
||||
|
||||
async function downloadModels() {
|
||||
let tracker = document.getElementById('loading-tracker');
|
||||
let oldText = tracker.innerText;
|
||||
@@ -285,7 +78,6 @@ theme_hdrs = [
|
||||
tracker.style.display = 'none';
|
||||
tracker.innerText = oldText;
|
||||
}
|
||||
|
||||
htmx.onLoad(function(content) {
|
||||
content.querySelectorAll('.bubble-ia:not(.rendered)').forEach(function(el) {
|
||||
el.innerHTML = marked.parse(el.textContent);
|
||||
@@ -297,29 +89,237 @@ theme_hdrs = [
|
||||
""")
|
||||
]
|
||||
|
||||
app, rt = fast_app(hdrs=theme_hdrs)
|
||||
# Activation des sessions chiffrées avec la clé définie dans le .env
|
||||
app, rt = fast_app(secret_key=os.getenv("SESSION_SECRET", "super-secret-key-fallback"), hdrs=theme_hdrs)
|
||||
|
||||
@rt('/session/new')
|
||||
# --- ROUTES D'AUTHENTIFICATION ---
|
||||
@rt('/login')
|
||||
def get():
|
||||
return Title("Connexion - AETHAS38"), Body(
|
||||
Div(
|
||||
H2("AETHAS 38 - Accès Sécurisé", style="color:#00e5ff; margin-bottom:20px; text-align:center;"),
|
||||
Form(
|
||||
Input(type="email", name="email", placeholder="E-mail", cls="input-box mb-3 w-full", required=True),
|
||||
Input(type="password", name="password", placeholder="Mot de passe", cls="input-box mb-3 w-full", required=True),
|
||||
Input(type="text", name="totp_code", placeholder="Code 2FA (6 chiffres)", cls="input-box mb-4 w-full", required=True),
|
||||
Button("Se connecter", type="submit", cls="send-btn w-full"),
|
||||
hx_post="/auth/login", hx_target="#auth-response"
|
||||
),
|
||||
Div(id="auth-response", style="color:red; margin-top:10px; text-align:center;"),
|
||||
cls="sidebar", style="margin: auto; width: 400px; height: auto; border-radius: 10px; margin-top: 15vh;"
|
||||
), style="background-color: #0f172a; height: 100vh; display: flex;"
|
||||
)
|
||||
|
||||
@rt('/auth/login')
|
||||
def post(email: str, password: str, totp_code: str, session):
|
||||
# On utilise VOS noms de variables exacts
|
||||
admin_email = os.getenv("ADMIN_EMAIL")
|
||||
admin_hash = os.getenv("ADMIN_PASSWORD_HASH")
|
||||
totp_secret = os.getenv("ADMIN_2FA_SECRET")
|
||||
|
||||
if not admin_email or not admin_hash or not totp_secret:
|
||||
return Div("Configuration système incomplète (.env).", style="color:red; font-weight:bold;")
|
||||
|
||||
if email == admin_email and verify_password(admin_hash, password) and verify_totp(totp_secret, totp_code):
|
||||
session["authenticated"] = True
|
||||
session["role"] = "admin"
|
||||
return Script("window.location.href = '/';")
|
||||
else:
|
||||
return Div("Identifiants ou code 2FA incorrects.", style="color:#ef4444; font-weight:bold;")
|
||||
|
||||
@rt('/logout')
|
||||
def get(session):
|
||||
session.clear()
|
||||
return RedirectResponse('/login')
|
||||
|
||||
@rt('/check_updates')
|
||||
def get():
|
||||
try:
|
||||
res = requests.get("https://api.github.com/repos/Reivax69008/Aethas-Multi-IA/commits/main", timeout=3).json()
|
||||
remote_commit = res.get("sha", "")
|
||||
local_commit = ""
|
||||
if os.path.exists(".git/refs/heads/main"):
|
||||
with open(".git/refs/heads/main", "r") as f:
|
||||
local_commit = f.read().strip()
|
||||
|
||||
if local_commit and remote_commit and remote_commit != local_commit:
|
||||
return Span("🚀 MAJ disponible (GitHub)", style="color:#fbbf24;")
|
||||
elif local_commit == remote_commit:
|
||||
return Span("🟢 Système à jour", style="color:#10b981;")
|
||||
else:
|
||||
return Span(f"🟢 Connecté (Git: {remote_commit[:7]})", style="color:#38bdf8;")
|
||||
except:
|
||||
return Span("⚪ Statut réseau inconnu", style="color:#94a3b8;")
|
||||
|
||||
# --- DOSSIERS & LOGS ---
|
||||
os.makedirs("sessions", exist_ok=True)
|
||||
os.makedirs("logs", exist_ok=True)
|
||||
APP_LAUNCH_TIME = datetime.now().strftime("%Y%m%d_%H%M%S")
|
||||
LOG_FILE = f"logs/AETHAS38_Run_{APP_LAUNCH_TIME}.log"
|
||||
|
||||
def log_event(session_id, category, action):
|
||||
sid_short = session_id[:8] if session_id and session_id != "system" else "SYSTEM"
|
||||
timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
|
||||
with open(LOG_FILE, "a", encoding="utf-8") as f:
|
||||
f.write(f"[{timestamp}] [Session:{sid_short}] [{category}] {action}\n")
|
||||
|
||||
# --- HISTORIQUE ---
|
||||
HISTORY_FILE = "history.json"
|
||||
def load_index():
|
||||
if not os.path.exists(HISTORY_FILE): return {}
|
||||
with open(HISTORY_FILE, "r", encoding="utf-8") as f:
|
||||
try:
|
||||
idx = json.load(f)
|
||||
for sid, data in list(idx.items()):
|
||||
if "messages" in data:
|
||||
save_session_data(sid, data["messages"])
|
||||
del data["messages"]
|
||||
return idx
|
||||
except: return {}
|
||||
def save_index(idx):
|
||||
with open(HISTORY_FILE, "w", encoding="utf-8") as f:
|
||||
json.dump(idx, f, indent=4, ensure_ascii=False)
|
||||
def load_session_data(session_id):
|
||||
path = f"sessions/{session_id}.json"
|
||||
if not os.path.exists(path): return []
|
||||
with open(path, "r", encoding="utf-8") as f:
|
||||
try: return json.load(f)
|
||||
except: return []
|
||||
def save_session_data(session_id, messages):
|
||||
with open(f"sessions/{session_id}.json", "w", encoding="utf-8") as f:
|
||||
json.dump(messages, f, indent=4, ensure_ascii=False)
|
||||
def create_new_session():
|
||||
session_id = str(uuid.uuid4())
|
||||
idx = load_index()
|
||||
idx[session_id] = {"title": f"Discussion du {datetime.now().strftime('%d/%m %H:%M')}", "pinned": False}
|
||||
save_index(idx)
|
||||
save_session_data(session_id, [])
|
||||
log_event(session_id, "USER_ACTION", "Nouvelle session initialisée.")
|
||||
return session_id
|
||||
def render_history_list():
|
||||
idx = load_index()
|
||||
sorted_sessions = sorted(idx.items(), key=lambda x: (not x[1].get('pinned', False), x[1]['title']), reverse=False)
|
||||
items = [Button("➕ Nouvelle Discussion", cls="history-btn w-full mb-3 text-center", style="background:#00e5ff; color:#0f172a; font-weight:bold;", hx_get="/session/new", hx_target="#chat-history")]
|
||||
for sid, data in sorted_sessions:
|
||||
pin_icon, pin_color = ("📍", "#10b981") if data.get('pinned') else ("📌", "#94a3b8")
|
||||
item = Div(
|
||||
Div(data['title'], cls="truncate flex-grow cursor-pointer hover:text-cyan-400", hx_get=f"/session/load/{sid}", hx_target="#chat-history"),
|
||||
Div(
|
||||
Button("✏️", cls="text-xs mx-1 hover:text-white", onclick=f"let name = prompt('Nouveau nom:'); if(name) {{ htmx.ajax('POST', '/history/rename/{sid}', {{values: {{title: name}}, target: '#history-list-container'}}); }}"),
|
||||
Button(pin_icon, style=f"color:{pin_color};", cls="text-xs mx-1 hover:text-white", hx_post=f"/history/pin/{sid}", hx_target="#history-list-container"),
|
||||
Button("🗑️", cls="text-xs hover:text-red-500", onclick=f"if(confirm('Supprimer définitivement cette discussion ?')) {{ htmx.ajax('POST', '/history/delete/{sid}', {{target: '#history-list-container'}}); }}"),
|
||||
cls="flex-shrink-0"
|
||||
),
|
||||
cls="flex justify-between items-center history-btn w-full mb-1"
|
||||
)
|
||||
items.append(item)
|
||||
return Div(*items, id="history-list-container", hx_swap_oob="true")
|
||||
|
||||
# --- CATALOGUE ---
|
||||
MODELS_DATA = []
|
||||
def init_models():
|
||||
global MODELS_DATA
|
||||
MODELS_DATA = [
|
||||
{"id": "groq|llama3-8b-8192", "name": "Groq - Llama 3 (8B)", "is_free": True, "description": "Modèle ultra-rapide hébergé par Groq.", "architecture": {"modality": "text"}},
|
||||
{"id": "gemini|gemini-1.5-flash", "name": "Google - Gemini 1.5 Flash", "is_free": True, "description": "Modèle multimodal léger de Google.", "architecture": {"modality": "text/vision"}},
|
||||
{"id": "deepseek|deepseek-coder", "name": "DeepSeek - Coder", "is_free": False, "description": "Génération de code complexe.", "architecture": {"modality": "text/code"}}
|
||||
]
|
||||
try:
|
||||
res = requests.get("https://openrouter.ai/api/v1/models", timeout=5)
|
||||
if res.status_code == 200:
|
||||
for m in res.json().get("data", []):
|
||||
m["id"], m["name"] = f"openrouter|{m['id']}", f"OR - {m['name']}"
|
||||
MODELS_DATA.append(m)
|
||||
except Exception as e: log_event("system", "ERROR", f"Erreur catalogue OR : {e}")
|
||||
init_models()
|
||||
def get_model_options(filter_type="all"):
|
||||
options = []
|
||||
for m in sorted(MODELS_DATA, key=lambda x: x['name']):
|
||||
is_free = m.get("is_free", "free" in m.get("id", "").lower())
|
||||
if filter_type == "free" and not is_free: continue
|
||||
if filter_type == "paid" and is_free: continue
|
||||
options.append(Option(f"{m['name']} ({'Gratuit' if is_free else 'Payant'})", value=m['id']))
|
||||
return Select(*options, name="model_id", cls="model-select", style="width: 100%;")
|
||||
|
||||
# --- MOTEUR IA ---
|
||||
def get_budget():
|
||||
mgmt_key = os.getenv("OPENROUTER_MANAGEMENT_KEY")
|
||||
if not mgmt_key: return "Clé manquante"
|
||||
try:
|
||||
data = requests.get("https://openrouter.ai/api/v1/credits", headers={"Authorization": f"Bearer {mgmt_key}"}, timeout=5).json().get("data", {})
|
||||
return f"{data.get('total_credits', 0) - data.get('total_usage', 0):.4f} $"
|
||||
except: return "Erreur lecture"
|
||||
|
||||
def handle_api_error(session_id, provider, raw_error):
|
||||
log_event(session_id, "ERROR", f"API {provider} : {raw_error}")
|
||||
sys_prompt = "Tu es un assistant technique. Traduis cette erreur d'API en français."
|
||||
try:
|
||||
qwen_res = requests.post("http://localhost:11434/api/generate", json={"model": "qwen2.5-coder:7b", "prompt": f"{sys_prompt}\n\nErreur ({provider}): {raw_error}", "stream": False}, timeout=15).json()["response"]
|
||||
return f"⚠️ **Alerte Serveur ({provider})**\n{qwen_res}"
|
||||
except: return f"⚠️ Erreur brute ({provider}) : {raw_error}"
|
||||
|
||||
async def async_ask_llm(session_id, provider, actual_model, msg):
|
||||
return await asyncio.to_thread(ask_llm, session_id, provider, actual_model, msg)
|
||||
|
||||
def ask_llm(session_id, provider, actual_model, msg):
|
||||
log_event(session_id, "API_REQ", f"Interrogation : {provider} | Modèle : {actual_model}")
|
||||
try:
|
||||
if provider == "openrouter":
|
||||
res = requests.post("https://openrouter.ai/api/v1/chat/completions", headers={"Authorization": f"Bearer {os.getenv('OPENROUTER_API_KEY')}"}, json={"model": actual_model, "messages": [{"role": "user", "content": msg}]})
|
||||
if res.status_code != 200: return handle_api_error(session_id, provider, res.text)
|
||||
return res.json()["choices"][0]["message"]["content"]
|
||||
elif provider == "groq":
|
||||
res = requests.post("https://api.groq.com/openai/v1/chat/completions", headers={"Authorization": f"Bearer {os.getenv('GROQ_API_KEY')}"}, json={"model": actual_model, "messages": [{"role": "user", "content": msg}]})
|
||||
if res.status_code != 200: return handle_api_error(session_id, provider, res.text)
|
||||
return res.json()["choices"][0]["message"]["content"]
|
||||
elif provider == "gemini":
|
||||
res = requests.post(f"https://generativelanguage.googleapis.com/v1beta/models/{actual_model}:generateContent?key={os.getenv('GEMINI_API_KEY')}", json={"contents": [{"parts": [{"text": msg}]}]})
|
||||
if res.status_code != 200: return handle_api_error(session_id, provider, res.text)
|
||||
return res.json()["candidates"][0]["content"]["parts"][0]["text"]
|
||||
return handle_api_error(session_id, provider, "Fournisseur non implémenté.")
|
||||
except Exception as e: return handle_api_error(session_id, provider, str(e))
|
||||
|
||||
def pipeline_minecraft(session_id, raw_msg):
|
||||
log_event(session_id, "PIPELINE", "Local Minecraft")
|
||||
try:
|
||||
en_prompt = requests.post("http://localhost:11434/api/generate", json={"model": "qwen2.5-coder:7b", "prompt": f"Translate this Minecraft modding request into technical English. Return ONLY English text.\n\n{raw_msg}", "stream": False}).json()["response"]
|
||||
phuzzy_res = requests.post("http://localhost:11434/api/generate", json={"model": "phuzzy:latest", "prompt": en_prompt, "stream": False}).json()["response"]
|
||||
final_fr = requests.post("http://localhost:11434/api/generate", json={"model": "qwen2.5-coder:7b", "prompt": f"Traduis les explications techniques en français. NE TRADUIS PAS le code.\n\n{phuzzy_res}", "stream": False}).json()["response"]
|
||||
return final_fr, "Qwen ➔ Phuzzy ➔ Qwen"
|
||||
except Exception as e: return handle_api_error(session_id, "Ollama Local", str(e)), "Erreur Locale"
|
||||
|
||||
async def pipeline_consolidated(session_id, raw_msg):
|
||||
log_event(session_id, "PIPELINE", "Consolidation Multi-IA")
|
||||
try:
|
||||
opt_prompt = requests.post("http://localhost:11434/api/generate", json={"model": "qwen2.5-coder:7b", "prompt": f"Optimise cette demande pour des LLM codeurs. Sois ultra précis.\n\n{raw_msg}", "stream": False}).json()["response"]
|
||||
rep_groq, rep_gemini = await asyncio.gather(
|
||||
async_ask_llm(session_id, "groq", "llama3-8b-8192", opt_prompt),
|
||||
async_ask_llm(session_id, "gemini", "gemini-1.5-flash", opt_prompt)
|
||||
)
|
||||
sys_synth = "Tu es un Architecte Logiciel Senior. Voici la même demande traitée par deux IA différentes. Lis leurs propositions, corrige les erreurs potentielles, garde le meilleur des deux, et génère le code final absolu et parfait en français."
|
||||
final_prompt = f"{sys_synth}\n\n--- IA 1 (Groq) ---\n{rep_groq}\n\n--- IA 2 (Gemini) ---\n{rep_gemini}"
|
||||
final_res = requests.post("http://localhost:11434/api/generate", json={"model": "qwen2.5-coder:7b", "prompt": final_prompt, "stream": False}).json()["response"]
|
||||
return final_res, "Qwen ➔ [Groq + Gemini] ➔ Synthèse Qwen"
|
||||
except Exception as e: return handle_api_error(session_id, "Pipeline Multi-IA", str(e)), "Erreur Consolidation"
|
||||
|
||||
# --- ROUTES PRINCIPALES (PROTÉGÉES) ---
|
||||
@rt('/session/new')
|
||||
def get(session):
|
||||
if not session.get("authenticated"): return RedirectResponse('/login')
|
||||
sid = create_new_session()
|
||||
new_hidden = Input(type="hidden", name="session_id", value=sid, id="current-session-id", hx_swap_oob="true")
|
||||
return Div(), new_hidden, render_history_list()
|
||||
return Div(), Input(type="hidden", name="session_id", value=sid, id="current-session-id", hx_swap_oob="true"), render_history_list()
|
||||
|
||||
@rt('/session/load/{sid}')
|
||||
def get(sid: str):
|
||||
def get(sid: str, session):
|
||||
if not session.get("authenticated"): return RedirectResponse('/login')
|
||||
log_event(sid, "USER_ACTION", "Chargement de la session.")
|
||||
messages = load_session_data(sid)
|
||||
bubbles = []
|
||||
for msg in messages:
|
||||
bubbles.append(Div(
|
||||
P("Vous", cls="msg-user"), Div(NotStr(f"{msg['user']}{msg.get('info_fichiers', '')}{msg.get('info_local', '')}"), cls="bubble-user"),
|
||||
P(f"AETHAS38 ({msg.get('nom_affichage', 'IA')})", cls="msg-ia"), Div(msg['ia'], cls="bubble-ia")
|
||||
))
|
||||
new_hidden = Input(type="hidden", name="session_id", value=sid, id="current-session-id", hx_swap_oob="true")
|
||||
return tuple(bubbles) + (new_hidden,)
|
||||
bubbles = [Div(P("Vous", cls="msg-user"), Div(NotStr(f"{msg['user']}{msg.get('info_fichiers', '')}{msg.get('info_local', '')}"), cls="bubble-user"), P(f"AETHAS38 ({msg.get('nom_affichage', 'IA')})", cls="msg-ia"), Div(msg['ia'], cls="bubble-ia")) for msg in messages]
|
||||
return tuple(bubbles) + (Input(type="hidden", name="session_id", value=sid, id="current-session-id", hx_swap_oob="true"),)
|
||||
|
||||
@rt('/history/pin/{sid}')
|
||||
def post_pin(sid: str):
|
||||
def post_pin(sid: str, session):
|
||||
if not session.get("authenticated"): return RedirectResponse('/login')
|
||||
idx = load_index()
|
||||
if sid in idx:
|
||||
idx[sid]['pinned'] = not idx[sid].get('pinned', False)
|
||||
@@ -328,7 +328,8 @@ def post_pin(sid: str):
|
||||
return render_history_list()
|
||||
|
||||
@rt('/history/rename/{sid}')
|
||||
def post_rename(sid: str, title: str):
|
||||
def post_rename(sid: str, title: str, session):
|
||||
if not session.get("authenticated"): return RedirectResponse('/login')
|
||||
idx = load_index()
|
||||
if sid in idx:
|
||||
idx[sid]['title'] = title
|
||||
@@ -337,77 +338,28 @@ def post_rename(sid: str, title: str):
|
||||
return render_history_list()
|
||||
|
||||
@rt('/history/delete/{sid}')
|
||||
def post_delete(sid: str):
|
||||
def post_delete(sid: str, session):
|
||||
if not session.get("authenticated"): return RedirectResponse('/login')
|
||||
idx = load_index()
|
||||
if sid in idx:
|
||||
del idx[sid]
|
||||
save_index(idx)
|
||||
if os.path.exists(f"sessions/{sid}.json"): os.remove(f"sessions/{sid}.json")
|
||||
log_event(sid, "USER_ACTION", "Session supprimée définitivement.")
|
||||
log_event(sid, "USER_ACTION", "Session supprimée.")
|
||||
return render_history_list()
|
||||
|
||||
@rt('/export_models')
|
||||
def get():
|
||||
log_event("system", "SYSTEM", "Demande d'exportation du catalogue de modèles.")
|
||||
now = datetime.now()
|
||||
txt_content = f"=== CATALOGUE COMPLET DES MODÈLES (Généré le {now.strftime('%d/%m/%Y à %H:%M:%S')}) ===\n\n"
|
||||
|
||||
google_banned = False # Sécurité pour ne pas figer si Google bloque l'IP
|
||||
|
||||
for m in MODELS_DATA:
|
||||
name = m.get('name', 'Inconnu')
|
||||
m_id = m.get('id', 'Inconnu')
|
||||
|
||||
usage = "Général / Texte"
|
||||
arch_mod = str(m.get('architecture', {}).get('modality', '')).lower()
|
||||
m_id_lower = m_id.lower()
|
||||
|
||||
if "vision" in arch_mod or "vision" in m_id_lower or "vl" in m_id_lower or "image" in arch_mod:
|
||||
usage = "Vision (Analyse d'images & Multimodal)"
|
||||
elif "coder" in m_id_lower or "code" in m_id_lower or "math" in m_id_lower:
|
||||
usage = "Codage & Mathématiques"
|
||||
elif "video" in m_id_lower:
|
||||
usage = "Vidéo"
|
||||
elif "rp" in m_id_lower or "uncensored" in m_id_lower or "roleplay" in m_id_lower:
|
||||
usage = "Non-censuré / Roleplay"
|
||||
|
||||
is_free = m.get("is_free")
|
||||
if is_free is None:
|
||||
pricing = m.get('pricing', {})
|
||||
p_prompt = float(pricing.get('prompt', -1))
|
||||
p_comp = float(pricing.get('completion', -1))
|
||||
cout = "Gratuit" if p_prompt == 0.0 and p_comp == 0.0 else f"Payant (In: {p_prompt}, Out: {p_comp})"
|
||||
else:
|
||||
cout = "Gratuit" if is_free else "Payant"
|
||||
|
||||
desc_courte = str(m.get('description', '')).replace('\n', ' ')[:150]
|
||||
desc_fr = desc_courte
|
||||
|
||||
if "openrouter" in m_id and desc_courte:
|
||||
if not google_banned:
|
||||
try:
|
||||
url = "https://translate.googleapis.com/translate_a/single?client=gtx&sl=en&tl=fr&dt=t&q=" + requests.utils.quote(desc_courte)
|
||||
trans_res = requests.get(url, timeout=1.5)
|
||||
if trans_res.status_code == 200:
|
||||
desc_fr = "".join([x[0] for x in trans_res.json()[0]])
|
||||
else:
|
||||
google_banned = True
|
||||
desc_fr += " [Traduction interrompue: Sécurité anti-spam]"
|
||||
except:
|
||||
google_banned = True
|
||||
desc_fr += " [Traduction interrompue: Délai d'attente]"
|
||||
else:
|
||||
desc_fr = desc_courte
|
||||
|
||||
if not desc_fr: desc_fr = "Aucune description fournie."
|
||||
|
||||
txt_content += f"Nom : {name}\nID : {m_id}\nTarif : {cout}\nSpécialité : {usage}\nDescription : {desc_fr}\n{'-'*60}\n"
|
||||
|
||||
def get(session):
|
||||
if not session.get("authenticated"): return RedirectResponse('/login')
|
||||
log_event("system", "SYSTEM", "Export catalogue.")
|
||||
txt_content = f"=== CATALOGUE ({datetime.now().strftime('%d/%m/%Y %H:%M:%S')}) ===\n\n"
|
||||
for m in MODELS_DATA: txt_content += f"Nom : {m.get('name')}\nID : {m.get('id')}\nDescription : {str(m.get('description', ''))[:150]}\n{'-'*60}\n"
|
||||
return Response(txt_content, media_type="text/plain")
|
||||
|
||||
|
||||
@rt('/')
|
||||
def get():
|
||||
def get(session):
|
||||
if not session.get("authenticated"): return RedirectResponse('/login')
|
||||
|
||||
session_id = create_new_session()
|
||||
return Title("AETHAS38 Multi-IA"), Body(
|
||||
Div(
|
||||
@@ -415,19 +367,22 @@ def get():
|
||||
H2("AETHAS 38", style="color:#00e5ff; font-weight:bold; font-size:24px;"),
|
||||
P("Tech Core - Multi IA", style="color:#a855f7; margin-bottom: 20px;"),
|
||||
|
||||
Div(
|
||||
P("STATUT SYSTÈME", style="color:#94a3b8; font-size:10px; font-weight:bold; margin-bottom:5px;"),
|
||||
Div("⏳ Vérification GitHub...", hx_get="/check_updates", hx_trigger="load", style="font-size:12px; font-weight:bold; margin-bottom:15px;"),
|
||||
),
|
||||
|
||||
Div(
|
||||
H3("Historique de Code", style="color:#94a3b8; font-size:12px; font-weight:bold; margin-bottom:10px;"),
|
||||
render_history_list(),
|
||||
style="display:flex; flex-direction:column; flex-grow:1; margin-bottom: 10px; overflow:hidden;"
|
||||
),
|
||||
|
||||
# Le bouton fait désormais appel à la fonction JavaScript de téléchargement
|
||||
Button("📥 Exporter Modèles (.txt)", type="button", cls="export-btn w-full mt-2", onclick="downloadModels()"),
|
||||
|
||||
A("Déconnexion", href="/logout", cls="export-btn w-full mt-2", style="color:#ef4444; border-color:#ef4444; margin-bottom:10px;"),
|
||||
Div(
|
||||
P("BUDGET OPENROUTER", style="color:#94a3b8; font-size:10px; font-weight:bold; margin-bottom:5px;"),
|
||||
P(get_budget(), id="budget-display", style="color:#10b981; font-size:18px; font-weight:bold;"),
|
||||
cls="budget-box", style="margin-top:15px;"
|
||||
cls="budget-box"
|
||||
),
|
||||
cls="sidebar"
|
||||
),
|
||||
@@ -438,9 +393,9 @@ def get():
|
||||
Form(
|
||||
Input(type="hidden", name="session_id", value=session_id, id="current-session-id"),
|
||||
Div(
|
||||
Label(Input(type="radio", name="context_type", value="generic", checked=True, onchange="updateUI()"), " 💻 Code Générique (Modèle Unique)"),
|
||||
Label(Input(type="radio", name="context_type", value="consolidated", onchange="updateUI()"), " 🧠 Consolidation (Multi-IA ➔ Qwen)"),
|
||||
Label(Input(type="radio", name="context_type", value="minecraft", onchange="updateUI()"), " ⛏ Minecraft (100% Local)"),
|
||||
Label(Input(type="radio", name="context_type", value="generic", checked=True, onchange="updateUI()"), " 💻 Code Générique"),
|
||||
Label(Input(type="radio", name="context_type", value="consolidated", onchange="updateUI()"), " 🧠 Consolidation"),
|
||||
Label(Input(type="radio", name="context_type", value="minecraft", onchange="updateUI()"), " ⛏ Minecraft"),
|
||||
cls="context-radio-group text-sm text-white flex gap-4"
|
||||
),
|
||||
Div(
|
||||
@@ -465,34 +420,31 @@ def get():
|
||||
)
|
||||
|
||||
@rt('/filter_models')
|
||||
def get(filter_type: str): return get_model_options(filter_type)
|
||||
def get(filter_type: str, session):
|
||||
if not session.get("authenticated"): return RedirectResponse('/login')
|
||||
return get_model_options(filter_type)
|
||||
|
||||
@rt('/chat')
|
||||
async def post(msg: str, model_id: str, session_id: str, context_type: str = "generic", fichiers: list[UploadFile] = None):
|
||||
log_event(session_id, "USER_ACTION", f"Envoi d'un message (Contexte: {context_type})")
|
||||
async def post(msg: str, model_id: str, session_id: str, session, context_type: str = "generic", fichiers: list[UploadFile] = None):
|
||||
if not session.get("authenticated"): return RedirectResponse('/login')
|
||||
log_event(session_id, "USER_ACTION", f"Message ({context_type})")
|
||||
|
||||
files_context = ""
|
||||
noms_fichiers = []
|
||||
files_context, noms_fichiers = "", []
|
||||
if fichiers:
|
||||
for f in fichiers:
|
||||
if f.filename:
|
||||
noms_fichiers.append(f.filename)
|
||||
try: files_context += f"\n--- {f.filename} ---\n{(await f.read()).decode('utf-8')}\n"
|
||||
except:
|
||||
files_context += f"\n--- {f.filename} (Binaire ignoré) ---\n"
|
||||
log_event(session_id, "FILE_SYS", f"Fichier non lisible ignoré : {f.filename}")
|
||||
if noms_fichiers: log_event(session_id, "FILE_SYS", f"Fichiers joints : {', '.join(noms_fichiers)}")
|
||||
except: files_context += f"\n--- {f.filename} (Binaire ignoré) ---\n"
|
||||
|
||||
full_req = f"Fichiers fournis:\n{files_context}\nDemande: {msg}" if files_context else msg
|
||||
|
||||
if context_type == "minecraft":
|
||||
ia_reponse, pipeline_info = pipeline_minecraft(session_id, full_req)
|
||||
nom_affichage = "Local - Phuzzy/Minecraft"
|
||||
|
||||
elif context_type == "consolidated":
|
||||
ia_reponse, pipeline_info = await pipeline_consolidated(session_id, full_req)
|
||||
nom_affichage = "Qwen Synthèse (via Groq/Gemini)"
|
||||
|
||||
else:
|
||||
opt_prompt = requests.post("http://localhost:11434/api/generate", json={"model": "qwen2.5-coder:7b", "prompt": f"Optimise cette demande pour un LLM codeur. Retourne UNIQUEMENT le prompt.\n\n{full_req}", "stream": False}).json()["response"]
|
||||
parts = model_id.split("|", 1)
|
||||
@@ -501,27 +453,15 @@ async def post(msg: str, model_id: str, session_id: str, context_type: str = "ge
|
||||
nom_affichage = f"{provider.capitalize()} - {actual_model.split('/')[-1]}"
|
||||
pipeline_info = "Qwen ➔ API Distante Unique"
|
||||
|
||||
info_fichiers = f"<br><span style='color:#a855f7; font-size:10px;'>📎 {len(noms_fichiers)} fichier(s) joint(s)</span>" if noms_fichiers else ""
|
||||
info_fichiers = f"<br><span style='color:#a855f7; font-size:10px;'>📎 {len(noms_fichiers)} fichier(s)</span>" if noms_fichiers else ""
|
||||
info_local = f"<br><span style='color:#38bdf8; font-size:10px;'>⚙️ Pipeline : {pipeline_info}</span>"
|
||||
|
||||
session_data = load_session_data(session_id)
|
||||
session_data.append({
|
||||
"user": msg,
|
||||
"ia": ia_reponse,
|
||||
"nom_affichage": nom_affichage,
|
||||
"info_fichiers": info_fichiers,
|
||||
"info_local": info_local
|
||||
})
|
||||
session_data.append({"user": msg, "ia": ia_reponse, "nom_affichage": nom_affichage, "info_fichiers": info_fichiers, "info_local": info_local})
|
||||
save_session_data(session_id, session_data)
|
||||
log_event(session_id, "SYSTEM", "Réponse sauvegardée avec succès.")
|
||||
|
||||
chat_bubble = Div(
|
||||
P("Vous", cls="msg-user"), Div(NotStr(f"{msg}{info_fichiers}{info_local}"), cls="bubble-user"),
|
||||
P(f"AETHAS38 ({nom_affichage})", cls="msg-ia"), Div(ia_reponse, cls="bubble-ia")
|
||||
)
|
||||
|
||||
updated_budget = P(get_budget(), id="budget-display", style="color:#10b981; font-size:18px; font-weight:bold;", hx_swap_oob="true")
|
||||
return chat_bubble, updated_budget
|
||||
chat_bubble = Div(P("Vous", cls="msg-user"), Div(NotStr(f"{msg}{info_fichiers}{info_local}"), cls="bubble-user"), P(f"AETHAS38 ({nom_affichage})", cls="msg-ia"), Div(ia_reponse, cls="bubble-ia"))
|
||||
return chat_bubble, P(get_budget(), id="budget-display", style="color:#10b981; font-size:18px; font-weight:bold;", hx_swap_oob="true")
|
||||
|
||||
if __name__ == '__main__':
|
||||
serve(port=5001)
|
||||
Reference in new issue
Block a user