issue: ComfyUI image generation broken in v0.6.23 #6160

GiteaMirror commented

@Haervwe commented on GitHub (Aug 22, 2025):

Experiencing the same issue. I cannot call any workflow using the Native implementation, it always returns NoneType is not iterable

@Haervwe commented on GitHub (Aug 22, 2025): Experiencing the same issue. I cannot call any workflow using the Native implementation, it always returns NoneType is not iterable

GiteaMirror commented

@dusty-du commented on GitHub (Aug 22, 2025):

yep same here

@dusty-du commented on GitHub (Aug 22, 2025): yep same here

GiteaMirror commented

@Codelica commented on GitHub (Aug 22, 2025):

Seeing the same with both Comfy and Automatic111 image generation. Reverting Open-WebUI back to 0.6.22 both work properly.

@Codelica commented on GitHub (Aug 22, 2025): Seeing the same with both Comfy and Automatic111 image generation. Reverting Open-WebUI back to 0.6.22 both work properly.

GiteaMirror commented

@GFYURI commented on GitHub (Aug 22, 2025):

This is only a temporary fix. If it gets fixed in a future update, it should be overwritten.
For context, I got this working with the help of an AI, just to clarify.
Before making any changes, create a backup of the file in case something goes wrong or the fix doesn’t work.

In open_webui/routers/images.py, look for the line:

if "x" in form_data.size:

and change it to:

if form_data.size and "x" in form_data.size:

If that fixes everything, great. If you encounter an error like:

TypeError: upload_file() missing 1 required positional argument: 'background_tasks'

try the following. In the same images.py, locate:

def upload_image(request, image_data, content_type, metadata, user):

and replace it with:

def upload_image(request, image_data, content_type, metadata, user):
image_format = mimetypes.guess_extension(content_type)
file = UploadFile(
file=io.BytesIO(image_data),
filename=f"generated-image{image_format}",
headers={
"content-type": content_type,
},
)

try:
    # Try with background_tasks (newer versions)
    from fastapi import BackgroundTasks
    background_tasks = BackgroundTasks()
    
    file_item = upload_file(
        request=request, 
        file=file, 
        metadata=metadata, 
        process=False, 
        user=user,
        background_tasks=background_tasks
    )
except TypeError:
    # Fallback for older versions
    file_item = upload_file(
        request=request, 
        file=file, 
        metadata=metadata, 
        process=False, 
        user=user
    )

url = request.app.url_path_for("get_file_content_by_id", id=file_item.id)
return url

Then just restart and test.

@GFYURI commented on GitHub (Aug 22, 2025): This is only a temporary fix. If it gets fixed in a future update, it should be overwritten. For context, I got this working with the help of an AI, just to clarify. Before making any changes, create a backup of the file in case something goes wrong or the fix doesn’t work. In open_webui/routers/images.py, look for the line: if "x" in form_data.size: and change it to: if form_data.size and "x" in form_data.size: If that fixes everything, great. If you encounter an error like: TypeError: upload_file() missing 1 required positional argument: 'background_tasks' try the following. In the same images.py, locate: def upload_image(request, image_data, content_type, metadata, user): and replace it with: def upload_image(request, image_data, content_type, metadata, user): image_format = mimetypes.guess_extension(content_type) file = UploadFile( file=io.BytesIO(image_data), filename=f"generated-image{image_format}", headers={ "content-type": content_type, }, ) try: # Try with background_tasks (newer versions) from fastapi import BackgroundTasks background_tasks = BackgroundTasks() file_item = upload_file( request=request, file=file, metadata=metadata, process=False, user=user, background_tasks=background_tasks ) except TypeError: # Fallback for older versions file_item = upload_file( request=request, file=file, metadata=metadata, process=False, user=user ) url = request.app.url_path_for("get_file_content_by_id", id=file_item.id) return url Then just restart and test.

GiteaMirror commented

