from Config.Config import OBS_BUCKET, OBS_SERVER
from LibLibUtil import LibLibUtil

if __name__ == '__main__':
    liblib = LibLibUtil()
    print("===== 测试生成任务处理流程 =====")
    test_generate_uuid = "df9bd6e03a204542b1daaba2f17c42e6"

    if test_generate_uuid:
        # 调用封装好的完整处理方法
        obs_url = liblib.process_generation_task(test_generate_uuid, interval=2)
        if obs_url:
            file_url = f"https://{OBS_BUCKET}.{OBS_SERVER}/{obs_url}"
            print(f"\n✅ 处理完成，OBS地址: {file_url}")
        else:
            print("\n❌ 任务处理失败")
    else:
        print("请先设置有效的生图任务UUID")