|
38 | 38 |
|
39 | 39 |
|
40 | 40 | async def handle_recraft_file_request(
|
41 |
| - image: torch.Tensor, |
42 |
| - path: str, |
43 |
| - mask: torch.Tensor=None, |
44 |
| - total_pixels=4096*4096, |
45 |
| - timeout=1024, |
46 |
| - request=None, |
47 |
| - auth_kwargs: dict[str,str] = None, |
48 |
| - ) -> list[BytesIO]: |
49 |
| - """ |
50 |
| - Handle sending common Recraft file-only request to get back file bytes. |
51 |
| - """ |
52 |
| - if request is None: |
53 |
| - request = EmptyRequest() |
54 |
| - |
55 |
| - files = { |
56 |
| - 'image': tensor_to_bytesio(image, total_pixels=total_pixels).read() |
57 |
| - } |
58 |
| - if mask is not None: |
59 |
| - files['mask'] = tensor_to_bytesio(mask, total_pixels=total_pixels).read() |
60 |
| - |
61 |
| - operation = SynchronousOperation( |
62 |
| - endpoint=ApiEndpoint( |
63 |
| - path=path, |
64 |
| - method=HttpMethod.POST, |
65 |
| - request_model=type(request), |
66 |
| - response_model=RecraftImageGenerationResponse, |
67 |
| - ), |
68 |
| - request=request, |
69 |
| - files=files, |
70 |
| - content_type="multipart/form-data", |
71 |
| - auth_kwargs=auth_kwargs, |
72 |
| - multipart_parser=recraft_multipart_parser, |
73 |
| - ) |
74 |
| - response: RecraftImageGenerationResponse = await operation.execute() |
75 |
| - all_bytesio = [] |
76 |
| - if response.image is not None: |
77 |
| - all_bytesio.append(await download_url_to_bytesio(response.image.url, timeout=timeout)) |
78 |
| - else: |
79 |
| - for data in response.data: |
80 |
| - all_bytesio.append(await download_url_to_bytesio(data.url, timeout=timeout)) |
| 41 | + image: torch.Tensor, |
| 42 | + path: str, |
| 43 | + mask: torch.Tensor=None, |
| 44 | + total_pixels=4096*4096, |
| 45 | + timeout=1024, |
| 46 | + request=None, |
| 47 | + auth_kwargs: dict[str,str] = None, |
| 48 | +) -> list[BytesIO]: |
| 49 | + """ |
| 50 | + Handle sending common Recraft file-only request to get back file bytes. |
| 51 | + """ |
| 52 | + if request is None: |
| 53 | + request = EmptyRequest() |
| 54 | + |
| 55 | + files = { |
| 56 | + 'image': tensor_to_bytesio(image, total_pixels=total_pixels).read() |
| 57 | + } |
| 58 | + if mask is not None: |
| 59 | + files['mask'] = tensor_to_bytesio(mask, total_pixels=total_pixels).read() |
| 60 | + |
| 61 | + operation = SynchronousOperation( |
| 62 | + endpoint=ApiEndpoint( |
| 63 | + path=path, |
| 64 | + method=HttpMethod.POST, |
| 65 | + request_model=type(request), |
| 66 | + response_model=RecraftImageGenerationResponse, |
| 67 | + ), |
| 68 | + request=request, |
| 69 | + files=files, |
| 70 | + content_type="multipart/form-data", |
| 71 | + auth_kwargs=auth_kwargs, |
| 72 | + multipart_parser=recraft_multipart_parser, |
| 73 | + ) |
| 74 | + response: RecraftImageGenerationResponse = await operation.execute() |
| 75 | + all_bytesio = [] |
| 76 | + if response.image is not None: |
| 77 | + all_bytesio.append(await download_url_to_bytesio(response.image.url, timeout=timeout)) |
| 78 | + else: |
| 79 | + for data in response.data: |
| 80 | + all_bytesio.append(await download_url_to_bytesio(data.url, timeout=timeout)) |
81 | 81 |
|
82 |
| - return all_bytesio |
| 82 | + return all_bytesio |
83 | 83 |
|
84 | 84 |
|
85 | 85 | def recraft_multipart_parser(data, parent_key=None, formatter: callable=None, converted_to_check: list[list]=None, is_list=False) -> dict:
|
|
0 commit comments