from pyrogram import filters
from pyrogram.errors import MessageNotModified, MessageTooLong

from misskaty import app
from misskaty.helper import http, post_to_telegraph
from misskaty.core.message_utils import *
from misskaty.core.decorator.ratelimiter import ratelimiter
from misskaty.vars import COMMAND_HANDLER, OPENAI_API


@app.on_message(filters.command("ask", COMMAND_HANDLER))
@ratelimiter
async def chatbot(c, m):
    if len(m.command) == 1:
        return await kirimPesan(m, f"Please use command <code>/{m.command[0]} [question]</code> to ask your question.")
    pertanyaan = m.text.split(" ", maxsplit=1)[1]
    headers = {
        "Content-Type": "application/json",
        "Authorization": f"Bearer {OPENAI_API}",
    }

    json_data = {
        "model": "gpt-3.5-turbo",
        "messages": [
            {
                "role": "user",
                "content": pertanyaan,
            },
        ],
    }
    msg = await kirimPesan(m, "Wait a moment looking for your answer..")
    try:
        response = (await http.post("https://api.openai.com/v1/chat/completions", headers=headers, json=json_data)).json()
        if err := response["error"]:
            return await editPesan(msg, err["message"])
        answer = response["choices"][0]["message"]["content"]
        await editPesan(msg, answer)
    except MessageTooLong:
        answerlink = await post_to_telegraph(False, "MissKaty ChatBot ", answer)
        await editPesan(msg, f"Question for your answer has exceeded TG text limit, check this link to view.\n\n{answerlink}", disable_web_page_preview=True)
    except (MessageNotModified, MessageIdInvalid):
        pass
    except Exception as err:
        await editPesan(msg, f"Oppss. ERROR: {str(err)}")