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

fix: handle json decode error
All checks were successful
ci/woodpecker/push/woodpecker Pipeline was successful
Details

Browse source
This commit is contained in:
0xMRTT 2023-05-26 20:01:17 +02:00
parent e36e58c1c8
commit 85fc78a222
No known key found for this signature in database
GPG key ID: 910B287304120902
1 changed files with 9 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
src/provider/stablediffusionlocal.py
View file

@ -36,14 +36,21 @@ class StableDiffusionLocalProvider(ImaginerProvider):
self.no_api_key()
return ""
elif response.status_code != 200:
self.no_api_key(title=response.json()["error"])
return ""
try:
self.no_api_key(title=response.json()["error"])
return ""
except json.decoder.JSONDecodeError:
self.no_api_key(title="No response")
return ""
response = response.json()
except KeyError:
pass
except socket.gaierror:
self.no_connection()
return ""
except json.decoder.JSONDecodeError:
self.no_api_key(title="No response")
return ""
else:
self.hide_banner()
if response: