API 文档

{
  "err_code": 0,
  "task_id": "字符串",
  "remaining_count": 5,
  "remaining_money": 300
}

def test_submit_vector_task():
    # 准备测试数据
    color_count = 5       # 矢量图颜色数量
    input_image_path = "input/logo.png"  # 替换为有效的图像路径

    with open(input_image_path, "rb") as image_file:
        try:
            response = httpx.post(
                f"{BASE_URL}/api/submit_vector_task",
                data={
                    "user_key": USER_KEY,
                    "color_count": color_count,
                },
                files={"input_image": image_file},
                timeout=httpx.Timeout(
                    connect=5.0,    # 连接超时
                    read=10.0,      # 读取超时
                    write=30.0,     # 写入超时
                    pool=None       # 连接池超时
                )
            )
            response.raise_for_status()
        except httpx.TimeoutException:
            print("请求超时，请检查网络连接或服务器状态")
            return -1
        except httpx.HTTPStatusError as e:
            print(f"HTTP请求失败: {e.response.status_code}")
            print(f"错误详情: {e.response.json()}")
            return -1
        except httpx.RequestError as e:
            print(f"请求错误: {str(e)}")
            return -1
        except Exception as e:
            print(f"发生未知错误: {str(e)}")
            return -1

    print(response.status_code)
    result = response.json()
    print(result)
    err_code = result.get("err_code")
    if err_code != 0:
        print(f"提交任务失败，错误码: {result.get('err_msg')}")
        return -1

    task_id = result.get("task_id")
    print(f"提交任务成功，任务ID: {task_id}")
    return task_id

if __name__ == "__main__":
    # 提交矢量图转换任务
    task_id = test_submit_vector_task()
    if task_id == -1:
        exit()
    # 使用之前定义的函数检查任务状态
    test_check_task_status(task_id)

{
  "err_code": 0,
  "task_id": "字符串",
  "remaining_count": 5,
  "remaining_money": 300
}

def test_submit_face_enhance_task():
    # 准备测试数据
    input_image_path = "input/face.png"  # 替换为有效的人脸图像路径

    with open(input_image_path, "rb") as image_file:
        try:
            response = httpx.post(
                f"{BASE_URL}/api/submit_face_enhance_task",
                data={
                    "user_key": USER_KEY,
                },
                files={"input_image": image_file},
                timeout=httpx.Timeout(
                    connect=5.0,    # 连接超时
                    read=10.0,      # 读取超时
                    write=30.0,     # 写入超时
                    pool=None       # 连接池超时
                )
            )
            response.raise_for_status()
        except httpx.TimeoutException:
            print("请求超时，请检查网络连接或服务器状态")
            return -1
        except httpx.HTTPStatusError as e:
            print(f"HTTP请求失败: {e.response.status_code}")
            print(f"错误详情: {e.response.json()}")
            return -1
        except httpx.RequestError as e:
            print(f"请求错误: {str(e)}")
            return -1
        except Exception as e:
            print(f"发生未知错误: {str(e)}")
            return -1

    print(response.status_code)
    result = response.json()
    print(result)
    err_code = result.get("err_code")
    if err_code != 0:
        print(f"提交任务失败，错误码: {result.get('err_msg')}")
        return -1

    task_id = result.get("task_id")
    print(f"提交任务成功，任务ID: {task_id}")
    return task_id

if __name__ == "__main__":
    # 提交人脸修复任务
    task_id = test_submit_face_enhance_task()
    if task_id == -1:
        exit()
    # 使用之前定义的函数检查任务状态
    test_check_task_status(task_id)

{
  "err_code": 0,
  "task_id": "字符串",
  "remaining_count": 5,
  "remaining_money": 300
}

