.gitea/workflows/check-code_quality.yaml

@@ -15,12 +15,12 @@ jobs:
       GITHUB: ${{ secrets.GITHUB }}
     steps:
       - name: Checkout code
-        uses: actions/checkout@v6
+        uses: actions/checkout@v4
 
       - name: Set up Python
-        uses: actions/setup-python@v6
+        uses: actions/setup-python@v5
         with:
-          python-version: '3.14'
+          python-version: '3.10'
 
       - name: Install dependencies
         run: |
@@ -85,7 +85,7 @@ jobs:
       
       # Ergebnisse als Artefakte speichern
       - name: Upload results as artifacts
-        uses: actions/upload-artifact@v5
+        uses: actions/upload-artifact@v3
         with:
           name: quality-reports
           path: /tmp/results

.gitea/workflows/release.yaml

@@ -10,7 +10,7 @@ jobs:
     runs-on: ubuntu-latest
     steps:
       - name: Checkout Repository
-        uses: actions/checkout@v6
+        uses: actions/checkout@v4
 
       - name: Verify Docker installation
         run: docker --version
@@ -20,15 +20,15 @@ jobs:
 
       - name: Build Docker Image
         run: |
-          docker build -f src/Dockerfile -t git.0day.agency/${{ secrets.REGISTRY_USERNAME }}/novelai-bot:${{ gitea.ref_name }} src/
+          docker build -f src/Dockerfile -t git.0day.agency/${{ secrets.REGISTRY_USERNAME }}/novelai-discord-bot:${{ gitea.ref_name }} src/
           # latest-Tag setzen
-          docker tag git.0day.agency/${{ secrets.REGISTRY_USERNAME }}/novelai-bot:${{ gitea.ref_name }} \
+          docker tag git.0day.agency/${{ secrets.REGISTRY_USERNAME }}/novelai-discord-bot:${{ gitea.ref_name }} \
-                     git.0day.agency/${{ secrets.REGISTRY_USERNAME }}/novelai-bot:latest
+                     git.0day.agency/${{ secrets.REGISTRY_USERNAME }}/novelai-discord-bot:latest
 
       - name: Push Docker Image mit Versionstag
         run: |
-          docker push git.0day.agency/${{ secrets.REGISTRY_USERNAME }}/novelai-bot:${{ gitea.ref_name }}
+          docker push git.0day.agency/${{ secrets.REGISTRY_USERNAME }}/novelai-discord-bot:${{ gitea.ref_name }}
 
       - name: Push Docker Image als latest
         run: |
-          docker push git.0day.agency/${{ secrets.REGISTRY_USERNAME }}/novelai-bot:latest
+          docker push git.0day.agency/${{ secrets.REGISTRY_USERNAME }}/novelai-discord-bot:latest

.github/renovate.json

@@ -1,20 +0,0 @@
-{
-    "$schema": "https://docs.renovatebot.com/renovate-schema.json",
-    "extends": [
-      "config:recommended"
-    ],
-    "baseBranches": [
-      "dev"
-    ],
-    "packageRules": [
-      {
-        "matchUpdateTypes": [
-          "minor",
-          "patch",
-          "pin",
-          "digest"
-        ],
-        "automerge": false
-      }
-    ]
-  }

README.md

@@ -1,52 +1,2 @@
-# NovelAI Discord Bot
+# novelai-bot
 
