hoelee/stable-diffusion-webui
1
0
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2025-08-08 13:19:54 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Download image client-side

Browse Source
This commit is contained in:
missionfloyd
2024-03-16 16:08:42 -06:00
parent 3da13f0cc9
commit 83a9dd82db
3 changed files with 20 additions and 45 deletions
Download Patch File Download Diff File Expand all files Collapse all files

35
modules/images.py
View File

@@ -772,31 +772,18 @@ Steps: {json_info["steps"]}, Sampler: {sampler}, CFG scale: {json_info["scale"]}
def image_data(data):
import gradio as gr
if not data:
return gr.update(), None
if isinstance(data, bytes):
try:
image = Image.open(io.BytesIO(data))
textinfo, _ = read_info_from_image(image)
return textinfo, None
except Exception:
pass
try:
text = data.decode('utf8')
assert len(text) < 10000
return text, None
except Exception:
pass
import requests
try:
r = requests.get(data, timeout=5)
if r.status_code == 200:
image = Image.open(io.BytesIO(r.content))
textinfo, _ = read_info_from_image(image)
return textinfo, None
image = read(io.BytesIO(data))
textinfo, _ = read_info_from_image(image)
return textinfo, None
except Exception:
pass
try:
text = data.decode('utf8')
assert len(text) < 10000
return text, None
except Exception:
pass