ti.sil/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui.git synced 2025-12-13 12:55:19 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

refac

Browse source
This commit is contained in:
Timothy Jaeryang Baek 2025-11-05 03:42:34 -05:00
parent 72f8539fd2
commit 5c1ba23026
1 changed files with 30 additions and 21 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

33
backend/open_webui/routers/images.py
View file

@ -737,7 +737,6 @@ async def image_edits(
form_data: EditImageForm,
user=Depends(get_verified_user),
):
size = None
width, height = None, None
if (
@ -757,29 +756,39 @@ async def image_edits(
else form_data.model
)
def load_url_image(string):
if string.startswith("http://") or string.startswith("https://"):
r = requests.get(string)
try:
async def load_url_image(data):
if data.startswith("http://") or data.startswith("https://"):
r = await asyncio.to_thread(requests.get, data)
r.raise_for_status()
image_data = base64.b64encode(r.content).decode("utf-8")
return f"data:{r.headers['content-type']};base64,{image_data}"
elif string.startswith("/api/v1/files"):
file_id = string.split("/api/v1/files/")[1].split("/content")[0]
file_response = get_file_content_by_id(file_id, user)
elif data.startswith("/api/v1/files"):
file_id = data.split("/api/v1/files/")[1].split("/content")[0]
file_response = await get_file_content_by_id(file_id, user)
if isinstance(file_response, FileResponse):
file_bytes = file_response.body
mime_type = file_response.headers.get("content-type", "image/png")
file_path = file_response.path
with open(file_path, "rb") as f:
file_bytes = f.read()
image_data = base64.b64encode(file_bytes).decode("utf-8")
mime_type, _ = mimetypes.guess_type(file_path)
return f"data:{mime_type};base64,{image_data}"
return string
return data
# Load image(s) from URL(s) if necessary
if isinstance(form_data.image, str):
form_data.image = load_url_image(form_data.image)
form_data.image = await load_url_image(form_data.image)
elif isinstance(form_data.image, list):
form_data.image = [load_url_image(img) for img in form_data.image]
form_data.image = [await load_url_image(img) for img in form_data.image]
except Exception as e:
raise HTTPException(status_code=400, detail=ERROR_MESSAGES.DEFAULT(e))
r = None
try: