<?php
// ==========================================
// تنظیمات پایه و اتصال به پایگاه داده SQLite
// ==========================================
define('AVALAI_API_KEY', 'aa-ivmdM4GeUcrYDCY51JJ4OQNLMLJrMAT3xgmECCI3O9h2AGi2'); // کلید API خود را اینجا وارد کنید
define('API_ENDPOINT', 'https://api.avalai.ir/v1/chat/completions');

session_start();

$db_file = __DIR__ . '/database.sqlite';
try {
    $db = new PDO("sqlite:" . $db_file);
    $db->setAttribute(PDO::ATTR_ERRMODE, PDO::ERRMODE_EXCEPTION);

    // ایجاد جداول
    $db->exec("CREATE TABLE IF NOT EXISTS users (
        id INTEGER PRIMARY KEY AUTOINCREMENT,
        username TEXT UNIQUE,
        password TEXT
    )");

    $db->exec("CREATE TABLE IF NOT EXISTS chats (
        id INTEGER PRIMARY KEY AUTOINCREMENT,
        user_id INTEGER,
        role TEXT,
        content TEXT,
        created_at DATETIME DEFAULT CURRENT_TIMESTAMP
    )");
} catch (PDOException $e) {
    die("خطا در اتصال به پایگاه داده: " . htmlspecialchars($e->getMessage()));
}

// خروج کاربر
if (isset($_GET['logout'])) {
    session_destroy();
    header("Location: index.php");
    exit;
}

$error = '';
$success = '';

// ==========================================
// پردازش فرم‌ها (ثبت‌نام، ورود، ارسال پیام)
// ==========================================
if ($_SERVER['REQUEST_METHOD'] === 'POST') {
    $action = $_POST['action'] ?? '';

    // ۱. ثبت‌نام
    if ($action === 'register') {
        $username = trim($_POST['username'] ?? '');
        $password = $_POST['password'] ?? '';

        if (empty($username) || empty($password)) {
            $error = 'لطفاً تمامی فیلدها را پر کنید.';
        } else {
            $hashed = password_hash($password, PASSWORD_DEFAULT);
            try {
                $stmt = $db->prepare("INSERT INTO users (username, password) VALUES (?, ?)");
                $stmt->execute([$username, $hashed]);
                $success = 'ثبت‌نام با موفقیت انجام شد. اکنون وارد شوید.';
            } catch (Exception $e) {
                $error = 'نام کاربری قبلاً ثبت شده است.';
            }
        }
    }

    // ۲. ورود
    elseif ($action === 'login') {
        $username = trim($_POST['username'] ?? '');
        $password = $_POST['password'] ?? '';

        $stmt = $db->prepare("SELECT * FROM users WHERE username = ?");
        $stmt->execute([$username]);
        $user = $stmt->fetch(PDO::FETCH_ASSOC);

        if ($user && password_verify($password, $user['password'])) {
            $_SESSION['user_id'] = $user['id'];
            $_SESSION['username'] = $user['username'];
            header("Location: index.php");
            exit;
        } else {
            $error = 'نام کاربری یا رمز عبور اشتباه است.';
        }
    }

    // ۳. ارسال پیام به هوش مصنوعی
    elseif ($action === 'ask' && isset($_SESSION['user_id'])) {
        $userPrompt = trim($_POST['message'] ?? '');
        $selectedModel = $_POST['model'] ?? 'gpt-4o-mini';

        if (!empty($userPrompt)) {
            // ذخیره پیام کاربر
            $stmt = $db->prepare("INSERT INTO chats (user_id, role, content) VALUES (?, 'user', ?)");
            $stmt->execute([$_SESSION['user_id'], $userPrompt]);

            // ارتباط با AvalAI API
            $payload = [
                'model' => $selectedModel,
                'messages' => [
                    ['role' => 'user', 'content' => $userPrompt]
                ]
            ];

            $ch = curl_init(API_ENDPOINT);
            curl_setopt_array($ch, [
                CURLOPT_RETURNTRANSFER => true,
                CURLOPT_POST => true,
                CURLOPT_POSTFIELDS => json_encode($payload, JSON_UNESCAPED_UNICODE),
                CURLOPT_HTTPHEADER => [
                    'Content-Type: application/json',
                    'Authorization: Bearer ' . AVALAI_API_KEY
                ],
                CURLOPT_TIMEOUT => 60
            ]);

            $response = curl_exec($ch);
            $result = json_decode($response, true);
            curl_close($ch);

            $aiReply = $result['choices'][0]['message']['content'] ?? 'خطا در دریافت پاسخ از هوش مصنوعی.';

            // ذخیره پاسخ هوش مصنوعی
            $stmt = $db->prepare("INSERT INTO chats (user_id, role, content) VALUES (?, 'assistant', ?)");
            $stmt->execute([$_SESSION['user_id'], $aiReply]);

            header("Location: index.php");
            exit;
        }
    }
}

