'commit'

dsLightRag/JiMeng/Kit/JmTxt2ImgUtil.py

@@ -1,7 +1,6 @@
 import base64
 import json
 import logging
-import os
 import uuid
 
 from Config.Config import OBS_SERVER, OBS_PREFIX, OBS_BUCKET
@@ -81,13 +80,9 @@ class JmTxt2Img:
             raise Exception(f"保存图片失败: {str(e)}")
         return None
 
-    @staticmethod
-    def main():
-        prompt = "雨天街头，一位头戴黄色安全帽、身穿荧光黄安全背心与棕色外套、搭配蓝色牛仔裤和棕色工装靴的中国工人，背着一个中国小女孩。工人脖子上侧挂一个小熊玩偶包，拖住小女孩臀部的手同时挂着一袋小熊饼干。小女孩双手搂着工人的脖子。小女孩穿红色雨衣，戴红色雨衣帽。两个人露出微笑。背景是居民楼和树木，地面潮湿，氛围温馨。"
-        # 保存图片路径
-        url = JmTxt2Img.generate_image(prompt)
-        logger.info(f"图片URL: {url}")
-
 
 if __name__ == "__main__":
-    JmTxt2Img.main()
+    prompt = "雨天街头，一位头戴黄色安全帽、身穿荧光黄安全背心与棕色外套、搭配蓝色牛仔裤和棕色工装靴的中国工人，背着一个中国小女孩。工人脖子上侧挂一个小熊玩偶包，拖住小女孩臀部的手同时挂着一袋小熊饼干。小女孩双手搂着工人的脖子。小女孩穿红色雨衣，戴红色雨衣帽。两个人露出微笑。背景是居民楼和树木，地面潮湿，氛围温馨。"
+    # 保存图片路径
+    url = JmTxt2Img.generate_image(prompt)
+    logger.info(f"图片URL: {url}")