<?php
require_once __DIR__ . '/config.php';

function getPDO() {
    return getPDO_instance(); // o il tuo getPDO attuale
}

// 1) Recupero i nuovi infortuni da classificare
$pdo = getPDO();

$anno = isset($_GET['anno']) ? (int)$_GET['anno'] : (int)date('Y');

// Limita il numero per chiamata (es. max 10 alla volta)
$maxPerBatch = 10;

// Se è stato premuto il pulsante di classificazione
$log = [];
if ($_SERVER['REQUEST_METHOD'] === 'POST' && isset($_POST['action']) && $_POST['action'] === 'classifica') {
    // prendo fino a $maxPerBatch infortuni senza processo
    $sql = "
        SELECT 
            e.id,
            e.data_infortunio,
            e.tipo,
            e.gravita,
            e.area,
            e.mansione,
            e.descrizione_evento,
            c.nome_centro
        FROM infortuni_estesi e
        LEFT JOIN centro_raccolta c ON e.id_centro = c.id_centro
        WHERE e.data_infortunio BETWEEN :da AND :a
          AND (e.processo IS NULL OR e.processo = '')
        ORDER BY e.data_infortunio
        LIMIT :lim
    ";

    $stmt = $pdo->prepare($sql);
    $data_da = $anno . "-01-01";
    $data_a  = $anno . "-12-31";
    $stmt->bindValue(':da', $data_da);
    $stmt->bindValue(':a', $data_a);
    $stmt->bindValue(':lim', $maxPerBatch, PDO::PARAM_INT);
    $stmt->execute();
    $infortuni = $stmt->fetchAll(PDO::FETCH_ASSOC);

    if ($infortuni) {
        foreach ($infortuni as $inf) {
            $risultato = chiamaChatGPT_per_infortunio($inf);
            if ($risultato && isset($risultato['processo'])) {
                // aggiorno DB
                $upd = $pdo->prepare("
                    UPDATE infortuni_estesi
                    SET 
                        processo = :processo,
                        famiglia_rischio = :famiglia_rischio,
                        dinamica_codificata = :dinamica_codificata,
                        fattore_rischio_principale = :fattore_rischio_principale
                    WHERE id = :id
                ");
                $upd->execute([
                    ':processo'                   => $risultato['processo'],
                    ':famiglia_rischio'           => $risultato['famiglia_rischio'],
                    ':dinamica_codificata'        => $risultato['dinamica_codificata'],
                    ':fattore_rischio_principale' => $risultato['fattore_rischio_principale'],
                    ':id'                         => $inf['id']
                ]);
                $log[] = "Infortunio ID {$inf['id']} classificato come [{$risultato['processo']} / {$risultato['famiglia_rischio']}]";
            } else {
                $log[] = "Infortunio ID {$inf['id']} – nessuna risposta valida dall'IA.";
            }
        }
    } else {
        $log[] = "Nessun infortunio da classificare per l'anno $anno.";
    }
}

// Conteggio quanti rimangono da classificare
$stmt = $pdo->prepare("
    SELECT COUNT(*) 
    FROM infortuni_estesi
    WHERE data_infortunio BETWEEN :da AND :a
      AND (processo IS NULL OR processo = '')
");
$stmt->execute([
    ':da' => $anno . "-01-01",
    ':a'  => $anno . "-12-31"
]);
$daClassificare = (int)$stmt->fetchColumn();

/**
 * Funzione che chiama ChatGPT API per un singolo infortunio
 */
function chiamaChatGPT_per_infortunio(array $inf) {
    $prompt = build_prompt($inf);

    $apiKey = OPENAI_API_KEY;
    $url = "https://api.openai.com/v1/chat/completions";

    $body = [
        "model" => "gpt-4.1-mini", // o il modello che vuoi usare
        "messages" => [
            [
                "role" => "system",
                "content" => "Sei un assistente per la sicurezza sul lavoro. Data la descrizione di un infortunio in azienda di raccolta rifiuti, devi classificare il caso restituendo un JSON con le chiavi: processo, famiglia_rischio, dinamica_codificata, fattore_rischio_principale. Scrivi solo JSON valido."
            ],
            [
                "role" => "user",
                "content" => $prompt
            ]
        ],
        "temperature" => 0.2
    ];

    $ch = curl_init($url);
    curl_setopt_array($ch, [
        CURLOPT_RETURNTRANSFER => true,
        CURLOPT_HTTPHEADER => [
            "Content-Type: application/json",
            "Authorization: Bearer " . $apiKey
        ],
        CURLOPT_POST => true,
        CURLOPT_POSTFIELDS => json_encode($body),
        CURLOPT_TIMEOUT => 30
    ]);

    $response = curl_exec($ch);
    if ($response === false) {
        return null;
    }
    $status = curl_getinfo($ch, CURLINFO_HTTP_CODE);
    curl_close($ch);

    if ($status < 200 || $status >= 300) {
        return null;
    }

    $data = json_decode($response, true);
    if (!isset($data['choices'][0]['message']['content'])) {
        return null;
    }

    $content = $data['choices'][0]['message']['content'];

    // Il modello risponde con JSON, lo decodifichiamo
    $json = json_decode($content, true);
    if (!$json) {
        // se per qualche motivo risponde testo normale, potresti fare parsing,
        // ma per ora torniamo null
        return null;
    }

    return [
        'processo'                   => $json['processo'] ?? null,
        'famiglia_rischio'           => $json['famiglia_rischio'] ?? null,
        'dinamica_codificata'        => $json['dinamica_codificata'] ?? null,
        'fattore_rischio_principale' => $json['fattore_rischio_principale'] ?? null,
    ];
}

function build_prompt(array $inf) {
    $txt = "Ti fornisco i dati di un infortunio in un'azienda che si occupa di raccolta e gestione rifiuti.\n";
    $txt .= "Devi classificare il caso scegliendo:\n";
    $txt .= "- processo tra: 'servizi esterni', 'centro raccolta', 'officina', 'sede aziendale', 'centro di trasferenza'\n";
    $txt .= "- famiglia_rischio (esempi: scivolamento/inciampo, movimentazione manuale carichi, schiacciamento, incidente stradale/investimento, corpo estraneo nell'occhio, urto/schiacciamento, puntura/taglio, morso animale, ecc.)\n";
    $txt .= "- dinamica_codificata (riassunto in 5–15 parole)\n";
    $txt .= "- fattore_rischio_principale (es. meteo/ghiaccio, appoggio instabile, coordinamento squadra, organizzazione, DPI, traffico veicolare, ecc.).\n\n";

    $txt .= "Dati caso:\n";
    $txt .= "ID: {$inf['id']}\n";
    $txt .= "Data: {$inf['data_infortunio']}\n";
    $txt .= "Centro: {$inf['nome_centro']}\n";
    $txt .= "Tipo evento: {$inf['tipo']}\n";
    $txt .= "Gravità: {$inf['gravita']}\n";
    $txt .= "Area: {$inf['area']}\n";
    $txt .= "Mansione: {$inf['mansione']}\n";
    $txt .= "Descrizione: {$inf['descrizione_evento']}\n\n";

    $txt .= "Rispondi SOLO con un JSON come questo esempio:\n";
    $txt .= "{\n";
    $txt .= "  \"processo\": \"servizi esterni\",\n";
    $txt .= "  \"famiglia_rischio\": \"scivolamento/inciampo\",\n";
    $txt .= "  \"dinamica_codificata\": \"Scende dal mezzo e distorce la caviglia\",\n";
    $txt .= "  \"fattore_rischio_principale\": \"appoggio instabile/discesa dal mezzo\"\n";
    $txt .= "}\n";

    return $txt;
}

?>
<!DOCTYPE html>
<html lang="it">
<head>
    <meta charset="UTF-8">
    <title>Classificazione automatica infortuni</title>
    <link href="https://cdn.jsdelivr.net/npm/bootstrap@5.3.2/dist/css/bootstrap.min.css" rel="stylesheet">
</head>
<body class="bg-light">
<nav class="navbar navbar-dark bg-dark mb-4">
    <div class="container-fluid">
        <span class="navbar-brand">Classificazione automatica infortuni <?php echo htmlspecialchars($anno); ?></span>
        <a href="dashboard.php" class="btn btn-outline-light btn-sm">← Dashboard</a>
    </div>
</nav>

<div class="container">
    <div class="card mb-4">
        <div class="card-body">
            <form class="row g-3" method="get">
                <div class="col-auto">
                    <label for="anno" class="col-form-label">Anno:</label>
                </div>
                <div class="col-auto">
                    <input type="number" class="form-control" id="anno" name="anno"
                           value="<?php echo htmlspecialchars($anno); ?>">
                </div>
                <div class="col-auto">
                    <button type="submit" class="btn btn-secondary">Cambia anno</button>
                </div>
            </form>
            <hr>
            <p class="mb-1">
                Infortuni dell'anno <?php echo htmlspecialchars($anno); ?> <strong>da classificare</strong>: 
                <strong><?php echo $daClassificare; ?></strong>
            </p>
            <p class="text-muted mb-2">
                Ad ogni clic verranno classificati al massimo <?php echo $maxPerBatch; ?> infortuni (per contenere l’uso dell’API).
            </p>
            <form method="post">
                <input type="hidden" name="action" value="classifica">
                <button type="submit" class="btn btn-primary"
                    <?php echo $daClassificare === 0 ? 'disabled' : ''; ?>>
                    Classifica un lotto di infortuni ora
                </button>
            </form>
        </div>
    </div>

    <?php if (!empty($log)): ?>
        <div class="card">
            <div class="card-header">
                Log operazione
            </div>
            <div class="card-body">
                <ul class="mb-0">
                    <?php foreach ($log as $riga): ?>
                        <li><?php echo htmlspecialchars($riga); ?></li>
                    <?php endforeach; ?>
                </ul>
            </div>
        </div>
    <?php endif; ?>
</div>
</body>
</html>