-![Code Quality](https://git.0day.agency/wirehack7/novelai-bot/actions/workflows/check-code_quality.yaml/badge.svg)
-![Safety Check](https://git.0day.agency/wirehack7/novelai-bot/actions/workflows/check-safety.yaml/badge.svg)
-![Release](https://git.0day.agency/wirehack7/novelai-bot/actions/workflows/release.yaml/badge.svg)
-
-Simple bot which registers a command to generate images via the NovelAI API.
-
-## Usage
-
-Just enter `/generate` in the desired channel.
-
-```text
-/generate 
-    *prompt: string
-    undesired_prompt: string
-    orientation: landscape or portrait
-    seed: integer
-```
-
-`*: required`
-
-## Run
-
-You might create environvent vars:
-
-- **NOVELAI_API_TOKEN**: your NovelAI user API key
-- **DISCORD_BOT_TOKEN**: token of your Discord bot
-- **DISCORD_CHANNEL_ID**: ID of the channel to enable this bot
-
-Don't forget to install the requirements in [requirements.txt](src/requirements.txt)
-
-## Docker Compose
-
-Or simply run it via **docker-compose**:
-
-```yaml
-services:
-  bot:
-    restart: unless-stopped
-    image: git.0day.agency/wirehack7/novelai-bot:latest
-    environment:
-      - NOVELAI_API_TOKEN=
-      - DISCORD_BOT_TOKEN=
-      - DISCORD_CHANNEL_ID=
-networks: {}
-```
-
-## Notes
-
-To create a Discord bot go to [Discord Developers](https://discord.com/developers/).
-You should make the bot private.

src/Dockerfile

@@ -1,4 +1,4 @@
-FROM python:3.14.2-slim
+FROM python:3.13.3-slim
 
 WORKDIR /app
 

src/main.py

@@ -1,115 +1,94 @@
-"""Discord Bot zur Bildgenerierung mit NovelAI API."""
-
-# flake8: noqa E501
-
 import io
-import logging
 import os
 import random
-import sys
-import time
 import zipfile
-
+import logging
-import aiohttp
 import discord
 from discord import app_commands
 from discord.ext import commands
+import requests
 
-# ENV Variablen
 NOVELAI_API_TOKEN = os.getenv("NOVELAI_API_TOKEN")
 BOT_TOKEN = os.getenv("DISCORD_BOT_TOKEN")
-ALLOWED_CHANNEL_ID = int(os.getenv("DISCORD_CHANNEL_ID", "0"))
+ALLOWED_CHANNEL_ID = int(os.getenv("DISCORD_CHANNEL_ID", "0"))  # falls nicht gesetzt → 0
-
-# Logging Setup
-logging.basicConfig(
-    stream=sys.stdout,
-    level=logging.INFO,
-    format="%(asctime)s - %(levelname)s - %(message)s",
-)
-logger = logging.getLogger(__name__)
 
 if not all([NOVELAI_API_TOKEN, BOT_TOKEN, ALLOWED_CHANNEL_ID]):
-    raise ValueError(
+    raise ValueError("Environment-Variablen NOVELAI_API_TOKEN, DISCORD_BOT_TOKEN und DISCORD_CHANNEL_ID müssen gesetzt sein.")
-        "NOVELAI_API_TOKEN, DISCORD_BOT_TOKEN and DISCORD_CHANNEL_ID have to be set."
-    )
 
 intents = discord.Intents.default()
 bot = commands.Bot(command_prefix=commands.when_mentioned, intents=intents)
 
+async def get_available_antlas(api_token):
+    headers = {
+        "Authorization": f"Bearer {api_token}"
+    }
+    response = requests.get("https://api.novelai.net/user/info", headers=headers)
+    if response.status_code == 200:
+        data = response.json()
+        return data.get("imageAnlas", 0)
+    else:
+        logging.error(f"Fehler beim Abrufen der Antlas: {response.status_code}")
+        return None
 
 @bot.event
 async def on_ready():
-    """Bot Startup Ereignis."""
+    print(f"Bot started as {bot.user}")
-    print(f"Bot gestartet als {bot.user}")
     try:
         synced = await bot.tree.sync()
-        logger.info("%d Slash-Commands synchronized.", len(synced))
+        logging.info(f"{len(synced)} Slash-Commands synchronized.")
-    except Exception as err:  # pylint: disable=W0718
+    except Exception as e:
-        logger.error("error while syncing the command: %s", err)
+        logging.error(e)
-
-    activity = discord.Game(name="generating juicy NovelAI images 🥵")
-    await bot.change_presence(status=discord.Status.online, activity=activity)
-
 
 @bot.tree.command(name="generate", description="Generate image with NovelAI v4 Full")
 @app_commands.describe(
     prompt="What should be generated?",
     undesired_prompt="What should be avoided? (optional)",
     orientation="portrait or landscape (Standard: portrait)",
-    seed="Optional seed (integer). If empty, a random one will be generated",
+    seed="Optional seed (integer). If empty, a random one will be generated"
 )
-# pylint: disable=too-many-locals,too-many-return-statements,too-many-statements
 async def generate(
     interaction: discord.Interaction,
     prompt: str,
     undesired_prompt: str = "",
     orientation: str = "portrait",
-    seed: int = None,
+    seed: int = None
 ):
-    """Slash-Command zur Bildgenerierung."""
+    
     default_negative = (
-        "blurry, lowres, error, film grain, scan artifacts, worst quality, "
+        "blurry, lowres, error, film grain, scan artifacts, worst quality, bad quality, jpeg artifacts, "
-        "bad quality, jpeg artifacts, very displeasing, chromatic aberration, "
+        "very displeasing, chromatic aberration, multiple views, logo, too many watermarks, white blank page, "
-        "multiple views, logo, too many watermarks, white blank page, blank page, "
+        "blank page, watermarks, watermark, text, "
-        "watermarks, watermark, text, "
     )
     
     if interaction.channel.id != ALLOWED_CHANNEL_ID:
         await interaction.response.send_message(
-            "This command isn't allowed here.", ephemeral=True
+            "This command isn't allowed here.",
+            ephemeral=True
         )
         return
     
-    max_prompt_length = 500
+    MAX_PROMPT_LENGTH = 500
-    negative_prompt = default_negative + undesired_prompt.strip()
 
-    if len(prompt) > max_prompt_length:
+    if len(prompt) > MAX_PROMPT_LENGTH:
-        await interaction.response.send_message(
+        await interaction.followup.send(f"Prompt too long! Maximum {MAX_PROMPT_LENGTH} chars allowed.")
-            f"Prompt too long! Max {max_prompt_length} characters.", ephemeral=True
-        )
         return
 
-    if len(negative_prompt) > max_prompt_length:
+    if len(default_negative + undesired_prompt.strip()) > MAX_PROMPT_LENGTH:
-        await interaction.response.send_message(
+        await interaction.followup.send(f"Negative Prompt too long! Maximum {MAX_PROMPT_LENGTH} chars allowed.")
-            f"Negative prompt too long! Max {max_prompt_length} characters.",
-            ephemeral=True,
-        )
         return
 
     await interaction.response.defer(thinking=True)
 
-    logger.info(
+    # Auflösung wählen
-        "User %s (%s, %s) requested image",
+    if orientation.lower() == "landscape":
-        interaction.user.display_name,
+        width, height = 1216, 832
-        interaction.user.name,
+    else:
-        interaction.user.id,
+        width, height = 832, 1216
-    )
-
-    width, height = (1216, 832) if orientation.lower() == "landscape" else (832, 1216)
 
     if seed is None:
         seed = random.randint(0, 2**32 - 1)
 
+    # API-Payload zusammenbauen
     payload = {
         "input": prompt,
         "model": "nai-diffusion-4-full",
@@ -139,88 +118,90 @@ async def generate(
             "seed": seed,
             "characterPrompts": [],
             "v4_prompt": {
-                "caption": {"base_caption": prompt, "char_captions": []},
+                "caption": {
+                    "base_caption": prompt,
+                    "char_captions": []
+                },
                 "use_coords": False,
-                "use_order": True,
+                "use_order": True
             },
             "v4_negative_prompt": {
-                "caption": {"base_caption": negative_prompt, "char_captions": []},
+                "caption": {
-                "legacy_uc": False,
+                    "base_caption": default_negative + undesired_prompt.strip(),
+                    "char_captions": []
+                },
+                "legacy_uc": False
             },
-            "negative_prompt": negative_prompt,
+            "negative_prompt": default_negative + undesired_prompt.strip(),
             "deliberate_euler_ancestral_bug": False,
-            "prefer_brownian": True,
+            "prefer_brownian": True
-        },
+        }
     }
 
     headers = {
         "Authorization": f"Bearer {NOVELAI_API_TOKEN}",
-        "Content-Type": "application/json",
+        "Content-Type": "application/json"
     }
 
-    start_time = time.monotonic()
+    response = requests.post(
-    try:
+        "https://image.novelai.net/ai/generate-image",
-        async with aiohttp.ClientSession() as session:
+        json=payload,
-            async with session.post(
+        headers=headers,
-                "https://image.novelai.net/ai/generate-image",
+        timeout=120
-                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
+    if response.status_code == 200:
-    logger.info("Image creation took %.2f seconds", duration)
+        antlas = get_available_antlas(NOVELAI_API_TOKEN)
+        logging.info("Generated image, current antlas: {antlas}")
+        response_bytes = response.content
 
-    if status_code == 200:
+        param_text = f"""```Prompt: {prompt}
-        param_text = (
+Undesired prompt: {default_negative + undesired_prompt.strip()}
-            f"```Prompt: {prompt}\n"
+Seed: {seed}
-            f"Undesired prompt: {negative_prompt}\n"
+Resolution: {width}x{height}
-            f"Seed: {seed}\n"
+Sampler: k_euler_ancestral
-            f"Resolution: {width}x{height}\n"
+Steps: 28
-            f"Sampler: k_euler_ancestral\n"
+Scale: 3.7
-            f"Steps: 28\n"
+Model: nai-diffusion-4-full```"""
-            f"Scale: 3.7\n"
-            f"Model: nai-diffusion-4-full```"
-        )
 
-        if response_bytes[:2] == b"PK":
+        # Prüfen ob ZIP (PK am Anfang)
+        if response_bytes[:2] == b'PK':
+            # ZIP-Archiv entpacken
             with zipfile.ZipFile(io.BytesIO(response_bytes)) as zip_file:
                 namelist = zip_file.namelist()
-                image_name = next(
+                # Erstes PNG suchen
-                    (name for name in namelist if name.endswith(".png")), None
+                image_name = next((name for name in namelist if name.endswith(".png")), None)
-                )
+
                 if image_name:
                     image_bytes = zip_file.read(image_name)
                     filename = "novelai.png"
                     file = discord.File(io.BytesIO(image_bytes), filename=filename)
                     await interaction.followup.send(content=param_text, file=file)
-                    return
+                else:
-            await interaction.followup.send("Found no PNG inside archive")
+                    await interaction.followup.send(
-            return
+                        "Found no PNG inside archive"
+                    )
 
-        if response_bytes[:4] == b"\x89PNG":
+        elif response_bytes[:4] == b'\x89PNG':
+            # Direkte PNG-Rückgabe
             filename = "novelai.png"
             file = discord.File(io.BytesIO(response_bytes), filename=filename)
             await interaction.followup.send(content=param_text, file=file)
-            return
 
-        await interaction.followup.send("API didn't send any file")
+        else:
-        return
+            await interaction.followup.send(
+                "API didn't send any file"
+            )
 
-    try:
+    else:
-        error_data = response_bytes.decode("utf-8")
+        try:
-    except Exception:  # pylint: disable=W0718
+            error_data = response.json()
-        error_data = "error message unreadable"
+        except Exception:
+            error_data = {"error": response.text}
 
-        error_message = (
+        error_message = f"Error {response.status_code} at API-request.\n"
-            f"Error {status_code} at API-request.\n" f"Answer: {error_data}\n"
+        for key, value in error_data.items():
-        )
+            error_message += f"**{key}**: {value}\n"
 
         await interaction.followup.send(error_message)
 

src/requirements.txt

@@ -1,2 +1,2 @@
 discord
-aiohttp
+requests