错误处理和日志处理

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()