@peiwenxu commented on GitHub (Aug 22, 2025):

Same issue here

@peiwenxu commented on GitHub (Aug 22, 2025): Same issue here

GiteaMirror commented

@ALIENvsROBOT commented on GitHub (Aug 22, 2025):

I think it might be issue because of this fix #16726

@ALIENvsROBOT commented on GitHub (Aug 22, 2025): I think it might be issue because of this fix #16726

GiteaMirror commented

@tjbck commented on GitHub (Aug 22, 2025):

Should be addressed with 88a5b5ff4c

@tjbck commented on GitHub (Aug 22, 2025): Should be addressed with 88a5b5ff4c40a339aecb796846cfc5223407ce61

GiteaMirror commented

@anthrotype commented on GitHub (Aug 22, 2025):

https://github.com/open-webui/open-webui/commit/88a5b5ff4c40a339aecb796846cfc5223407ce61 only fixed the

TypeError: argument of type 'NoneType' is not iterable

But after fixing that, I am still getting another error, same as @GFYURI wrote above

TypeError: upload_file() missing 1 required positional argument: 'background_tasks'

Note that this is not just ComfyUI image generation but any image generation (I'm using Gemini)

@anthrotype commented on GitHub (Aug 22, 2025): https://github.com/open-webui/open-webui/commit/88a5b5ff4c40a339aecb796846cfc5223407ce61 only fixed the ``` TypeError: argument of type 'NoneType' is not iterable ``` But after fixing that, I am still getting another error, same as @GFYURI wrote above ``` TypeError: upload_file() missing 1 required positional argument: 'background_tasks' ``` Note that this is not just ComfyUI image generation but any image generation (I'm using Gemini)

GiteaMirror commented

@fighter3005 commented on GitHub (Aug 22, 2025):

Yes, I just tried and can confirm. Still the same error.

I am a bit confused, since the upload_file method is only used for uploading images!? I guess

file_item = upload_file(
        request, None, file=file, metadata=metadata, process=False, user=user
    )

should work in image.py

def upload_image(request, image_data, content_type, metadata, user):
    image_format = mimetypes.guess_extension(content_type)
    file = UploadFile(
        file=io.BytesIO(image_data),
        filename=f"generated-image{image_format}",  # will be converted to a unique ID on upload_file
        headers={
            "content-type": content_type,
        },
    )
    file_item = upload_file(
        request, file=file, metadata=metadata, process=False, user=user  // <------------- HERE
    )
    url = request.app.url_path_for("get_file_content_by_id", id=file_item.id)
    return url

@fighter3005 commented on GitHub (Aug 22, 2025): Yes, I just tried and can confirm. Still the same error. I am a bit confused, since the upload_file method is only used for uploading images!? I guess ``` file_item = upload_file( request, None, file=file, metadata=metadata, process=False, user=user ) ``` should work in image.py ``` def upload_image(request, image_data, content_type, metadata, user): image_format = mimetypes.guess_extension(content_type) file = UploadFile( file=io.BytesIO(image_data), filename=f"generated-image{image_format}", # will be converted to a unique ID on upload_file headers={ "content-type": content_type, }, ) file_item = upload_file( request, file=file, metadata=metadata, process=False, user=user // <------------- HERE ) url = request.app.url_path_for("get_file_content_by_id", id=file_item.id) return url ```

GiteaMirror commented

@nihilimbo commented on GitHub (Aug 22, 2025):

Same here, the prompt is correctly sent to ComfyUI, the image is generated (I can see it in the ComfyUI Dashboard) but OWUI fails to retrieve and show the generated image.

So much for a critital bug fix...

@nihilimbo commented on GitHub (Aug 22, 2025): Same here, the prompt is correctly sent to ComfyUI, the image is generated (I can see it in the ComfyUI Dashboard) but OWUI fails to retrieve and show the generated image. So much for a critital bug fix...

GiteaMirror commented