mirrors/gpt4free
1
Fork 0
mirror of https://github.com/xtekky/gpt4free.git synced 2025-12-06 02:30:41 -08:00
Code Issues Projects Releases Packages Wiki Activity

Fix Azure image

Browse source
This commit is contained in:
hlohaus 2025-08-06 08:16:27 +02:00
parent 05784da883
commit fe79b11070
2 changed files with 7 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

8
g4f/Provider/needs_auth/Azure.py
View file

@ -87,7 +87,8 @@ class Azure(OpenaiTemplate):
if media: if media:
form = FormData() form = FormData()
form.add_field("prompt", prompt) form.add_field("prompt", prompt)
form.add_field("size", f"{width}x{height}") form.add_field("width", width)
form.add_field("height", height)
output_format = "png" output_format = "png"
for i in range(len(media)): for i in range(len(media)):
if media[i][1] is None and isinstance(media[i][0], str): if media[i][1] is None and isinstance(media[i][0], str):
@ -96,10 +97,12 @@ class Azure(OpenaiTemplate):
for image, image_name in media: for image, image_name in media:
form.add_field(f"image", image, filename=image_name) form.add_field(f"image", image, filename=image_name)
else: else:
api_endpoint = api_endpoint.replace("/edits", "/generations")
data = { data = {
"prompt": prompt, "prompt": prompt,
"n": 1, "n": 1,
"size": f"{width}x{height}", "width": width,
"height": height,
"output_format": output_format, "output_format": output_format,
} }
async with StreamSession(proxy=kwargs.get("proxy"), headers={ async with StreamSession(proxy=kwargs.get("proxy"), headers={
@ -108,7 +111,6 @@ class Azure(OpenaiTemplate):
}) as session: }) as session:
async with session.post(api_endpoint, data=form, json=data) as response: async with session.post(api_endpoint, data=form, json=data) as response:
data = await response.json() data = await response.json()
cls.raise_error(data, response.status)
await raise_for_status(response, data) await raise_for_status(response, data)
async for chunk in save_response_media(data["data"][0]["b64_json"], prompt, content_type=f"image/{output_format}"): async for chunk in save_response_media(data["data"][0]["b64_json"], prompt, content_type=f"image/{output_format}"):
yield chunk yield chunk

4
g4f/image/__init__.py
View file

@ -324,9 +324,9 @@ def get_width_height(
if aspect_ratio == "1:1": if aspect_ratio == "1:1":
return width or 1024, height or 1024 return width or 1024, height or 1024
elif aspect_ratio == "16:9": elif aspect_ratio == "16:9":
return width or 1792, height or 1024 return width or 832, height or 480
elif aspect_ratio == "9:16": elif aspect_ratio == "9:16":
return width or 1024, height or 1792 return width or 480, height or 832,
return width, height return width, height
class ImageRequest: class ImageRequest: