mirror of
https://github.com/RYGhub/royalnet.git
synced 2024-11-27 13:34:28 +00:00
43 lines
1.7 KiB
Python
43 lines
1.7 KiB
Python
|
from typing import *
|
||
|
|
from royalnet.commands import *
|
||
|
|
import markovify
|
||
|
|
import os
|
||
|
|
import re
|
||
|
|
|
||
|
|
|
||
|
|
class MarkovCommand(Command):
|
||
|
|
name: str = "markov"
|
||
|
|
|
||
|
|
description: str = "Genera una frase da una catena di Markov."
|
||
|
|
|
||
|
|
syntax: str = "[modello]"
|
||
|
|
|
||
|
|
_texts: Dict[str, markovify.NewlineText] = {}
|
||
|
|
|
||
|
|
def __init__(self, interface: CommandInterface):
|
||
|
|
super().__init__(interface)
|
||
|
|
if interface.name == "telegram":
|
||
|
|
files: List[str] = tuple(os.walk(self.config["Markov"]["models_directory"]))[0][2]
|
||
|
|
for file in files:
|
||
|
|
match = re.match(r"(\S+)\.json$", file)
|
||
|
|
if match is None:
|
||
|
|
continue
|
||
|
|
model_name = match.group(1)
|
||
|
|
with open(os.path.join(self.config["Markov"]["models_directory"], file)) as f:
|
||
|
|
self._texts[model_name] = markovify.NewlineText.from_json(f.read())
|
||
|
|
|
||
|
|
async def run(self, args: CommandArgs, data: CommandData) -> None:
|
||
|
|
if self.interface.name != "telegram":
|
||
|
|
raise UnsupportedError("[c]markov[/c] funziona solo su Telegram.")
|
||
|
|
model_name = args.optional(0, self.config["Markov"]["default_model"])
|
||
|
|
try:
|
||
|
|
sentence = self._texts[model_name].make_sentence()
|
||
|
|
except KeyError:
|
||
|
|
models = "\n- ".join([model_name for model_name in self._texts])
|
||
|
|
raise InvalidInputError("Il modello richiesto non esiste."
|
||
|
|
f"Modelli disponibili: {models}")
|
||
|
|
if sentence is None or sentence == "":
|
||
|
|
await data.reply(f"💭 Il bot ([c]{model_name}[/c])... non dice niente. Riprova!")
|
||
|
|
else:
|
||
|
|
await data.reply(f'💬 Il bot ([c]{model_name}[/c]) dice:\n{sentence}')
|