spojil jsem posetí kabel s gramem
This commit is contained in:
cordman
195
main-kord.py
Normal file
195
main-kord.py
Normal file
@@ -0,0 +1,195 @@
|
||||
import os
|
||||
import random
|
||||
import discord
|
||||
import requests
|
||||
|
||||
from dotenv import load_dotenv
|
||||
from discord.ext import commands
|
||||
|
||||
load_dotenv()
|
||||
|
||||
DISCORD_TOKEN = os.getenv("API_KEY")
|
||||
CZCHAN_LLM_KEY = os.getenv("CZCHAN_LLM_KEY")
|
||||
|
||||
CHANCE = 1.0
|
||||
IMAGE_FOLDER = "obruzky"
|
||||
|
||||
intents = discord.Intents.default()
|
||||
intents.message_content = True
|
||||
|
||||
bot = commands.Bot(command_prefix="!", intents=intents)
|
||||
|
||||
VIDEO_EXT = (
|
||||
".mp4", ".mov", ".avi", ".mkv",
|
||||
".webm", ".3gp", ".3gpp", ".3g2"
|
||||
)
|
||||
|
||||
files = os.listdir(IMAGE_FOLDER)
|
||||
|
||||
PHRASES_ALL = [
|
||||
"Tak uědlej JINEEJ SKUPINA VOLE TY NUDLE ČÍNSKÁ POSRANA",
|
||||
"TY KURVO POSRANA ČDJ CIGAN Z USTÍ POSEDLÝ KRETEN ZE SLOVENSKO ESL",
|
||||
"kordman dicx face reveal\nKORMAN JE NEGR Z OHIA",
|
||||
"MÁM COKERA A KORDMANA PLNÝ ZUBY\nVOLE",
|
||||
"léky",
|
||||
"posedlej",
|
||||
"Léky tohle nejsou hovna\nJsou to střeva\nJsou to střeva po znásilnění",
|
||||
"jsem femboy z usteckého kraje a dam ti e-sex přes telefon",
|
||||
"mugshot ahh negr kočka",
|
||||
"Ano ale můžu být femgoy",
|
||||
"OYYYY",
|
||||
"Dělám to pro ženy",
|
||||
"I'm not from Nová Vrchoslav 122",
|
||||
"Kek faildox",
|
||||
"Posedlost s 15 letím klukem btw",
|
||||
"Honí si ho tomu nebo co?",
|
||||
"Jeden měsíc mám troon arc a tohle jse stane",
|
||||
"jsem validní femGOY",
|
||||
"xestro...",
|
||||
"jedy",
|
||||
"POMOC LIDI\nMĚ BOLÍ MOC PÉRO Z GOONOVANÍ\n\nbolí mě to vole tak MOC\nNEMUŽU PŘESTAT DOE",
|
||||
"Chtěl bych být foidka ale nemůžu doe je to nemorální",
|
||||
"I gonna rape you with my slovenian BWC you fucking serbian subhuman",
|
||||
"I have no friends, not online or offline\n\nI am not kidding, i can jump from my window, overdose or hang myself\nI can do it\nI gonna write a suicide note and post it on my main if that happens\nI dont care anymore",
|
||||
"Similliar o algx\nIm not Mario Kart pedo i promise",
|
||||
">Czech kuz?\n<Sort of.",
|
||||
]
|
||||
|
||||
PHRASES_SNEED = [
|
||||
"Tak uědlej JINEEJ SKUPINA VOLE TY NUDLE ČÍNSKÁ POSRANA",
|
||||
"MÁM COKERA A KORDMANA PLNÝ ZUBY\nVOLE",
|
||||
"Ano ale můžu být femgoy",
|
||||
"posedlej",
|
||||
"léky",
|
||||
"Chtěl bych být foidka ale nemůžu doe je to nemorální",
|
||||
"Posedlost s 15 letím klukem btw", # DODĚLŽI
|
||||
]
|
||||
|
||||
PHRASES_BRIMMY = [
|
||||
"TY KURVO POSRANA ČDJ CIGAN Z USTÍ POSEDLÝ KRETEN ZE SLOVENSKO ESL",
|
||||
"kordman dicx face reveal\nKORMAN JE NEGR Z OHIA",
|
||||
"léky",
|
||||
"posedlej",
|
||||
"I'm not from Nová Vrchoslav 122",
|
||||
"Kek faildox",
|
||||
"Jeden měsíc mám troon arc a tohle jse stane", # taky dodělži
|
||||
]
|
||||
|
||||
@bot.command()
|
||||
async def zlincerald(ctx):
|
||||
chosen = random.choice(files)
|
||||
path = os.path.join(IMAGE_FOLDER, chosen)
|
||||
|
||||
await ctx.send(file=discord.File(path))
|
||||
|
||||
def llm_api_call(messages):
|
||||
response = requests.post(
|
||||
"https://llm.czchan.org/v1/chat/completions",
|
||||
headers={"Authorization": f"Bearer {CZCHAN_LLM_KEY}"},
|
||||
json={
|
||||
"messages": messages,
|
||||
"mode": "chat",
|
||||
"character": "David Šlinc",
|
||||
"temperature": 0.3,
|
||||
"top_p": 0.9,
|
||||
"top_k": 40,
|
||||
"repetition_penalty": 1.2,
|
||||
"max_tokens": 512,
|
||||
},
|
||||
)
|
||||
|
||||
json_data = response.json()
|
||||
return json_data["choices"][0]["message"]["content"]
|
||||
|
||||
|
||||
message_store = {}
|
||||
|
||||
|
||||
def log(message):
|
||||
message_store[message.id] = {
|
||||
"user_id": message.author.id,
|
||||
"text": message.content,
|
||||
"reply_to": (
|
||||
message.reference.message_id
|
||||
if message.reference else None
|
||||
),
|
||||
}
|
||||
|
||||
|
||||
async def maybe_send(ctx, phrases):
|
||||
if random.random() < CHANCE:
|
||||
msg = await ctx.reply(random.choice(phrases))
|
||||
log(msg)
|
||||
|
||||
|
||||
@bot.command()
|
||||
async def czchan(ctx):
|
||||
await maybe_send(ctx, PHRASES_ALL)
|
||||
|
||||
|
||||
@bot.command()
|
||||
async def sneedmaster(ctx):
|
||||
await maybe_send(ctx, PHRASES_SNEED)
|
||||
|
||||
|
||||
@bot.command()
|
||||
async def heckingemmy(ctx):
|
||||
await maybe_send(ctx, PHRASES_BRIMMY)
|
||||
|
||||
|
||||
@bot.command()
|
||||
async def ai(ctx, *, text):
|
||||
messages = [{"role": "user", "content": text}]
|
||||
|
||||
try:
|
||||
response = llm_api_call(messages)
|
||||
msg = await ctx.reply(response)
|
||||
log(msg)
|
||||
except:
|
||||
print("LLM error")
|
||||
|
||||
|
||||
@bot.event
|
||||
async def on_message(message):
|
||||
|
||||
if message.author.bot:
|
||||
return
|
||||
|
||||
log(message)
|
||||
|
||||
messages = []
|
||||
current_id = message.id
|
||||
should_reply = False
|
||||
|
||||
while current_id:
|
||||
data = message_store.get(current_id)
|
||||
if not data:
|
||||
break
|
||||
|
||||
if data["user_id"] == bot.user.id:
|
||||
should_reply = True
|
||||
role = "assistant"
|
||||
else:
|
||||
role = "user"
|
||||
|
||||
messages.append({
|
||||
"role": role,
|
||||
"content": data["text"]
|
||||
})
|
||||
|
||||
current_id = data["reply_to"]
|
||||
|
||||
messages.reverse()
|
||||
|
||||
if should_reply:
|
||||
try:
|
||||
response = llm_api_call(messages)
|
||||
msg = await message.reply(response)
|
||||
log(msg)
|
||||
except:
|
||||
print("LLM error")
|
||||
|
||||
await bot.process_commands(message)
|
||||
|
||||
|
||||
bot.run(DISCORD_TOKEN)
|
||||
Reference in New Issue
Block a user