|
@@ -517,7 +517,13 @@ async def image_generations(
|
|
|
images = []
|
|
|
|
|
|
for image in res["data"]:
|
|
|
- image_data, content_type = load_b64_image_data(image["b64_json"])
|
|
|
+ if "url" in image:
|
|
|
+ image_data, content_type = load_url_image_data(
|
|
|
+ image["url"], headers
|
|
|
+ )
|
|
|
+ else:
|
|
|
+ image_data, content_type = load_b64_image_data(image["b64_json"])
|
|
|
+
|
|
|
url = upload_image(request, data, image_data, content_type, user)
|
|
|
images.append({"url": url})
|
|
|
return images
|