添加获取token数和计费

main.py

@@ -4,6 +4,7 @@ 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 controllers.calculate_cost import calc_cost
 from models.logger import setup_logger
 
 logger = setup_logger()
@@ -19,6 +20,7 @@ emoji = yield_emoji_path(emoji_folder)
 gen = Generator(api_key)
 
 def main():
+    count_cost = 0
     for i in range(get_emoji_count(emoji_folder)):
         emoji_file = next(emoji)
         emoji_uuid = ''
@@ -30,13 +32,16 @@ def main():
 
         if check_emoji(emoji_uuid):
             try:
-                description = gen.process_local_image(str(image_path))
+                description, tokens = gen.get_data(str(image_path))
+                cost = calc_cost(tokens)
+                input_tokens, output_tokens = tokens
             except AttributeError:
                 logger.warning("AI模型未响应，请检查是否欠费！")
                 break
             update_description(emoji_uuid, description)
             if description:
-                logger.info(f"图片 {emoji_file} 的描述词生成完毕，序号：{i + 1}， 其 description 为: {description}")
+                logger.info(f"图片 {emoji_file} 的描述词生成完毕，序号：{i + 1}, 其 description 为: {description}")
+                logger.info(f"本次生成输入token：{input_tokens}, 输出token：{output_tokens}, 花费：{cost} 元")
             else:
                 logger.warning(f"图片 {emoji_file} 的描述词生成失败！序号：{i + 1}")