OpenWebUI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2025-06-14 10:20:52 +00:00
Code Issues Actions 4 Packages Projects Releases Wiki Activity

fix: image generation with allowed file extensions

Browse Source
This commit is contained in:
Timothy Jaeryang Baek 2025-05-23 02:53:08 +04:00
parent 4c55ad71a7
commit 7a593b63b2
2 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
backend/open_webui/routers/files.py
View File

@ -85,12 +85,12 @@ def upload_file(
request: Request, request: Request,
file: UploadFile = File(...), file: UploadFile = File(...),
user=Depends(get_verified_user), user=Depends(get_verified_user),
file_metadata: dict = None, metadata: dict = None,
process: bool = Query(True), process: bool = Query(True),
): ):
log.info(f"file.content_type: {file.content_type}") log.info(f"file.content_type: {file.content_type}")
file_metadata = file_metadata if file_metadata else {} file_metadata = metadata if metadata else {}
try: try:
unsanitized_filename = file.filename unsanitized_filename = file.filename
filename = os.path.basename(unsanitized_filename) filename = os.path.basename(unsanitized_filename)
@ -99,7 +99,7 @@ def upload_file(
# Remove the leading dot from the file extension # Remove the leading dot from the file extension
file_extension = file_extension[1:] if file_extension else "" file_extension = file_extension[1:] if file_extension else ""
if request.app.state.config.ALLOWED_FILE_EXTENSIONS: if not file_metadata and request.app.state.config.ALLOWED_FILE_EXTENSIONS:
request.app.state.config.ALLOWED_FILE_EXTENSIONS = [ request.app.state.config.ALLOWED_FILE_EXTENSIONS = [
ext for ext in request.app.state.config.ALLOWED_FILE_EXTENSIONS if ext ext for ext in request.app.state.config.ALLOWED_FILE_EXTENSIONS if ext
] ]

6
backend/open_webui/routers/images.py
View File

@ -451,7 +451,7 @@ def load_url_image_data(url, headers=None):
return None return None
def upload_image(request, image_metadata, image_data, content_type, user): def upload_image(request, image_data, content_type, metadata, user):
image_format = mimetypes.guess_extension(content_type) image_format = mimetypes.guess_extension(content_type)
file = UploadFile( file = UploadFile(
file=io.BytesIO(image_data), file=io.BytesIO(image_data),
@ -460,7 +460,7 @@ def upload_image(request, image_metadata, image_data, content_type, user):
"content-type": content_type, "content-type": content_type,
}, },
) )
file_item = upload_file(request, file, user, file_metadata=image_metadata) file_item = upload_file(request, file, user, metadata=metadata)
url = request.app.url_path_for("get_file_content_by_id", id=file_item.id) url = request.app.url_path_for("get_file_content_by_id", id=file_item.id)
return url return url
@ -527,7 +527,7 @@ async def image_generations(
else: else:
image_data, content_type = load_b64_image_data(image["b64_json"]) image_data, content_type = load_b64_image_data(image["b64_json"])
url = upload_image(request, data, image_data, content_type, user) url = upload_image(request, image_data, content_type, data, user)
images.append({"url": url}) images.append({"url": url})
return images return images