调整测试代码

main.py

@@ -30,7 +30,7 @@ def main():
 
         if check_emoji(emoji_uuid):
             try:
-                description = gen.process_single_image(str(image_path))
+                description = gen.process_local_image(str(image_path))
             except AttributeError:
                 logger.warning("AI模型未响应，请检查是否欠费！")
                 break

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,9 +18,9 @@ emoji = yield_emoji_path(emoji_folder)
 
 gen = Generator(api_key)
 
-fp = open("descriptions.txt", "w", encoding="utf-8")
-
-for i in range(get_emoji_count(emoji_folder)):
+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)
@@ -28,8 +29,15 @@ for i in range(get_emoji_count(emoji_folder)):
         image = compress_image(emoji_file)
         image_path = Path(root_folder) / image
 
-    description = gen.process_single_image(str(image_path))
+        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()
 
-fp.close()
+generate_description_to_txt()
+
+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))