From 85fc78a222ba0a5fd430a0cb77715cd93e822d77 Mon Sep 17 00:00:00 2001
From: 0xMRTT <0xMRTT@proton.me>
Date: Fri, 26 May 2023 20:01:17 +0200
Subject: [PATCH] fix: handle json decode error

---
 src/provider/stablediffusionlocal.py | 11 +++++++++--
 1 file changed, 9 insertions(+), 2 deletions(-)

diff --git a/src/provider/stablediffusionlocal.py b/src/provider/stablediffusionlocal.py
index 39409bb..61c89b9 100644
--- a/src/provider/stablediffusionlocal.py
+++ b/src/provider/stablediffusionlocal.py
@@ -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: