From 9954db0823c2c7a937021bf9dfd7518ae520251b Mon Sep 17 00:00:00 2001 From: HuangHai <10402852@qq.com> Date: Wed, 3 Sep 2025 15:32:20 +0800 Subject: [PATCH] 'commit' --- dsLightRag/Liblib/LiblibUtil.py | 97 +++++++++++++----- dsLightRag/Liblib/T2.py | 39 ++++--- dsLightRag/Liblib/T3.py | 61 +++++++++++ .../__pycache__/LiblibUtil.cpython-310.pyc | Bin 0 -> 4823 bytes 4 files changed, 157 insertions(+), 40 deletions(-) create mode 100644 dsLightRag/Liblib/T3.py create mode 100644 dsLightRag/Liblib/__pycache__/LiblibUtil.cpython-310.pyc diff --git a/dsLightRag/Liblib/LiblibUtil.py b/dsLightRag/Liblib/LiblibUtil.py index e4533d68..764efcc7 100644 --- a/dsLightRag/Liblib/LiblibUtil.py +++ b/dsLightRag/Liblib/LiblibUtil.py @@ -79,33 +79,76 @@ class LiblibUtil: payload = {"generateUuid": generate_uuid} return self.post_request(uri, payload) - def wait_for_generation_completion(self, generate_uuid, check_interval=5, max_wait_time=300): - """等待生图任务完成并获取最终结果""" - start_time = time.time() - - while time.time() - start_time < max_wait_time: + def wait_for_generation_completion(self, generate_uuid, interval=5): + """等待生成完成并返回最终状态""" + while True: status_data = self.get_generation_status(generate_uuid) - - if status_data: - status = status_data.get('status') - progress = status_data.get('percentCompleted', 0) * 100 - print(f"生成进度：{progress:.2f}%") - - if status == 'completed': - print("生图任务已完成") - images = status_data.get('images', []) - if images: - print("生成的最终图片：") - for img in images: - print(f"- {img.get('imageUrl')}") - return status_data - elif status == 'failed': - print(f"生图任务失败: {status_data.get('errorMessage', '未知错误')}") - return None - else: - print(f"生图任务处理中，当前状态: {status}") - - time.sleep(check_interval) + if not status_data: + print("获取状态失败，重试中...") + time.sleep(interval) + continue + + # 获取当前状态和进度 + generate_status = status_data.get('generateStatus') + percent = status_data.get('percentCompleted', 0) * 100 + status_text = { + 0: "初始化", + 1: "处理中", + 2: "排队中", + 3: "已取消", + 4: "失败", + 5: "完成" + }.get(generate_status, f"未知状态({generate_status})") + + # 显示进度和状态 + print(f"生成进度：{percent:.2f}% | 状态：{status_text}") + + # 检查是否完成 + if generate_status in [3, 4, 5]: # 取消、失败或完成 + return status_data + + time.sleep(interval) print(f"生图任务超时（{max_wait_time}秒）") - return None \ No newline at end of file + return None + + @staticmethod + def download_and_upload_to_obs(image_url, generate_uuid): + """下载图片并上传到OBS""" + try: + # 1. 清理URL（去除可能的引号和空格） + clean_url = image_url.strip('` ') + print(f"清理后的图片URL: {clean_url}") + + # 2. 下载图片到临时文件 + response = requests.get(clean_url, stream=True) + response.raise_for_status() # 检查HTTP错误 + + # 创建临时文件 + with tempfile.NamedTemporaryFile(delete=False, suffix='.png') as temp_file: + for chunk in response.iter_content(chunk_size=8192): + temp_file.write(chunk) + temp_file_path = temp_file.name + print(f"图片已下载至临时文件: {temp_file_path}") + + # 3. 上传到OBS + obs_uploader = ObsUploader() + obs_object_key = f"HuangHai/Liblib/{generate_uuid}.jpg" + success, result = obs_uploader.upload_file( + object_key=obs_object_key, + file_path=temp_file_path + ) + + # 4. 清理临时文件 + os.unlink(temp_file_path) + + if success: + print(f"✅ 图片成功上传至OBS") + return obs_object_key + else: + print(f"❌ OBS上传失败: {result}") + return None + + except Exception as e: + print(f"处理图片时发生错误: {str(e)}") + return None diff --git a/dsLightRag/Liblib/T2.py b/dsLightRag/Liblib/T2.py index fe386cab..655307d9 100644 --- a/dsLightRag/Liblib/T2.py +++ b/dsLightRag/Liblib/T2.py @@ -1,20 +1,33 @@ -from .LiblibUtil import LiblibUtil - +import json +import requests +import os +import tempfile +from LiblibUtil import LiblibUtil +from Util.ObsUtil import ObsUploader # 导入OBS上传工具 if __name__ == '__main__': liblib = LiblibUtil() - print("===== 测试查询生图结果 =====") - test_generate_uuid = "061d402be5af492d9949d53719912737" - + print("===== 测试查询生图结果并上传OBS =====") + test_generate_uuid = "df9bd6e03a204542b1daaba2f17c42e6" + if test_generate_uuid: - # 直接查询状态 - status_data = liblib.get_generation_status(test_generate_uuid) - if status_data: - print(f"生图任务状态：{status_data}") + # 等待任务完成 + status_data = liblib.wait_for_generation_completion(test_generate_uuid) + print(f"生图状态: {json.dumps(status_data, ensure_ascii=False, indent=2)}") + + # 检查生成状态是否为5（完成） + if status_data and status_data.get('generateStatus') == 5: + # 提取图片URL + if status_data.get('images') and len(status_data['images']) > 0: + image_url = status_data['images'][0]['imageUrl'] + # 下载并上传到OBS + # 使用实例调用而非类调用 + obs_path = liblib.download_and_upload_to_obs(image_url, test_generate_uuid) + if obs_path: + print(f"\n最终OBS路径: {obs_path}") + else: + print("❌ 未找到图片数据") else: - print("查询生图任务状态失败") - - # 等待任务完成（如需测试请取消注释） - # liblib.wait_for_generation_completion(test_generate_uuid) + print(f"❌ 生图未完成或失败，状态码: {status_data.get('generateStatus')}") else: print("请先设置有效的生图任务UUID") \ No newline at end of file diff --git a/dsLightRag/Liblib/T3.py b/dsLightRag/Liblib/T3.py new file mode 100644 index 00000000..e36580ad --- /dev/null +++ b/dsLightRag/Liblib/T3.py @@ -0,0 +1,61 @@ +import json +from LiblibUtil import LiblibUtil + +def generate_image_with_controlnet(): + # 初始化Liblib工具类 + liblib_util = LiblibUtil() + + # 构建生成参数 + generate_params = { + "templateUuid": "6f7c4652458d4802969f8d089cf5b91f", + "generateParams": { + "prompt": "filmfotos, Asian portrait,A young woman wearing a green baseball cap,covering one eye with her hand", + "steps": 20, + "width": 768, + "height": 1024, + "imgCount": 1, + "controlNet": [ + { + "unitOrder": 0, + "sourceImage": "https://dsideal.obs.cn-north-1.myhuaweicloud.com/HuangHai/Backup/HuangWanQiao.jpg", + "width": 768, + "height": 1024, + "preprocessor": 0, + "annotationParameters": {"none": {}}, + "model": "405836d1ae2646b4ba2716ed6bd5453a", + "controlWeight": 1, + "startingControlStep": 0, + "endingControlStep": 1, + "pixelPerfect": 1, + "controlMode": 0, + "resizeMode": 1 + } + ] + } + } + + # 调用生成接口 + try: + response = liblib_util.post_request( + endpoint="/api/generate/webui/text2img", + params=generate_params + ) + + print(f"API响应: {json.dumps(response, ensure_ascii=False, indent=2)}") + + # 修复成功条件判断 - 检查是否返回generateUuid + if response and "generateUuid" in response: + print("✅ 图像生成任务已成功提交！") + print(f"生成UUID: {response['generateUuid']}") + print("提示: 可使用T2.py中的get_generation_status方法查询生成进度") + return response + else: + error_msg = response.get('message', '未知错误') if response else 'API无响应' + print(f"❌ 图像生成失败: {error_msg}") + return None + except Exception as e: + print(f"请求发生异常: {str(e)}") + return None + +if __name__ == "__main__": + generate_image_with_controlnet() \ No newline at end of file diff --git a/dsLightRag/Liblib/__pycache__/LiblibUtil.cpython-310.pyc b/dsLightRag/Liblib/__pycache__/LiblibUtil.cpython-310.pyc new file mode 100644 index 0000000000000000000000000000000000000000..116420cf50ca3aeec6270072ebfbfa443bf9a8b6 GIT binary patch literal 4823 zcma)A{c{{;9p7hnZ*On!CD$|*s8v%HuZgul^b?MK4oG5ov{G?rkoqM%bR7FkKuMHa0|y2STnUFLmE zkMUm772d~n^@oBbS@IFVl4nIdVX;vm7h3^OVI;>w*_$%&%hPP_XI1t-tlEOkEI}77 zW{F1x$m84~FJ>v2$E>)eqF1bhrJ;{oNh^h3wKiHAtM3TY6Sih$tqn(nS*9nge#jpP zwLQ*w(HYLJV2Kr({3s0Y(tDq@&z^FY z;W^jY-G3)D?#N|B$@AS%a>{-vlj=e_2plVnQM)aa3l*mv%2OpXA4+9=+9e%?ie1iE zEZe2m4He6ov^_sm$b1iO3pc=mo;hI~;}33&=t9MX45tcGo1V+XU5cwv3S1{lkWkMz zOVzN?b0*8CAGo$ru9Wk37|&PAzFqd+JBeJTZEnn{Nzr1WWFE8)+S!1qM{5IOie=dr zaR8>O4XiQKDrugAr}rV#(s8(*QqW3mPhDsVb=G1Bq=wXx8*t+So5RW+EQhg(9#%pt z)n&ADJvJwt*BT0io5LD$UnWOz$KDQ`15&8+;kMX`lJo%HfyyK07)Vxl;T1`;lG;;a#=lqdR$Y zK4Z`9s3q?n?OBl~V?FEcsAX^LjF5SE)KbU-WZ!$oW~z2AYgVg8CvW;rr952lD&?F6 z|3e?fr)(4M=D7)+C6u}Ek^6?SW%_177Q{JL&i#7%(xvT#VH^(SnUnTNkiky(#MfSF zFDz^yf$MUyS=V%spC&4<0eAHS z&;16`N|Ux9#;Ps_kY*ps+f_Qh7iRZGJa+e%+$@nNR_D1Sj{=u6Vi>QQGsTK&xnIM~ z{VFwsw2lxdx_8kCXY#6GZ0qUX$e_0mnY}(HQgmpe`wm~i!#qoiVwT;*RD^|!Hp5bE zKvY;iz8R(-&UAb6i`wmLLS`+eo3a~aHc}XmnkO-^kj)Sk4E~P-fcQa#w8Vy7m))(% zUX1b-@8xcf?8WB_Vjw;*!&g;gu0(JLm@S{a(mwM>=bcvjt=E=bJ>HrBSbC}}$Ds7cP3PW+ZC&HMP%PjcnMv?Y~@KTe01_aB=z4Z=fNl zXYuOQ_KDWgiOZeCFSlPmxpejCoww#|n{TDCC{yUE-W!v$+=svqKF4cdy0^2=qtoIG? zoNV>i6%;s`xPxlri-NneE};P4ki>EW!8dWJrT1QKe{!_zpnp0ESkwN?<>gP$wddaI z%pYF5dOWhmSU|RlvPaiGZ-21-(Z$cNom@F~YWdPH78l+d8X5{$-F1{_ay?r`T0kM% zRe?|2zHPZwUREqL0VV(T@$>E9zt%qaMl|E)1FZP0bNcm_UtQ?U+E+fL%ys#AKxS!k z>Cup%I;TXOIeUUnD-t#2iSDClf;5`W{2!JsT#U|t&vljy`mh+i`|_uA?Z5u+^J{O` zq@nvKIQ3nF^}&c1%Q?w?f_Uwwh8RNNBQ(C3ntn84%q!Y%eyr>hi?`T%Imdr$h<{Lh1;FSb-xw;5zZlofHBE?)0bF6T*+;K#OuyC2WpB zxw|eG5)B3DO_PysW+f(+)%yhRp=s>W_j%&SfcGd6qA=5Dg+|;;akQB}F9A8m3(2}# zSFFq_@qpM!G&GL>Q8y7b33n3D_wf)?IeMh<(t6xTGRKXXnUg?&NsbwrwZZSJC)Z&{ zt&w~wNz@h}O*5ds{(#`a;=*goAAcOx1%%z=!imLeO+dhBA0MrWKN$>gH5NJxM+s7$ zJ-zhik%$sky6uBwxP~b}P_|imvL+8z%ab{gt^pKe*a#*joI{Q<2tQ?^mY)jB2Z7~l zKyFwPx)QH^7#(2w*q;{{KI;7Pa_5a>i&rmiAFO4c3e584Q>HT!i9 zY*nVqbZWyaTU}mMzro7`xkoTdlr-SbjHI$GVm*bp-}0}FtVk*30E&9;3r(TIrmzf< zYDSEZDiG>jYa;)@%fbInq7;~-dps%AFhb2BEEd2aj4)*wF9c>W+HooKMi&8&`nJ>b zVQP5(d