# flake8: noqa
import asyncio

from aiogram import Bot
from aiogram.types import Message

from ocab_modules.external.yandexgpt.yandexgpt import *
from ocab_modules.standard.config.config import (
    get_yandexgpt_catalog_id,
    get_yandexgpt_prompt,
    get_yandexgpt_token,
)
from ocab_modules.standard.database.db_api import add_message


async def answer_to_message(message: Message, bot: Bot):
    # print("answer_to_message")
    await log("answer_to_message")
    yagpt = YandexGPT(get_yandexgpt_token(), get_yandexgpt_catalog_id())
    text = message.text
    prompt = get_yandexgpt_prompt()
    # response = await yagpt.async_yandexgpt(system_prompt=prompt, input_messages=text)
    response = await yagpt.yandexgpt_request(
        chat_id=message.chat.id, message_id=message.message_id, type="yandexgpt"
    )
    reply = await message.reply(response, parse_mode="Markdown")
    add_message(reply, message_ai_model="yandexgpt")