def test_submit_seamless_task():
    # 准备测试数据
    fit = 0.3
    direction = 0        # 四周无缝
    expand_top = 0.2     # 向上扩图20%
    expand_bottom = 0.1  # 向下扩图10%
    expand_left = 0.0    # 不向左扩图
    expand_right = 0.0   # 不向右扩图
    input_image_path = "input/seamless.jpeg"  # 替换为有效的图像路径

    with open(input_image_path, "rb") as image_file:
        try:
            response = httpx.post(
                f"{BASE_URL}/api/submit_seamless_task",
                data={
                    "user_key": USER_KEY,
                    "fit": fit,
                    "direction": direction,
                    "expand_top": expand_top,
                    "expand_bottom": expand_bottom,
                    "expand_left": expand_left,
                    "expand_right": expand_right,
                },
                files={"input_image": image_file},
                timeout=httpx.Timeout(
                    connect=5.0,    # 连接超时
                    read=10.0,      # 读取超时
                    write=30.0,     # 写入超时
                    pool=None       # 连接池超时
                )
            )
            response.raise_for_status()
        except httpx.TimeoutException:
            print("请求超时，请检查网络连接或服务器状态")
            return -1
        except httpx.HTTPStatusError as e:
            print(f"HTTP请求失败: {e.response.status_code}")
            print(f"错误详情: {e.response.json()}")
            return -1
        except httpx.RequestError as e:
            print(f"请求错误: {str(e)}")
            return -1
        except Exception as e:
            print(f"发生未知错误: {str(e)}")
            return -1

    print(response.status_code)
    result = response.json()
    print(result)
    err_code = result.get("err_code")
    if err_code != 0:
        print(f"提交任务失败，错误码: {result.get('err_msg')}")
        return -1

    task_id = result.get("task_id")
    print(f"提交任务成功，任务ID: {task_id}")
    return task_id

if __name__ == "__main__":
    # 提交无缝拼图任务
    task_id = test_submit_seamless_task()
    if task_id == -1:
        exit()
    # 使用之前定义的函数检查任务状态
    test_check_task_status(task_id)

{
  "err_code": 0,
  "task_id": "字符串",
  "remaining_count": 5,
  "remaining_money": 300
}

def test_submit_remove_fabric_task():
    # 准备测试数据
    input_image_path = "input/fabric_pattern.png"  # 替换为有效的图像路径

    with open(input_image_path, "rb") as image_file:
        try:
            response = httpx.post(
                f"{BASE_URL}/api/submit_remove_fabric_task",
                data={
                    "user_key": USER_KEY,
                },
                files={"input_image": image_file},
                timeout=httpx.Timeout(
                    connect=5.0,    # 连接超时
                    pool=None       # 连接池超时
                )
            )
            response.raise_for_status()
        except httpx.TimeoutException:
            print("请求超时，请检查网络连接或服务器状态")
            return -1
        except httpx.HTTPStatusError as e:
            print(f"HTTP请求失败: {e.response.status_code}")
            print(f"错误详情: {e.response.json()}")
            return -1
        except httpx.RequestError as e:
            print(f"请求错误: {str(e)}")
            return -1
        except Exception as e:
            print(f"发生未知错误: {str(e)}")
            return -1

    print(response.status_code)
    result = response.json()
    print(result)
    err_code = result.get("err_code")
    if err_code != 0:
        print(f"提交任务失败，错误码: {result.get('err_msg')}")
        return -1

    task_id = result.get("task_id")
    print(f"提交任务成功，任务ID: {task_id}")
    return task_id

if __name__ == "__main__":
    # 提交移除布纹任务
    task_id = test_submit_remove_fabric_task()
    if task_id == -1:
        exit()
    # 使用之前定义的函数检查任务状态
    test_check_task_status(task_id)

{
  "err_code": 0,
  "task_id": "字符串",
  "remaining_count": 5,
  "remaining_money": 300
}

def test_submit_edit_image_task():
    # 准备测试数据
    prompt = "将这张图片转换为水彩画风格"
    model_version = "v2"  # 模型版本，默认v1
    input_image_path = "input/flower2.jpg"  # 替换为有效的图像路径

    with open(input_image_path, "rb") as image_file:
        try:
            response = httpx.post(
                f"{BASE_URL}/api/submit_edit_image_task",
                data={
                    "user_key": USER_KEY,
                    "prompt": prompt,
                    "model_version": model_version,
                },
                files={"input_image": image_file},
                timeout=httpx.Timeout(
                    connect=5.0,    # 连接超时
                    read=10.0,      # 读取超时
                    write=30.0,     # 写入超时
                    pool=None       # 连接池超时
                )
            )
            response.raise_for_status()
        except httpx.TimeoutException:
            print("请求超时，请检查网络连接或服务器状态")
            return -1
        except httpx.HTTPStatusError as e:
            print(f"HTTP请求失败: {e.response.status_code}")
            print(f"错误详情: {e.response.json()}")
            return -1
        except httpx.RequestError as e:
            print(f"请求错误: {str(e)}")
            return -1
        except Exception as e:
            print(f"发生未知错误: {str(e)}")
            return -1

    print(response.status_code)
    result = response.json()
    print(result)
    err_code = result.get("err_code")
    if err_code != 0:
        print(f"提交任务失败，错误码: {result.get('err_msg')}")
        return -1

    task_id = result.get("task_id")
    print(f"提交AI改图任务成功，任务ID: {task_id}")
    return task_id

