Loki/backend/app/skills.py

"""Skills : méthodes expertes injectées automatiquement selon la tâche.

Sélection lexicale instantanée (aucun appel LLM) : au plus UNE skill par
message, injectée en message système pour ce tour uniquement.
"""
from __future__ import annotations

import os
import re

_SKILLS_DIR = os.path.join(os.path.dirname(__file__), "..", "skills")

# Ponytail (https://github.com/DietrichGebert/ponytail) : philosophie de code
# « paresseux » (anti sur-ingénierie). Adaptée en méthode transverse injectée
# pour toute tâche de code — indépendante du sélecteur de skill (mono-skill).
PONYTAIL_TITLE = "Ponytail · code minimal"
PONYTAIL_GUIDANCE = (
    "Méthode Ponytail — code minimal, anti sur-ingénierie. Avant d'écrire du "
    "code, descends l'échelle de décision et arrête-toi au premier échelon qui "
    "suffit :\n"
    "1. NE PAS coder ce qui n'est pas explicitement demandé (YAGNI) ;\n"
    "2. RÉUTILISER l'existant (fichiers/fonctions déjà présents) ;\n"
    "3. UTILISER les fonctions natives du langage / du navigateur ;\n"
    "4. en DERNIER recours seulement, écrire le minimum de code nécessaire.\n"
    "Livre la solution la plus simple qui fonctionne : aucune dépendance ni "
    "bibliothèque à installer, aucune abstraction prématurée, aucune "
    "fonctionnalité en plus non demandée (pas de « moteur IA », d'options ou de "
    "configuration superflues). Préfère un seul fichier clair à une "
    "architecture élaborée."
)

# Seuil : nombre minimal de mots-clés distincts trouvés dans le message.
_MIN_HITS = 2


def _load() -> dict[str, dict]:
    out: dict[str, dict] = {}
    if not os.path.isdir(_SKILLS_DIR):
        return out
    for fname in sorted(os.listdir(_SKILLS_DIR)):
        if not fname.endswith(".md"):
            continue
        with open(os.path.join(_SKILLS_DIR, fname), encoding="utf-8") as f:
            raw = f.read()
        m = re.match(r"^---\n(.*?)\n---\n(.*)$", raw, re.S)
        if not m:
            continue
        meta: dict[str, str] = {}
        for line in m.group(1).splitlines():
            if ":" in line:
                key, _, value = line.partition(":")
                meta[key.strip()] = value.strip()
        keywords = [
            k.strip().lower()
            for k in meta.get("keywords", "").split(",")
            if k.strip()
        ]
        out[meta.get("name", fname[:-3])] = {
            "name": meta.get("name", fname[:-3]),
            "title": meta.get("title", fname[:-3]),
            "keywords": keywords,
            "body": m.group(2).strip(),
        }
    return out


_WEB_RE = re.compile(
    r"\b(html|css|javascript|js|page|site|web|appli|application|interface|"
    r"bouton|formulaire|canvas|animation|jeu|game|échiquier|echiquier|"
    r"dashboard|landing|maquette|ui|front)\b",
    re.I,
)


def is_web_task(message: str) -> bool:
    """La demande porte-t-elle sur une page / appli web ?"""
    return bool(_WEB_RE.search(message or ""))


# Contraintes « appli web » : le sandbox de prévisualisation est HORS-LIGNE et
# isolé (pas de CDN, pas de WASM téléchargé). Sans ces règles, les modèles
# promettent des libs (Stockfish, React…) qui ne se chargent jamais, éclatent
# le code en plusieurs fichiers incohérents, puis déclarent « ça marche ».
WEBAPP_GUIDANCE = (
    "Contraintes STRICTES pour une page/appli web :\n"
    "1. UN SEUL fichier index.html AUTONOME : tout le CSS dans <style>, tout le "
    "JS dans <script> à la fin du <body>. Pas de fichiers .css/.js séparés.\n"
    "2. AUCUNE ressource externe : pas de CDN, pas de <script src=\"https://…\">, "
    "pas de bibliothèque à télécharger (jQuery, React, chess.js, Stockfish, "
    ".wasm…). Le sandbox est hors-ligne : ça ne se charge JAMAIS. Si une "
    "fonctionnalité a besoin d'une lib, écris une version simple toi-même en "
    "JavaScript natif.\n"
    "3. Le rendu doit s'afficher AU CHARGEMENT : construis réellement le DOM "
    "dans un script qui s'exécute (ex. génère les 64 cases de l'échiquier en "
    "boucle). Pas de fonction jamais appelée, pas de conteneur laissé vide.\n"
    "4. Ne prétends JAMAIS qu'une fonctionnalité marche sans l'avoir codée, et "
    "ne « simule » pas un moteur en le faisant passer pour réel : dis "
    "clairement ce qui est réel et ce qui ne l'est pas.\n"
    "5. Design sobre, lisible et moderne — mais la priorité est que ça "
    "FONCTIONNE vraiment."
)


ALL_SKILLS: dict[str, dict] = _load()


def pick_skill(message: str) -> dict | None:
    """Meilleure skill pour ce message, ou None si rien d'assez net."""
    low = message.lower()
    best, best_hits = None, 0
    for skill in ALL_SKILLS.values():
        hits = sum(1 for kw in skill["keywords"] if kw in low)
        if hits > best_hits:
            best, best_hits = skill, hits
    if best is None or best_hits < _MIN_HITS:
        return None
    return {"name": best["name"], "title": best["title"], "body": best["body"]}