import asyncio
import json
import os.path
import threading
from typing import Dict, List
from openai import OpenAI
import aiohttp
import qqbot


from qqbot.core.util.yaml_util import YamlUtil
from qqbot.model.message import MessageEmbed, MessageEmbedField, MessageEmbedThumbnail, CreateDirectMessageRequest, MessageArk, MessageArkKv, MessageArkObj, MessageArkObjKv

def load_openai_key(filepath="openai_key.txt"):
    with open(filepath, 'r') as f:
        return f.read().strip()

openai_key = load_openai_key()
test_config = YamlUtil.read(os.path.join(os.path.dirname(__file__), "config.yaml"))

clinet = OpenAI(api_key=openai_key)

gpt_instruction = YamlUtil.read(os.path.join(os.path.dirname(__file__), "gpt_instruction.yaml"))

async def _message_handler(event, message: qqbot.Message):
    msg_api = qqbot.AsyncMessageAPI(t_token, False)
    qqbot.logger.info("event %s" % event + ",receive message %s" % message.content)
    response = clinet.chat.completions.create(
        model="gpt-3.5-turbo-0125",
        messages=[
            {"role": "system", "content": gpt_instruction},
            {"role": "user", "content": message.content}
        ]
    )
    reply = response.choices[0].message
    print(reply)
    message_to_send = qqbot.MessageSendRequest(content=reply.content, msg_id=message.id)
    await msg_api.post_message(message.channel_id, message_to_send)

#write a funtion for the bot: when a message begins with "/copy", the bot will copy the message and reply it.

if __name__ == "__main__":
    t_token = qqbot.Token(test_config["token"]["appid"], test_config["token"]["token"])
    qqbot_handler = qqbot.Handler(
        qqbot.HandlerType.AT_MESSAGE_EVENT_HANDLER, _message_handler
    )

    qqbot.async_listen_events(t_token, False, qqbot_handler)