diff --git a/dsLightRag/JiMeng/T1_JmTxt2Img.py b/dsLightRag/JiMeng/Kit/JmTxt2ImgUtil.py similarity index 77% rename from dsLightRag/JiMeng/T1_JmTxt2Img.py rename to dsLightRag/JiMeng/Kit/JmTxt2ImgUtil.py index 8ecb48fa..8603e0e0 100644 --- a/dsLightRag/JiMeng/T1_JmTxt2Img.py +++ b/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}")