aho/welcomecenterbot
2
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

nosegm

Browse Source
This commit is contained in:
Untone
2024-09-29 09:55:58 +03:00
parent 22ed5f6335
commit 338f8584db
3 changed files with 4 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
handlers/messages_routing.py
View File

@@ -114,19 +114,9 @@ async def messages_routing(msg, state):
toxic_score = detector(normalized_text)
toxic_perc = math.floor(toxic_score * 100)
logger.info(f"\text: {normalized_text}\ntoxic: {toxic_perc}%")
nospaces_text = text.replace(" ", "")
if nospaces_text != text:
nospaces_normalized_text = normalize(nospaces_text)
nospaces_text_score = detector(nospaces_normalized_text)
nospaces_text_perc = math.floor(nospaces_text_score * 100)
if check_stopwords(nospaces_normalized_text):
logger.info('stopword detected with no spaces, toxicity +40%')
nospaces_text_perc += 40
logger.info(f"\nospaces_text: {nospaces_normalized_text}\nnospaces_toxic: {nospaces_text_perc}%")
if nospaces_text_score > toxic_score or nospaces_text_perc > 95:
text_perc = nospaces_text_perc
if check_stopwords(normalized_text):
logger.info('stopword detected with no spaces, toxicity +40%')
toxic_perc += 40
await redis.set(f"toxic:{cid}", mid)
await redis.set(f"toxic:{cid}:{uid}:{mid}", toxic_perc, ex=60 * 60 * 24 * 3)