if __name__ == "__main__":
    # 提交AI改图任务
    task_id = test_submit_edit_image_task()
    if task_id == -1:
        exit()
    # 使用之前定义的函数检查任务状态
    test_check_task_status(task_id)

{
  "err_code": 0,
  "task_id": "字符串",
  "remaining_count": 5,
  "remaining_money": 300
}

def test_submit_extract_pattern_task():
    # 准备测试数据
    aspect_ratio = "1:1"  # 出图比例，可选值：1:1, 4:3, 16:9 等
    input_image_path = "input/flower.jpeg"  # 替换为有效的服装图像路径

    with open(input_image_path, "rb") as image_file:
        try:
            response = httpx.post(
                f"{BASE_URL}/api/submit_extract_pattern_task",
                data={
                    "user_key": USER_KEY,
                    "aspect_ratio": aspect_ratio,
                },
                files={"input_image": image_file},
                timeout=httpx.Timeout(
                    connect=5.0,    # 连接超时
                    read=10.0,      # 读取超时
                    write=30.0,     # 写入超时
                    pool=None       # 连接池超时
                )
            )
            response.raise_for_status()
        except httpx.TimeoutException:
            print("请求超时，请检查网络连接或服务器状态")
            return -1
        except httpx.HTTPStatusError as e:
            print(f"HTTP请求失败: {e.response.status_code}")
            print(f"错误详情: {e.response.json()}")
            return -1
        except httpx.RequestError as e:
            print(f"请求错误: {str(e)}")
            return -1
        except Exception as e:
            print(f"发生未知错误: {str(e)}")
            return -1

    print(response.status_code)
    result = response.json()
    print(result)
    err_code = result.get("err_code")
    if err_code != 0:
        print(f"提交任务失败，错误码: {result.get('err_msg')}")
        return -1

    task_id = result.get("task_id")
    print(f"提交提取印花任务成功，任务ID: {task_id}")
    return task_id

if __name__ == "__main__":
    # 提交提取印花任务
    task_id = test_submit_extract_pattern_task()
    if task_id == -1:
        exit()
    # 使用之前定义的函数检查任务状态
    test_check_task_status(task_id)

{
  "err_code": 0,
  "task_id": "字符串",
  "remaining_count": 5,
  "remaining_money": 300
}

def test_submit_remove_diamonds_task():
    # 准备测试数据
    input_image_path = "input/diamonds.jpg"  # 替换为有效的图像路径

    with open(input_image_path, "rb") as image_file:
        try:
            response = httpx.post(
                f"{BASE_URL}/api/submit_remove_diamonds_task",
                data={
                    "user_key": USER_KEY,
                },
                files={"input_image": image_file},
                timeout=httpx.Timeout(
                    connect=5.0,    # 连接超时
                    read=10.0,      # 读取超时
                    write=30.0,     # 写入超时
                    pool=None       # 连接池超时
                )
            )
            response.raise_for_status()
        except httpx.TimeoutException:
            print("请求超时，请检查网络连接或服务器状态")
            return -1
        except httpx.HTTPStatusError as e:
            print(f"HTTP请求失败: {e.response.status_code}")
            print(f"错误详情: {e.response.json()}")
            return -1
        except httpx.RequestError as e:
            print(f"请求错误: {str(e)}")
            return -1
        except Exception as e:
            print(f"发生未知错误: {str(e)}")
            return -1

    print(response.status_code)
    result = response.json()
    print(result)
    err_code = result.get("err_code")
    if err_code != 0:
        print(f"提交任务失败，错误码: {result.get('err_msg')}")
        return -1

    task_id = result.get("task_id")
    print(f"提交移除烫钻任务成功，任务ID: {task_id}")
    return task_id

