import os
from pathlib import Path
from models.generator import Generator
from models.emoji_db import check_emoji, update_description
from controllers.localfile_handles import get_emoji_count, yield_emoji_path
from controllers.compress_image import compress_image
from models.logger import setup_logger

logger = setup_logger()

emoji_folders = "D:\\Python\\mcunc\\EmojiTextGenerator\\emojis\\"
emoji = yield_emoji_path(emoji_folders)

api_key = os.getenv('DASHSCOPE_API_KEY')

def main():
    gen = Generator(api_key)
    for f in range(get_emoji_count(emoji_folders)):
        emoji_file = next(emoji)
        emoji_uuid = ''
        p = Path(emoji_file)
        if p.suffix == ".jpg" or p.suffix == ".jpeg":
            emoji_uuid = p.stem
        # print(emoji_uuid)
        image = compress_image(emoji_file)

        # if not check_emoji(emoji_uuid):
        #     logger.error(f"图片 {emoji_uuid} 不在数据库中！")
        # else:
        description = gen.process_single_image(image)
        if update_description(emoji_uuid, description)
            logger.info(f"图片 {emoji_file} 的描述词生成完毕， 其 description 为: {description}")


if __name__ == "__main__":
    main()