<?php
session_start();

require_once __DIR__ . '/../ai/LocalAIApi.php';
require_once __DIR__ . '/../db/config.php';

header('Content-Type: application/json');

$questions = [
    "I wake up feeling tired, even after a full night’s sleep.",
    "By midday, I already feel mentally drained and out of energy.",
    "I feel emotionally numb, detached, or ‘on autopilot’ most of the time.",
    "Things that used to excite me now feel pointless or like a chore.",
    "I feel irritated or cynical about people I work or study with.",
    "I struggle to focus and constantly procrastinate, even on important tasks.",
    "I feel guilty for not doing ‘enough’, no matter how much I actually do.",
    "I often think about quitting everything for a while or disappearing from social media and work.",
    "I use caffeine, sugar, nicotine, alcohol, or scrolling to ‘numb out’ instead of resting.",
    "I feel like my life is just surviving, not really living"
];

if (!isset($_SESSION['question_index'])) {
    $_SESSION['question_index'] = 0;
    $_SESSION['answers'] = [];
}

$response = [];

if ($_SERVER['REQUEST_METHOD'] === 'POST') {
    $data = json_decode(file_get_contents('php://input'), true);
    $answer = $data['answer'] ?? null;

    if ($answer !== null && $_SESSION['question_index'] < count($questions)) {
        $_SESSION['answers'][$_SESSION['question_index']] = $answer;
        $_SESSION['question_index']++;
    }
}

if ($_SESSION['question_index'] < count($questions)) {
    $response['question'] = $questions[$_SESSION['question_index']];
} else {
    // Survey is complete, analyze the answers
    $formatted_answers = [];
    foreach ($_SESSION['answers'] as $index => $answer) {
        $formatted_answers[] = "Q: " . $questions[$index] . "\nA: " . $answer;
    }
    $prompt = "You are a burnout analysis expert. A user has completed a burnout survey. The user answered on a scale of 1-5 where 1 is 'Never' and 5 is 'Always'. Analyze the following survey answers and provide a detailed analysis.\n\nReturn the response as a JSON object. The JSON object must have the following keys: \"scores\", \"analysis\", \"recommendations\", and \"nextSteps\".\n\n- The 'scores' key should contain an object with 'Exhaustion', 'Cynicism', and 'Inefficacy' as keys, each with a numeric score from 0 to 5.\n- The 'analysis' key should contain an object with 'overallSummary', 'exhaustionSummary', 'cynicismSummary', and 'inefficacySummary' as keys, each containing a string with the corresponding analysis.\n- The 'recommendations' key should contain an array of objects, where each object has a 'title' and a 'description'.\n- The 'nextSteps' key should contain an array of strings.\n\nAnswers:";
$prompt .= implode("\n\n", $formatted_answers);

    $ai_response = LocalAIApi::createResponse([
        'input' => [
            ['role' => 'system', 'content' => 'You are a burnout analysis expert.'],
            ['role' => 'user', 'content' => $prompt],
        ],
    ]);

    if (!empty($ai_response['success'])) {
        $decoded_response = LocalAIApi::decodeJsonFromResponse($ai_response);
        $_SESSION['results'] = $decoded_response;

        if (isset($_SESSION['user_id'])) {
            try {
                $pdo = db();
                $stmt = $pdo->prepare('INSERT INTO survey_results (user_id, results_json) VALUES (:user_id, :results_json)');
                $stmt->execute([
                    ':user_id' => $_SESSION['user_id'],
                    ':results_json' => json_encode($decoded_response)
                ]);
            } catch (PDOException $e) {
                // Optionally handle or log the database error
            }
        }
    } else {
        // Handle AI API error, maybe provide default results
        $_SESSION['results'] = [
            'scores' => ['Exhaustion' => 0, 'Cynicism' => 0, 'Inefficacy' => 0],
            'recommendations' => ['Could not analyze results at this time. Please try again later.']
        ];
    }
    
    $response['redirect'] = 'results.php';
}

echo json_encode($response);