From f9e2eb6d76a33a980e49a67cc245a35d05639262 Mon Sep 17 00:00:00 2001
From: ljh938527 <2689819155@qq.com>
Date: Sun, 14 Dec 2025 00:05:04 +0800
Subject: [PATCH] =?UTF-8?q?=E9=94=99=E8=AF=AF=E5=A4=84=E7=90=86=E5=92=8C?=
 =?UTF-8?q?=E6=97=A5=E5=BF=97=E5=A4=84=E7=90=86?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 main.py | 33 +++++++++++++++++++++------------
 1 file changed, 21 insertions(+), 12 deletions(-)

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