if __name__ == "__main__":
    # 提交移除烫钻任务
    task_id = test_submit_remove_diamonds_task()
    if task_id == -1:
        exit()
    # 使用之前定义的函数检查任务状态
    test_check_task_status(task_id)

{
  "err_code": 0,
  "task_id": "字符串",
  "remaining_count": 5,
  "remaining_money": 300
}

def test_submit_expand_image_task():
    # 准备测试数据
    expand_top = 0.2     # 向上扩图20%
    expand_bottom = 0.1  # 向下扩图10%
    expand_left = 0.0    # 不向左扩图
    expand_right = 0.0   # 不向右扩图
    prompt = "延续原图的风格和内容"  # 扩展部分的生成提示
    input_image_path = "input/test.png"  # 替换为有效的图像路径

    with open(input_image_path, "rb") as image_file:
        try:
            response = httpx.post(
                f"{BASE_URL}/api/submit_expand_image_task",
                data={
                    "user_key": USER_KEY,
                    "expand_top": expand_top,
                    "expand_bottom": expand_bottom,
                    "expand_left": expand_left,
                    "expand_right": expand_right,
                    "prompt": prompt
                },
                files={"input_image": image_file},
                timeout=300.0
            )
        except Exception as e:
            print(f"请求失败: {str(e)}")
            return -1

    print(response.status_code)
    result = response.json()
    print(result)
    err_code = result.get("err_code")
    if err_code != 0:
        print(f"提交任务失败，错误码: {result.get('err_msg')}")
        return -1

    task_id = result.get("task_id")
    print(f"提交扩图任务成功，任务ID: {task_id}")
    return task_id

if __name__ == "__main__":
    # 提交扩图任务
    task_id = test_submit_expand_image_task()
    if task_id == -1:
        exit()
    # 使用之前定义的函数检查任务状态
    test_check_task_status(task_id)

{
  "err_code": 0,
  "task_id": "字符串",
  "remaining_count": 5,
  "remaining_money": 300
}

def test_submit_remove_watermark_task():
    # 准备测试数据
    input_image_path = "input/watermarked.png"  # 替换为有效的图像路径

    with open(input_image_path, "rb") as image_file:
        try:
            response = httpx.post(
                f"{BASE_URL}/api/submit_remove_watermark_task",
                data={
                    "user_key": USER_KEY
                },
                files={"input_image": image_file},
                timeout=300.0
            )
        except Exception as e:
            print(f"请求失败: {str(e)}")
            return -1

    print(response.status_code)
    result = response.json()
    print(result)
    err_code = result.get("err_code")
    if err_code != 0:
        print(f"提交任务失败，错误码: {result.get('err_msg')}")
        return -1

    task_id = result.get("task_id")
    print(f"提交去水印任务成功，任务ID: {task_id}")
    return task_id

if __name__ == "__main__":
    # 提交去水印任务
    task_id = test_submit_remove_watermark_task()
    if task_id == -1:
        exit()
    # 使用之前定义的函数检查任务状态
    test_check_task_status(task_id)

{
  "err_code": 0,
  "status": "Completed",
  "images_url": "https://api.domain.com/api/images/xxx/task123",
  "vector_url": "https://api.domain.com/api/images/xxx/task123/vector",
  "grid_url": "https://api.domain.com/api/images/xxx/task123/grid",
  "generated_prompt": "high quality, detailed, sharp, professional photography, beautiful lighting",
  "remaining_count": 5,
  "remaining_money": 300
}

{
  "err_code": 0,
  "status": "Completed",
  "images_url": "https://api.domain.com/api/images/xxx/task123",
  "images_urls": [
    "https://api.domain.com/api/images/xxx/task123",
    "https://api.domain.com/api/images/xxx/task123/alternative/1"
  ],
  "vector_url": "",
  "grid_url": "",
  "generated_prompt": "vibrant pattern, textile design, seamless repeat, colorful motifs",
  "remaining_count": 5,
  "remaining_money": 300
}