Take current input into account for autocomplete + reformat
All checks were successful
Build Heidi Docker image / build-docker (push) Successful in 17s
All checks were successful
Build Heidi Docker image / build-docker (push) Successful in 17s
This commit is contained in:
171
bot.py
171
bot.py
@ -5,7 +5,7 @@ from discord import app_commands
|
||||
from discord.app_commands import Choice
|
||||
from functools import reduce
|
||||
from dotenv import load_dotenv
|
||||
from typing import Optional, Union
|
||||
from typing import List, Optional, Union
|
||||
|
||||
# TODO: Reenable + extend textgen
|
||||
# from textgen import textgen
|
||||
@ -17,28 +17,27 @@ from typing import Optional, Union
|
||||
|
||||
# We're fancy today
|
||||
from rich.traceback import install
|
||||
|
||||
install(show_locals=True)
|
||||
|
||||
# ================================================================================================ #
|
||||
# ================================================================================================ #
|
||||
# NOTE: Always set this correctly: #
|
||||
DOCKER = True #
|
||||
DOCKER = False #
|
||||
# ================================================================================================ #
|
||||
# ================================================================================================ #
|
||||
|
||||
# DONE: Migrate back to discord.py
|
||||
# DONE: Rewrite bot with slash commands (and making actual use of discord.py)
|
||||
# TODO: Insult statistics (you have insulted 20 times)
|
||||
# TODO: Only post in heidi-spam channel
|
||||
# TODO: yt-dlp music support
|
||||
# TODO: Somehow upload voicelines more easily (from discord voice message?), also need to be distributed to folders so no more than 25 lines per folder
|
||||
# TODO: Reenable text/quote generation, allow uploading of training text files, allow switching "personalities" (/elon generates elon quote?)
|
||||
# TODO: Zalgo generator
|
||||
# TODO: Somehow upload voicelines more easily (from discord voice message?)
|
||||
|
||||
# IDs of the servers Heidi is used on
|
||||
LINUS_GUILD = discord.Object(id=431154792308408340)
|
||||
TEST_GUILD = discord.Object(id=821511861178204161)
|
||||
|
||||
|
||||
class HeidiClient(discord.Client):
|
||||
def __init__(self, *, intents: discord.Intents):
|
||||
super().__init__(status="Nur eine kann GNTM werden!", intents=intents)
|
||||
@ -53,33 +52,33 @@ class HeidiClient(discord.Client):
|
||||
# if the predicate is true the action is performed
|
||||
self.auto_triggers = {
|
||||
# lambda m: m.author.nick.lower() in self.models.get_in_names(): self.autoreact_to_girls,
|
||||
lambda m: "jeremy" in m.author.nick.lower(): self._autoreact_to_jeremy
|
||||
lambda m: "jeremy"
|
||||
in m.author.nick.lower(): self._autoreact_to_jeremy
|
||||
}
|
||||
|
||||
# Textgen
|
||||
self.textgen_models: dict[str, textgen] = {
|
||||
# The name must correspond to the name of the training text file
|
||||
# "kommunistisches_manifest": LSTMTextGenerator(10),
|
||||
# "musk": LSTMTextGenerator(10),
|
||||
# "bibel": LSTMTextGenerator(10)
|
||||
# self.textgen_models: dict[str, textgen] = {
|
||||
# # The name must correspond to the name of the training text file
|
||||
# "kommunistisches_manifest": LSTMTextGenerator(10),
|
||||
# "musk": LSTMTextGenerator(10),
|
||||
# "bibel": LSTMTextGenerator(10)
|
||||
# "bibel": MarkovTextGenerator(3), # Prefix length of 3
|
||||
# "kommunistisches_manifest": MarkovTextGenerator(3),
|
||||
# "musk": MarkovTextGenerator(3)
|
||||
# }
|
||||
|
||||
# "bibel": MarkovTextGenerator(3), # Prefix length of 3
|
||||
# "kommunistisches_manifest": MarkovTextGenerator(3),
|
||||
# "musk": MarkovTextGenerator(3)
|
||||
}
|
||||
# for name, model in self.textgen_models.items():
|
||||
# model.init(name) # Loads the textfile
|
||||
|
||||
for name, model in self.textgen_models.items():
|
||||
model.init(name) # Loads the textfile
|
||||
# if os.path.exists(f"weights/{name}_lstm_model.pt"):
|
||||
# model.load()
|
||||
# elif not DOCKER:
|
||||
# model.train()
|
||||
# else:
|
||||
# print("Error: Can't load model", name)
|
||||
|
||||
if os.path.exists(f"weights/{name}_lstm_model.pt"):
|
||||
model.load()
|
||||
elif not DOCKER:
|
||||
model.train()
|
||||
else:
|
||||
print("Error: Can't load model", name)
|
||||
|
||||
print("Generating test sentence for", name)
|
||||
self.textgen_models[name].generate_sentence()
|
||||
# print("Generating test sentence for", name)
|
||||
# self.textgen_models[name].generate_sentence()
|
||||
|
||||
# Synchronize commands to guilds
|
||||
async def setup_hook(self):
|
||||
@ -113,7 +112,6 @@ class HeidiClient(discord.Client):
|
||||
# picture.set_footer(text=name)
|
||||
# await message.channel.send(embed=picture)
|
||||
|
||||
|
||||
# Automatic Actions --------------------------------------------------------------------------
|
||||
|
||||
# @staticmethod
|
||||
@ -130,15 +128,16 @@ class HeidiClient(discord.Client):
|
||||
"""
|
||||
await message.add_reaction("🧀")
|
||||
|
||||
|
||||
# ------------------------------------------------------------------------------------------------
|
||||
|
||||
# Log to file
|
||||
handler = logging.FileHandler(filename='discord.log', encoding='utf-8', mode='w')
|
||||
handler = logging.FileHandler(filename="discord.log", encoding="utf-8", mode="w")
|
||||
|
||||
# Intents specification is no longer optional
|
||||
intents = discord.Intents.default()
|
||||
intents.members = True # Allow to react to member join/leave etc
|
||||
intents.message_content = True # Allow to read message content from arbitrary messages
|
||||
intents.members = True # Allow to react to member join/leave etc
|
||||
intents.message_content = True # Allow to read message content from arbitrary messages
|
||||
|
||||
# Setup our client
|
||||
client = HeidiClient(intents=intents)
|
||||
@ -146,9 +145,14 @@ client = HeidiClient(intents=intents)
|
||||
# Events -----------------------------------------------------------------------------------------
|
||||
# NOTE: I defined the events outside of the Client class, don't know if I like it or not...
|
||||
|
||||
|
||||
@client.event
|
||||
async def on_ready():
|
||||
print(f"{client.user} (id: {client.user.id}) has connected to Discord!")
|
||||
if client.user != None:
|
||||
print(f"{client.user} (id: {client.user.id}) has connected to Discord!")
|
||||
else:
|
||||
print(f"client.user is None!")
|
||||
|
||||
|
||||
@client.event
|
||||
async def on_message(message):
|
||||
@ -164,14 +168,22 @@ async def on_message(message):
|
||||
await action(message)
|
||||
break
|
||||
|
||||
|
||||
# Commands ---------------------------------------------------------------------------------------
|
||||
|
||||
@client.tree.command(name = "giblinkbruder", description = "Heidi hilft mit dem Link zu deiner Lieblingsshow im Qualitätsfernsehen.")
|
||||
|
||||
@client.tree.command(
|
||||
name="giblinkbruder",
|
||||
description="Heidi hilft mit dem Link zu deiner Lieblingsshow im Qualitätsfernsehen.",
|
||||
)
|
||||
async def show_link(interaction: discord.Interaction):
|
||||
link_pro7 = "https://www.prosieben.de/tv/germanys-next-topmodel/livestream"
|
||||
link_joyn = "https://www.joyn.de/serien/germanys-next-topmodel"
|
||||
|
||||
await interaction.response.send_message(f"ProSieben: {link_pro7}\nJoyn: {link_joyn}")
|
||||
await interaction.response.send_message(
|
||||
f"ProSieben: {link_pro7}\nJoyn: {link_joyn}"
|
||||
)
|
||||
|
||||
|
||||
@client.tree.command(name="heidi", description="Heidi!")
|
||||
async def heidi_exclaim(interaction: discord.Interaction):
|
||||
@ -182,13 +194,14 @@ async def heidi_exclaim(interaction: discord.Interaction):
|
||||
"Dann zieh dich mal aus!",
|
||||
"Warum denn so schüchtern?",
|
||||
"Im TV ist das legal!",
|
||||
"Das Stroh ist nur fürs Shooting!"
|
||||
"Das Stroh ist nur fürs Shooting!",
|
||||
]
|
||||
await interaction.response.send_message(random.choice(messages))
|
||||
|
||||
|
||||
@client.tree.command(name="miesmuschel", description="Was denkt Heidi?")
|
||||
@app_commands.rename(question = "frage")
|
||||
@app_commands.describe(question = "Heidi wird es beantworten!")
|
||||
@app_commands.rename(question="frage")
|
||||
@app_commands.describe(question="Heidi wird es beantworten!")
|
||||
async def magic_shell(interaction: discord.Interaction, question: str):
|
||||
choices = [
|
||||
"Ja!",
|
||||
@ -202,17 +215,23 @@ async def magic_shell(interaction: discord.Interaction, question: str):
|
||||
]
|
||||
question = question.strip()
|
||||
question_mark = "" if question[-1] == "?" else "?"
|
||||
await interaction.response.send_message(f"{question}{question_mark}\nHeidi sagt: {random.choice(choices)}")
|
||||
await interaction.response.send_message(
|
||||
f"{question}{question_mark}\nHeidi sagt: {random.choice(choices)}"
|
||||
)
|
||||
|
||||
|
||||
# TODO: Allow , separated varargs, need to parse manually as slash commands don't support varargs
|
||||
@client.tree.command(name="wähle", description="Heidi trifft die Wahl!")
|
||||
@app_commands.rename(option_a = "entweder")
|
||||
@app_commands.describe(option_a = "Ist es vielleicht dies?")
|
||||
@app_commands.rename(option_b = "oder")
|
||||
@app_commands.describe(option_b = "Oder doch eher das?")
|
||||
@app_commands.rename(option_a="entweder")
|
||||
@app_commands.describe(option_a="Ist es vielleicht dies?")
|
||||
@app_commands.rename(option_b="oder")
|
||||
@app_commands.describe(option_b="Oder doch eher das?")
|
||||
async def choose(interaction: discord.Interaction, option_a: str, option_b: str):
|
||||
options = [option_a.strip(), option_b.strip()]
|
||||
await interaction.response.send_message(f"{options[0]} oder {options[1]}?\nHeidi sagt: {random.choice(options)}")
|
||||
await interaction.response.send_message(
|
||||
f"{options[0]} oder {options[1]}?\nHeidi sagt: {random.choice(options)}"
|
||||
)
|
||||
|
||||
|
||||
# async def quote_model_autocomplete(interaction: discord.Interaction, current: str) -> list[Choice[str]]:
|
||||
# models = client.textgen_models.keys()
|
||||
@ -239,22 +258,33 @@ async def choose(interaction: discord.Interaction, option_a: str, option_b: str)
|
||||
# joined_quote = " ".join(generated_quote)
|
||||
# await interaction.response.send_message(f"Heidi sagt: \"{joined_quote}\"")
|
||||
|
||||
SOUNDDIR: str = "/sounds/" if DOCKER else "./voicelines/"
|
||||
SOUNDDIR: str = "/sounds/" if DOCKER else "./heidi-sounds/"
|
||||
|
||||
|
||||
# Example: https://discordpy.readthedocs.io/en/latest/interactions/api.html?highlight=autocomplete#discord.app_commands.autocomplete
|
||||
async def board_autocomplete(interaction: discord.Interaction, current: str) -> list[Choice[str]]:
|
||||
boards = os.listdir(SOUNDDIR)
|
||||
return [Choice(name=board, value=board) for board in boards]
|
||||
|
||||
async def sound_autocomplete(interaction: discord.Interaction, current: str) -> list[Choice[str]]:
|
||||
board = interaction.namespace.board # TODO: Can't work?
|
||||
sounds = map(lambda x: x.split(".")[0], os.listdir(SOUNDDIR + board + "/"))
|
||||
return [Choice(name=sound, value=sound) for sound in sounds]
|
||||
async def board_autocomplete(
|
||||
interaction: discord.Interaction, current: str
|
||||
) -> list[Choice[str]]:
|
||||
boards: List[str] = os.listdir(SOUNDDIR)
|
||||
|
||||
@client.tree.command(name = "sag", description = "Heidi drückt den Knopf auf dem Soundboard.")
|
||||
@app_commands.describe(sound = "Was soll Heidi sagen?")
|
||||
@app_commands.autocomplete(board = board_autocomplete)
|
||||
@app_commands.autocomplete(sound = sound_autocomplete)
|
||||
return [Choice(name=board, value=board) for board in boards if board.startswith(current)]
|
||||
|
||||
|
||||
async def sound_autocomplete(
|
||||
interaction: discord.Interaction, current: str
|
||||
) -> list[Choice[str]]:
|
||||
board: str = interaction.namespace.board
|
||||
sounds: List[str] = list(map(lambda x: x.split(".")[0], os.listdir(SOUNDDIR + board + "/")))
|
||||
|
||||
return [Choice(name=sound, value=sound) for sound in sounds if sound.startswith(current)]
|
||||
|
||||
|
||||
@client.tree.command(
|
||||
name="sag", description="Heidi drückt den Knopf auf dem Soundboard."
|
||||
)
|
||||
@app_commands.describe(sound="Was soll Heidi sagen?")
|
||||
@app_commands.autocomplete(board=board_autocomplete)
|
||||
@app_commands.autocomplete(sound=sound_autocomplete)
|
||||
async def say_voiceline(interaction: discord.Interaction, board: str, sound: str):
|
||||
# Only Members can access voice channels
|
||||
if not isinstance(interaction.user, discord.Member):
|
||||
@ -265,7 +295,11 @@ async def say_voiceline(interaction: discord.Interaction, board: str, sound: str
|
||||
member: discord.Member = interaction.user
|
||||
|
||||
# Member needs to be in voice channel to hear audio (Heidi needs to know the channel to join)
|
||||
if (not member.voice) or (not member.voice.channel) or (not isinstance(member.voice.channel, discord.VoiceChannel)):
|
||||
if (
|
||||
(not member.voice)
|
||||
or (not member.voice.channel)
|
||||
or (not isinstance(member.voice.channel, discord.VoiceChannel))
|
||||
):
|
||||
print("User not in (valid) voice channel!")
|
||||
await interaction.response.send_message("Heidi sagt: Komm in den Channel!")
|
||||
return
|
||||
@ -276,12 +310,16 @@ async def say_voiceline(interaction: discord.Interaction, board: str, sound: str
|
||||
open(SOUNDDIR + board + "/" + sound + ".mkv")
|
||||
except IOError:
|
||||
print("Error: Invalid soundfile!")
|
||||
await interaction.response.send_message(f"Heidi sagt: \"{board}/{sound}\" kanninich finden bruder")
|
||||
await interaction.response.send_message(
|
||||
f'Heidi sagt: "{board}/{sound}" kanninich finden bruder'
|
||||
)
|
||||
return
|
||||
|
||||
await interaction.response.send_message(f"Heidi sagt: \"{board}/{sound}\"")
|
||||
await interaction.response.send_message(f'Heidi sagt: "{board}/{sound}"')
|
||||
|
||||
audio_source = discord.FFmpegPCMAudio(SOUNDDIR + board + "/" + sound + ".mkv") # only works from docker
|
||||
audio_source = discord.FFmpegPCMAudio(
|
||||
SOUNDDIR + board + "/" + sound + ".mkv"
|
||||
) # only works from docker
|
||||
voice_client = await voice_channel.connect()
|
||||
voice_client.play(audio_source)
|
||||
|
||||
@ -290,12 +328,16 @@ async def say_voiceline(interaction: discord.Interaction, board: str, sound: str
|
||||
|
||||
await voice_client.disconnect()
|
||||
|
||||
|
||||
# Contextmenu ------------------------------------------------------------------------------------
|
||||
|
||||
|
||||
# TODO: More insults
|
||||
# Callable on members
|
||||
@client.tree.context_menu(name="beleidigen")
|
||||
async def insult(interaction: discord.Interaction, member: discord.Member): # with message: discord.Message this can be called on a message
|
||||
async def insult(
|
||||
interaction: discord.Interaction, member: discord.Member
|
||||
): # with message: discord.Message this can be called on a message
|
||||
if not member.dm_channel:
|
||||
await member.create_dm()
|
||||
|
||||
@ -317,11 +359,14 @@ async def insult(interaction: discord.Interaction, member: discord.Member): # wi
|
||||
"Opfer!",
|
||||
"Du miese Raupe!",
|
||||
"Geh Steckdosen befruchten!",
|
||||
"Richtiger Gesichtsgünther ey!"
|
||||
"Richtiger Gesichtsgünther ey!",
|
||||
]
|
||||
|
||||
await member.dm_channel.send(random.choice(insults))
|
||||
await interaction.response.send_message("Anzeige ist raus!") # with ephemeral = True only the caller can see the answer
|
||||
await interaction.response.send_message(
|
||||
"Anzeige ist raus!"
|
||||
) # with ephemeral = True only the caller can see the answer
|
||||
|
||||
|
||||
# ------------------------------------------------------------------------------------------------
|
||||
|
||||
|
||||
Reference in New Issue
Block a user