调整测试代码

test.py

@@ -4,6 +4,7 @@ from models.generator import Generator
 from controllers.localfile_handles import get_emoji_count, yield_emoji_path
 from controllers.compress_image import compress_image
 from models.logger import setup_logger
+from models.emoji_db import get_emoji_url
 
 logger = setup_logger()
 
@@ -17,19 +18,26 @@ emoji = yield_emoji_path(emoji_folder)
 
 gen = Generator(api_key)
 
-fp = open("descriptions.txt", "w", encoding="utf-8")
+def generate_description_to_txt():
+    fp = open("descriptions.txt", "w", encoding="utf-8")
+    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
 
-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
+        description = gen.process_local_image(str(image_path))
+        logger.info(f"图片 {emoji_file} 的描述词生成完毕，序号：{i}， 其 description 为: {description}")
+        fp.write(f"{emoji_uuid}: {description}\n")
+    fp.close()
 
-    description = gen.process_single_image(str(image_path))
-    logger.info(f"图片 {emoji_file} 的描述词生成完毕，序号：{i}， 其 description 为: {description}")
-    fp.write(f"{emoji_uuid}: {description}\n")
+generate_description_to_txt()
 
-fp.close()
+content = get_emoji_url("c3151774-2c4c-48f6-a3ac-b8802cf95498")
+print(content)
+
+url = 'https://pic1.arkoo.com/56D0B40F99F841DF8A2425762AE2565D/picture/o_1i4qop009177v1tgf14db15he1iaj1is.jpg'
+print(gen.process_link_image(url))