diff --git a/other_ext/kamaz_ai.py b/other_ext/kamaz_ai.py index 25f1be4..9ed64e9 100644 --- a/other_ext/kamaz_ai.py +++ b/other_ext/kamaz_ai.py @@ -194,6 +194,19 @@ class KamazAI: class Extension: core = None + action_translate = { + 'ban': "забанить игрока", + 'inspect': "проверить профиль стима и решить что делать дальше", + 'kick': "кикнуть игрока", + 'ban30': "легкий бан на 30 минут", + 'ban120': "бан на пару часов", + 'noreason': "не вводить причину", + 'author_kick': "кикнуть автора репорта", + 'mute': "замьютить игрока", + 'unban': "разбанить игрока если тот в бане, ебанутое решение", + 'author_inspect': 'глянуть профиль автора репорта' + } + def __init__(self, core): self.core = core self.kamazai = None @@ -244,13 +257,14 @@ class Extension: suggestions = result.get("suggestions", []) s = [] for ss in suggestions: - s.append(f'{ss["action"]} c вероятностью {ss["confidence"]}') + #s.append(f'{ss["action"]} c вероятностью {ss["confidence"]}') + s.append(f'{round(ss["confidence"] * 100)}) {self.action_translate.get(ss["action"], ss["action"])}') - content = f'KamazAI решил что с этим репортом надо сделать: ' + ",".join(s) + "\nОцените решение камаза где :thumbsup: - норм или :thumbsdown: - не очень" + content = f'KamazAI решил что с этим репортом надо сделать: \n' + "\n".join(s) + "\nОцените решение камаза где :thumbsup: - норм или :thumbsdown: - не очень" response = await message.reply(content=content) try: - await response.add_reaction(':thumbsup:') - await response.add_reaction(':thumbsdown:') + await response.add_reaction('👍') + await response.add_reaction('👎') except: pass