"""
AI module for Perplexity API integration.
Sends user queries and receives AI responses.
"""

import requests
from config import PERPLEXITY_API_KEY, PERPLEXITY_MODEL, PERPLEXITY_API_URL


# System prompt for the AI
SYSTEM_PROMPT = """Ты — Александр, умный голосовой ассистент с человеческим поведением.
Веди себя как живой человек: будь дружелюбным, естественным и немного эмоциональным, где это уместно.
Твоя главная цель — помогать пользователю и поддерживать интересный диалог.
Отвечай кратко и по существу, на русском языке.
Избегай длинных списков, сложного форматирования и спецсимволов, так как твои ответы озвучиваются голосом.
Пиши в разговорном стиле, как при живом общении, но не забывай о вежливости и правильности твоих ответов."""


def ask_ai(messages_history: list) -> str:
    """
    Send a message history to Perplexity AI and get a response.

    Args:
        messages_history: List of dictionaries with role and content
                         e.g., [{"role": "user", "content": "Hi"}]

    Returns:
        AI response text
    """
    if not messages_history:
        return "Извините, я не расслышал вашу команду."

    # Extract the last user message for logging
    last_user_message = next(
        (m["content"] for m in reversed(messages_history) if m["role"] == "user"),
        "Unknown",
    )
    print(f"🤖 Запрос к AI: {last_user_message}")

    headers = {
        "Authorization": f"Bearer {PERPLEXITY_API_KEY}",
        "Content-Type": "application/json",
    }

    # Prepend system prompt to the history
    messages = [{"role": "system", "content": SYSTEM_PROMPT}] + list(messages_history)

    payload = {
        "model": PERPLEXITY_MODEL,
        "messages": messages,
        "max_tokens": 500,
        "temperature": 1.0,
    }

    try:
        response = requests.post(
            PERPLEXITY_API_URL, headers=headers, json=payload, timeout=30
        )
        response.raise_for_status()

        data = response.json()
        ai_response = data["choices"][0]["message"]["content"]
        print(f"💬 Ответ AI: {ai_response[:100]}...")
        return ai_response

    except requests.exceptions.Timeout:
        return "Извините, сервер не отвечает. Попробуйте позже."
    except requests.exceptions.RequestException as e:
        print(f"❌ Ошибка API: {e}")
        return "Произошла ошибка при обращении к AI. Попробуйте ещё раз."
    except (KeyError, IndexError) as e:
        print(f"❌ Ошибка парсинга ответа: {e}")
        return "Не удалось обработать ответ от AI."