diff --git a/dsLightRag/Liblib/T2.py b/dsLightRag/Liblib/T2.py index 288deb85..eabe37f8 100644 --- a/dsLightRag/Liblib/T2.py +++ b/dsLightRag/Liblib/T2.py @@ -1,4 +1,5 @@ from Liblib.LibLibGenerator import LibLibGenerator +import Config.Config if __name__ == '__main__': generate_params = { @@ -22,10 +23,43 @@ if __name__ == '__main__': } liblib = LibLibGenerator() + # 调用生成接口 response = liblib.post_request( "/api/generate/webui/text2img", generate_params ) - # {'generateUuid': 'f16463d55e144214ae12d5177f8256fc'} - print(response) + + if response and "generateUuid" in response: + generate_uuid = response["generateUuid"] + print(f"✅ 图像生成任务已成功提交!生成UUID: {generate_uuid}") + print("开始轮询生成状态...") + + # 等待生成完成(每2秒检查一次) + status_data = liblib.wait_for_generation_completion(generate_uuid, interval=2) + + # 检查生成状态 + if status_data and status_data.get("generateStatus") == 5: + print("🎉 图像生成完成!开始处理文件...") + + # 提取图片URL + if status_data.get("images") and len(status_data["images"]) > 0: + image_url = status_data["images"][0]["imageUrl"] + + # 下载并上传到OBS + obs_object_key = liblib.download_and_upload_to_obs(image_url, generate_uuid) + + if obs_object_key: + # 构建完整的OBS URL + file_url = f"https://{Config.Config.OBS_BUCKET}.{Config.Config.OBS_SERVER}/{obs_object_key}" + print(f"✅ 文件处理完成,OBS地址: {file_url}") + else: + print("❌ 文件上传OBS失败") + else: + print("❌ 未找到生成的图片数据") + else: + error_msg = status_data.get('message', '未知错误') if status_data else '生成状态查询失败' + print(f"❌ 图像生成失败: {error_msg}") + else: + error_msg = response.get('message', '未知错误') if response else 'API无响应' + print(f"❌ 图像生成失败: {error_msg}")