diff --git a/main.py b/main.py index feb6184..f14967e 100644 --- a/main.py +++ b/main.py @@ -18,17 +18,26 @@ emoji = yield_emoji_path(emoji_folder) gen = Generator(api_key) -for i in range(get_emoji_count(emoji_folder)): - emoji_file = next(emoji) - emoji_uuid = '' - p = Path(emoji_file) - if p.suffix == ".jpg" or p.suffix == ".jpeg": - emoji_uuid = p.stem - image = compress_image(emoji_file) - image_path = Path(root_folder) / image +def main(): + for i in range(get_emoji_count(emoji_folder)): + emoji_file = next(emoji) + emoji_uuid = '' + p = Path(emoji_file) + if p.suffix == ".jpg" or p.suffix == ".jpeg": + emoji_uuid = p.stem + image = compress_image(emoji_file) + image_path = Path(root_folder) / image - if check_emoji(emoji_uuid): - description = gen.process_single_image(str(image_path)) - update_description(emoji_uuid, description) - logger.info(f"图片 {emoji_file} 的描述词生成完毕,序号:{i}, 其 description 为: {description}") + if check_emoji(emoji_uuid): + description = gen.process_single_image(str(image_path)) + update_description(emoji_uuid, description) + if description: + logger.info(f"图片 {emoji_file} 的描述词生成完毕,序号:{i + 1}, 其 description 为: {description}") + else: + logger.warning(f"图片 {emoji_file} 的描述词生成失败!序号:{i + 1}") +if __name__ == '__main__': + try: + main() + except KeyboardInterrupt: + exit() \ No newline at end of file