// دریافت تاریخچه پیام‌های کاربر
$history = [];
if (isset($_SESSION['user_id'])) {
    $stmt = $db->prepare("SELECT * FROM chats WHERE user_id = ? ORDER BY created_at ASC");
    $stmt->execute([$_SESSION['user_id']]);
    $history = $stmt->fetchAll(PDO::FETCH_ASSOC);
}
?>
<!DOCTYPE html>
<html lang="fa" dir="rtl">
<head>
    <meta charset="UTF-8">
    <meta name="viewport" content="width=device-width, initial-scale=1.0">
    <title>پنل هوشمند چت‌بات</title>
    <script src="https://cdn.tailwindcss.com"></script>
    <style>
        body { font-family: system-ui, -apple-system, sans-serif; }
    </style>
</head>
<body class="bg-gray-100 min-h-screen flex items-center justify-center p-4">

<div class="w-full max-w-2xl bg-white rounded-2xl shadow-xl overflow-hidden">

    <?php if (!empty($error)): ?>
        <div class="bg-red-50 text-red-600 p-3 text-center text-sm border-b border-red-100"><?php echo htmlspecialchars($error); ?></div>
    <?php endif; ?>

    <?php if (!empty($success)): ?>
        <div class="bg-green-50 text-green-600 p-3 text-center text-sm border-b border-green-100"><?php echo htmlspecialchars($success); ?></div>
    <?php endif; ?>

    <?php if (!isset($_SESSION['user_id'])): ?>
        <!-- فرم ورود و ثبت‌نام -->
        <div class="p-8">
            <h2 class="text-2xl font-bold text-center text-gray-800 mb-6">ورود / ثبت‌نام</h2>
            <form method="POST" class="space-y-4">
                <div>
                    <label class="block text-sm font-medium text-gray-700 mb-1">نام کاربری</label>
                    <input type="text" name="username" required class="w-full px-4 py-2 border rounded-lg focus:ring-2 focus:ring-indigo-500 focus:outline-none">
                </div>
                <div>
                    <label class="block text-sm font-medium text-gray-700 mb-1">رمز عبور</label>
                    <input type="password" name="password" required class="w-full px-4 py-2 border rounded-lg focus:ring-2 focus:ring-indigo-500 focus:outline-none">
                </div>
                <div class="flex gap-3 pt-2">
                    <button type="submit" name="action" value="login" class="flex-1 bg-indigo-600 hover:bg-indigo-700 text-white font-medium py-2 rounded-lg transition">ورود</button>
                    <button type="submit" name="action" value="register" class="flex-1 bg-slate-600 hover:bg-slate-700 text-white font-medium py-2 rounded-lg transition">ثبت‌نام کاربر جدید</button>
                </div>
            </form>
        </div>

    <?php else: ?>
        <!-- پنل چت هوش مصنوعی -->
        <div class="p-6">
            <div class="flex justify-between items-center border-b pb-4 mb-4">
                <span class="font-medium text-gray-700">خوش آمدید، <b><?php echo htmlspecialchars($_SESSION['username']); ?></b></span>
                <a href="?logout=1" class="text-sm text-red-500 hover:underline">خروج</a>
            </div>

            <!-- نمایش پیام‌ها -->
            <div class="h-96 overflow-y-auto space-y-4 p-4 bg-gray-50 rounded-xl mb-4 border">
                <?php if (empty($history)): ?>
                    <p class="text-center text-gray-400 my-auto pt-36">هنوز پیامی ندارید. گفتگو را شروع کنید!</p>
                <?php else: ?>
                    <?php foreach ($history as $msg): ?>
                        <?php if ($msg['role'] === 'user'): ?>
                            <div class="flex justify-end">
                                <div class="bg-indigo-600 text-white px-4 py-2 rounded-2xl rounded-br-none max-w-[80%] text-sm">
                                    <?php echo nl2br(htmlspecialchars($msg['content'])); ?>
                                </div>
                            </div>
                        <?php else: ?>
                            <div class="flex justify-start">
                                <div class="bg-white border text-gray-800 px-4 py-2 rounded-2xl rounded-bl-none max-w-[80%] text-sm shadow-sm">
                                    <?php echo nl2br(htmlspecialchars($msg['content'])); ?>
                                </div>
                            </div>
                        <?php endif; ?>
                    <?php endforeach; ?>
                <?php endif; ?>
            </div>

            <!-- فرم ارسال پیام -->
            <form method="POST" class="space-y-3">
                <input type="hidden" name="action" value="ask">
                <select name="model" class="w-full px-3 py-2 border rounded-lg text-sm bg-white focus:outline-none focus:ring-2 focus:ring-indigo-500">
                    <option value="gpt-4o">GPT-4o (اصلی)</option>
                    <option value="gpt-4o-mini" selected>GPT-4o Mini (سریع و به‌صرفه)</option>
                    <option value="claude-3-5-sonnet-20241022">Claude 3.5 Sonnet</option>
                </select>

                <textarea name="message" rows="2" placeholder="پیام خود را بنویسید..." required class="w-full p-3 border rounded-lg focus:ring-2 focus:ring-indigo-500 focus:outline-none text-sm resize-none"></textarea>

                <button type="submit" class="w-full bg-indigo-600 hover:bg-indigo-700 text-white font-medium py-2 rounded-lg transition">ارسال به هوش مصنوعی</button>
            </form>
        </div>
    <?php endif; ?>

</div>

</body>
</html>