From 90c97d3f54323c7ec4eb996dd332590f2df62d64 Mon Sep 17 00:00:00 2001
From: wirehack7 <wirehack7@protonmail.ch>
Date: Thu, 1 May 2025 19:25:48 +0200
Subject: [PATCH 1/3] better timeout handling

---
 src/main.py | 28 ++++++++++++++++------------
 1 file changed, 16 insertions(+), 12 deletions(-)

diff --git a/src/main.py b/src/main.py
index ec7718d..31a993f 100644
--- a/src/main.py
+++ b/src/main.py
@@ -159,16 +159,20 @@ async def generate(
     }
 
     start_time = time.monotonic()
-
-    async with aiohttp.ClientSession() as session:
-        async with session.post(
-            "https://image.novelai.net/ai/generate-image",
-            json=payload,
-            headers=headers,
-            timeout=aiohttp.ClientTimeout(total=120),
-        ) as response:
-            response_bytes = await response.read()
-            status_code = response.status
+    try:
+        async with aiohttp.ClientSession() as session:
+            async with session.post(
+                "https://image.novelai.net/ai/generate-image",
+                json=payload,
+                headers=headers,
+                timeout=aiohttp.ClientTimeout(total=120),
+            ) as response:
+                response_bytes = await response.read()
+                status_code = response.status
+    except aiohttp.TimeoutError:
+        await interaction.followup.send("Error: the request timed out.")
+    except aiohttp.ClientError as e:
+        await interaction.followup.send(f"Error: {e}")
 
     duration = time.monotonic() - start_time
     logger.info("Image creation took %.2f seconds", duration)
@@ -214,9 +218,9 @@ async def generate(
     except Exception:  # pylint: disable=W0718
         error_data = "error message unreadable"
 
-    error_message = f"Error {status_code} at API-request.\n" f"Answer: {error_data}\n"
+        error_message = f"Error {status_code} at API-request.\n" f"Answer: {error_data}\n"
 
-    await interaction.followup.send(error_message)
+        await interaction.followup.send(error_message)
 
 
 bot.run(BOT_TOKEN)

From b3045bf7243b5acf663c46ebdbfa0a0531f11f97 Mon Sep 17 00:00:00 2001
From: wirehack7 <wirehack7@protonmail.ch>
Date: Thu, 1 May 2025 19:28:14 +0200
Subject: [PATCH 2/3] reformatted

---
 src/main.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/src/main.py b/src/main.py
index 31a993f..c9984e8 100644
--- a/src/main.py
+++ b/src/main.py
@@ -218,7 +218,9 @@ async def generate(
     except Exception:  # pylint: disable=W0718
         error_data = "error message unreadable"
 
-        error_message = f"Error {status_code} at API-request.\n" f"Answer: {error_data}\n"
+        error_message = (
+            f"Error {status_code} at API-request.\n" f"Answer: {error_data}\n"
+        )
 
         await interaction.followup.send(error_message)
 

From 9145ac1bb4b08ed00883bb327b47d9adc899d4ff Mon Sep 17 00:00:00 2001
From: wirehack7 <wirehack7@protonmail.ch>
Date: Thu, 1 May 2025 19:30:45 +0200
Subject: [PATCH 3/3] =?UTF-8?q?pylint,=20fuck=20you!=20=F0=9F=96=95?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/main.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/main.py b/src/main.py
index c9984e8..c99f54a 100644
--- a/src/main.py
+++ b/src/main.py
@@ -58,7 +58,7 @@ async def on_ready():
     orientation="portrait or landscape (Standard: portrait)",
     seed="Optional seed (integer). If empty, a random one will be generated",
 )
-# pylint: disable=too-many-locals,too-many-return-statements
+# pylint: disable=too-many-locals,too-many-return-statements,too-many-statements
 async def generate(
     interaction: discord.Interaction,
     prompt: str,