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

Sync with main #4

Merged
wirehack7 merged 12 commits from main into dev 2025-05-01 14:07:15 +02:00
Conversation 0 Commits 12 Files Changed 2 +13 -18
1 changed files with 13 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files
Showing only changes of commit fb52c427e3 - Show all commits

31
src/main.py
View File

@ -19,18 +19,18 @@ bot = commands.Bot(command_prefix=commands.when_mentioned, intents=intents)
@bot.event @bot.event
async def on_ready(): async def on_ready():
print(f"Bot gestartet als {bot.user}") print(f"Bot started as {bot.user}")
try: try:
synced = await bot.tree.sync() synced = await bot.tree.sync()
print(f"{len(synced)} Slash-Commands synchronisiert.") print(f"{len(synced)} Slash-Commands synchronized.")
except Exception as e: except Exception as e:
print(e) print(e)
@bot.tree.command(name="generate", description="Generiere ein Bild mit NovelAI v4") @bot.tree.command(name="generate", description="Generate image with NovelAI v4 Full")
@app_commands.describe( @app_commands.describe(
prompt="Was soll dargestellt werden?", prompt="What should be generated?",
undesired_prompt="Was soll vermieden werden? (optional)", undesired_prompt="What should be avoided? (optional)",
orientation="portrait oder landscape (Standard: portrait)" orientation="portrait or landscape (Standard: portrait)"
) )
async def generate( async def generate(
interaction: discord.Interaction, interaction: discord.Interaction,
@ -40,7 +40,7 @@ async def generate(
): ):
if interaction.channel.id != ALLOWED_CHANNEL_ID: if interaction.channel.id != ALLOWED_CHANNEL_ID:
await interaction.response.send_message( await interaction.response.send_message(
"Dieser Befehl ist in diesem Kanal nicht erlaubt.", "This command isn't allowed here.",
ephemeral=True ephemeral=True
) )
return return
@ -56,12 +56,7 @@ async def generate(
default_negative = ( default_negative = (
"blurry, lowres, error, film grain, scan artifacts, worst quality, bad quality, jpeg artifacts, " "blurry, lowres, error, film grain, scan artifacts, worst quality, bad quality, jpeg artifacts, "
"very displeasing, chromatic aberration, multiple views, logo, too many watermarks, white blank page, " "very displeasing, chromatic aberration, multiple views, logo, too many watermarks, white blank page, "
"blank page, " "blank page, watermarks, watermark, text, "
)
negative = undesired_prompt.strip() or (
"blurry, lowres, error, film grain, scan artifacts, worst quality, bad quality, jpeg artifacts, "
"very displeasing, chromatic aberration, multiple views, logo, too many watermarks, white blank page, "
"blank page"
) )
seed = random.randint(0, 2**32 - 1) seed = random.randint(0, 2**32 - 1)
@ -93,7 +88,7 @@ async def generate(
"use_coords": False, "use_coords": False,
"legacy_uc": False, "legacy_uc": False,
"normalize_reference_strength_multiple": True, "normalize_reference_strength_multiple": True,
#"seed": seed, "seed": seed,
"characterPrompts": [], "characterPrompts": [],
"v4_prompt": { "v4_prompt": {
"caption": { "caption": {
@ -132,7 +127,7 @@ async def generate(
response_bytes = response.content response_bytes = response.content
param_text = f"""```Prompt: {prompt} param_text = f"""```Prompt: {prompt}
Negativer Prompt: {negative} Negativer Prompt: {default_negative + undesired_prompt.strip()}
Seed: {seed} Seed: {seed}
Auflösung: {width}x{height} Auflösung: {width}x{height}
Sampler: k_euler_ancestral Sampler: k_euler_ancestral
@ -155,7 +150,7 @@ Model: nai-diffusion-4-full```"""
await interaction.followup.send(content=param_text, file=file) await interaction.followup.send(content=param_text, file=file)
else: else:
await interaction.followup.send( await interaction.followup.send(
"ZIP erhalten, aber kein PNG-Bild gefunden." "Found no PNG inside archive"
) )
elif response_bytes[:4] == b'\x89PNG': elif response_bytes[:4] == b'\x89PNG':
@ -166,7 +161,7 @@ Model: nai-diffusion-4-full```"""
else: else:
await interaction.followup.send( await interaction.followup.send(
"Die API hat Daten gesendet, aber es war weder ein Bild noch ein gültiges ZIP-Archiv." "API didn't send any file"
) )
else: else:
@ -175,7 +170,7 @@ Model: nai-diffusion-4-full```"""
except Exception: except Exception:
error_data = {"error": response.text} error_data = {"error": response.text}
error_message = f"Fehler {response.status_code} bei der API-Anfrage.\n" error_message = f"Error {response.status_code} at API-request.\n"
for key, value in error_data.items(): for key, value in error_data.items():
error_message += f"**{key}**: {value}\n" error_message += f"**{key}**: {value}\n"