104 lines
3.2 KiB
Python
104 lines
3.2 KiB
Python
import os
|
|
|
|
from discord.ext import commands
|
|
from dotenv import load_dotenv
|
|
|
|
import random
|
|
|
|
max_len = 6
|
|
|
|
with open('text-faust.txt', 'r', encoding='utf-8') as f:
|
|
training_text = f.read()
|
|
|
|
def train(text, max_len=5):
|
|
model = dict()
|
|
for i in range(1, max_len+1):
|
|
model[i] = dict()
|
|
for k in range(len(text) - i - 1):
|
|
if not text[k:k+i] in model[i]:
|
|
model[i][text[k:k+i]] = dict()
|
|
if text[k+i] in model[i][text[k:k+i]]:
|
|
model[i][text[k:k+i]][text[k+i]] += 1
|
|
else:
|
|
model[i][text[k:k+i]][text[k+i]] = 1
|
|
return model
|
|
|
|
def query(model, previous):
|
|
for l in range(len(previous), 0, -1):
|
|
if previous[-l:] not in model[l]:
|
|
continue
|
|
freqs = model[l][previous[-l:]]
|
|
return random.choices(list(freqs.keys()), list(freqs.values()))[0]
|
|
#return random.choices(list(freqs.keys()))[0]
|
|
|
|
model = train(training_text, max_len=max_len)
|
|
|
|
load_dotenv()
|
|
TOKEN = os.getenv('DISCORD_TOKEN')
|
|
|
|
bot = commands.Bot(command_prefix='-')
|
|
|
|
@bot.event
|
|
async def on_ready():
|
|
print('We have logged in as {0.user}'.format(bot))
|
|
|
|
@bot.command(name='isthis')
|
|
async def isthis_command(ctx, person, schmetterling, frage):
|
|
await ctx.send(f"https://api.memegen.link/images/pigeon/{person}/{schmetterling}/{frage}.png")
|
|
|
|
@bot.command(name='pictures')
|
|
async def pictures_command(ctx, a, auch_a, person):
|
|
await ctx.send(f"https://api.memegen.link/images/same/{a}/{auch_a}/{person}.png")
|
|
|
|
@bot.command(name='drake')
|
|
async def drake_command(ctx, no, yes):
|
|
await ctx.send(f"https://api.memegen.link/images/drake/{no}/{yes}.jpg")
|
|
|
|
@bot.command(name='obiwan')
|
|
async def obiwan_command(ctx, zitat):
|
|
await ctx.send(f"https://api.memegen.link/images/chosen/{zitat}.png")
|
|
|
|
@bot.command(name='picard')
|
|
async def picard_command(ctx, oben, unten):
|
|
await ctx.send(f"https://api.memegen.link/images/facepalm/{oben}/{unten}.png")
|
|
|
|
@bot.command(name='umdreh')
|
|
async def umdreh_command(ctx, yay, kerl, nay):
|
|
await ctx.send(f"https://api.memegen.link/images/db/{yay}/{kerl}/{nay}.png")
|
|
|
|
@bot.command(name='changemind')
|
|
async def changemind_command(ctx, these):
|
|
await ctx.send(f"https://api.memegen.link/images/cmm/{these}.png")
|
|
|
|
@bot.command(name='boromir')
|
|
async def boromir_command(ctx, notsimply, mordor):
|
|
await ctx.send(f"https://api.memegen.link/images/mordor/{notsimply}/{mordor}.png")
|
|
|
|
@bot.command(name='brains')
|
|
async def brains_command(ctx, eins, zwei, drei, pfier):
|
|
await ctx.send(f"https://api.memegen.link/images/gb/{eins}/{zwei}/{drei}/{pfier}.png")
|
|
|
|
@bot.command(name='summonbeleidiger')
|
|
async def test(ctx):
|
|
await ctx.send("+beleidiger")
|
|
|
|
@bot.command(name='faust')
|
|
async def test(ctx, current_text):
|
|
n_lines = 500
|
|
for _ in range(n_lines):
|
|
current_text += query(model, current_text[-6:])
|
|
|
|
await ctx.send(current_text)
|
|
|
|
@bot.command(name='faustmeme')
|
|
async def test(ctx, current_text):
|
|
current_text = current_text.replace('-', ' ')
|
|
n_lines = 100
|
|
for _ in range(n_lines):
|
|
current_text += query(model, current_text[-6:])
|
|
|
|
current_text = current_text.replace('-', '--').replace('\n', '-').replace(' ', '-')
|
|
await ctx.send(f"https://api.memegen.link/images/chosen/{current_text}.png""")
|
|
|
|
bot.run(TOKEN)
|