Imaginer
/
Imaginer
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: show button for prefs when hitting rate limit

This commit is contained in:
0xMRTT 2023-05-26 17:42:04 +02:00
parent dbc49f99f4
commit 67919251c8
Signed by: 0xMRTT
GPG Key ID: 910B287304120902
1 changed files with 2 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
src/provider/huggingface.py
View File

@ -36,9 +36,7 @@ class BaseHFProvider(ImaginerProvider):
self.no_api_key()
return ""
elif response.status_code != 200:
self.win.banner.props.title = response.json()["error"]
self.win.banner.props.button_label = ""
self.win.banner.set_revealed(True)
self.no_api_key(title=response.json()["error"])
return ""
response = response.content
except KeyError:
@ -53,8 +51,7 @@ class BaseHFProvider(ImaginerProvider):
return Image.open(io.BytesIO(response))
except UnidentifiedImageError:
error = json.loads(response)["error"]
self.win.banner.set_title(error)
self.win.banner.set_revealed(True)
self.no_api_key(title=error)
return None
else:
print("No response")