from .LiblibUtil import LiblibUtil


if __name__ == '__main__':
    liblib = LiblibUtil()
    print("===== 测试查询生图结果 =====")
    test_generate_uuid = "061d402be5af492d9949d53719912737"
    
    if test_generate_uuid:
        # 直接查询状态
        status_data = liblib.get_generation_status(test_generate_uuid)
        if status_data:
            print(f"生图任务状态：{status_data}")
        else:
            print("查询生图任务状态失败")
        
        # 等待任务完成（如需测试请取消注释）
        # liblib.wait_for_generation_completion(test_generate_uuid)
    else:
        print("请先设置有效的生图任务UUID")