wirehack7/novelai-bot
1
0
Fork 0
Code Issues 1 Pull Requests 1 Actions Packages Releases Activity

more tests
Some checks failed
Safety Check / safety-check (push) Successful in -1s
Details
Build and Push Docker Image / build (push) Successful in -10s
Details
Code Quality Check / quality-check (push) Failing after 14s
Details

Browse Source
This commit is contained in:
wirehack7 2025-05-01 13:53:14 +02:00
parent a288d06345
commit f530071bb2
Signed by: wirehack7
GPG Key ID: C9F4CF85BA505ADC
1 changed files with 8 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
src/main.py
View File

@ -1,5 +1,6 @@
import io import io
import os import os
import sys
import random import random
import zipfile import zipfile
import logging import logging
@ -12,14 +13,13 @@ NOVELAI_API_TOKEN = os.getenv("NOVELAI_API_TOKEN")
BOT_TOKEN = os.getenv("DISCORD_BOT_TOKEN") BOT_TOKEN = os.getenv("DISCORD_BOT_TOKEN")
ALLOWED_CHANNEL_ID = int(os.getenv("DISCORD_CHANNEL_ID", "0")) # falls nicht gesetzt → 0 ALLOWED_CHANNEL_ID = int(os.getenv("DISCORD_CHANNEL_ID", "0")) # falls nicht gesetzt → 0
root = logging.getLogger() logging.basicConfig(
root.setLevel(logging.INFO) stream=sys.stdout,
level=logging.INFO,
format="%(asctime)s - %(levelname)s - %(message)s"
)
handler = logging.StreamHandler(sys.stdout) logger = logging.getLogger(__name__)
handler.setLevel(logging.DEBUG)
formatter = logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s')
handler.setFormatter(formatter)
root.addHandler(handler)
if not all([NOVELAI_API_TOKEN, BOT_TOKEN, ALLOWED_CHANNEL_ID]): if not all([NOVELAI_API_TOKEN, BOT_TOKEN, ALLOWED_CHANNEL_ID]):
raise ValueError("Environment-Variablen NOVELAI_API_TOKEN, DISCORD_BOT_TOKEN und DISCORD_CHANNEL_ID müssen gesetzt sein.") raise ValueError("Environment-Variablen NOVELAI_API_TOKEN, DISCORD_BOT_TOKEN und DISCORD_CHANNEL_ID müssen gesetzt sein.")
@ -160,7 +160,7 @@ async def generate(
) )
if response.status_code == 200: if response.status_code == 200:
antlas = get_available_antlas(NOVELAI_API_TOKEN) antlas = await get_available_antlas(NOVELAI_API_TOKEN)
logging.info("Generated image, current antlas: {antlas}") logging.info("Generated image, current antlas: {antlas}")
response_bytes = response.content response_bytes = response.content