<?php

require_once __DIR__ . '/vendor/autoload.php';

use Classifier\Lib\MessageQualifier;

require_once __DIR__ . '/Lib/MessageQualifier.php';


$samples = [
    // --- Catégorie ALERTE ---
    ["Urgent : Incendie majeur détecté zone industrielle. Évacuez."],
    ["Alerte inondation : la rivière déborde, montez dans les étages."],
    ["Alerte attentat : restez confinés chez vous, évitez les déplacements."],
    ["Fuite de gaz signalée rue de la Paix. Périmètre de sécurité en cours."],
    ["ALERTE : Risque d'effondrement bâtiment mairie. Ne pas approcher."],
    ["Incident chimique usine Sud. Fermez les fenêtres et bouchez les aérations."],

    // --- Catégorie INFO (Le "bruit" quotidien) ---
    ["Le ramassage des ordures sera décalé à mardi prochain."],
    ["Inauguration de la nouvelle bibliothèque ce samedi à 14h."],
    ["Information : Travaux sur la voirie prévus du 12 au 15 mars."],
    ["La coupure d'eau est terminée, merci de votre patience."],
    ["Rappel : Pensez à inscrire vos enfants à la cantine scolaire."],
    ["Le montant de votre allocation a été mis à jour dans votre espace."],
    ["Ceci est un test technique du système d'alerte, aucune action requise."],
];

$labels = [
    'ALERTE',
    'ALERTE',
    'ALERTE',
    'ALERTE',
    'ALERTE',
    'ALERTE',
    'INFO',
    'INFO',
    'INFO',
    'INFO',
    'INFO',
    'INFO',
    'INFO'
];

$classifier = new MessageQualifier();
$classifier->train($samples, $labels);

echo "Le cerveau de l'IA est prêt !";