no-process-score2

This commit is contained in:
Untone 2024-09-26 15:38:14 +03:00
parent 5658992a88
commit f51548dc07

View File

@ -33,34 +33,34 @@ async def messages_routing(msg, state):
elif bool(text): elif bool(text):
mid = msg.get("message_id") mid = msg.get("message_id")
if text.startswith('/score'): if text == '/score@welcomecenter_bot':
await telegram_api( await telegram_api(
"sendMessage", "sendMessage",
chat_id=cid, chat_id=cid,
reply_to=mid, reply_to=mid,
text=f"{latest_toxic.get(cid, 0)}%" text=f"{latest_toxic.get(cid, 0)}%"
) )
return else:
toxic_score = detector(text) toxic_score = detector(text)
toxic_perc = math.floor(toxic_score*100) toxic_perc = math.floor(toxic_score*100)
latest_toxic[cid] = toxic_perc latest_toxic[cid] = toxic_perc
logger.info(f'\ntext: {text}\ntoxic: {toxic_perc}%') logger.info(f'\ntext: {text}\ntoxic: {toxic_perc}%')
if toxic_score > 0.81: if toxic_score > 0.81:
if toxic_score > 0.90: if toxic_score > 0.90:
await redis.set(f"removed:{uid}:{cid}:{mid}", text) await redis.set(f"removed:{uid}:{cid}:{mid}", text)
await telegram_api( await telegram_api(
"deleteMessage", "deleteMessage",
chat_id=cid, chat_id=cid,
message_id=mid message_id=mid
) )
else: else:
await telegram_api( await telegram_api(
"setMessageReaction", "setMessageReaction",
chat_id=cid, chat_id=cid,
is_big=True, is_big=True,
message_id=mid, message_id=mid,
reaction=f'[{{"type":"emoji", "emoji":"🙉"}}]' reaction=f'[{{"type":"emoji", "emoji":"🙉"}}]'
) )
else: else:
pass pass