added image ai-moderation

ai-moderation/app/moderation/profanity/detector.py

@@ -0,0 +1,51 @@
+import re
+
+from app.moderation.profanity.dictionary import ProfanityDictionary
+from app.moderation.profanity.lemmatizer import Lemmatizer
+
+
+
+class ProfanityDetector:
+
+
+  def __init__(
+      self,
+      dictionary: ProfanityDictionary,
+      lemmatizer: Lemmatizer
+  ):
+
+    self.dictionary = dictionary
+    self.lemmatizer = lemmatizer
+
+
+
+  def detect(
+      self,
+      text: str
+  ) -> list[str]:
+
+
+    words = re.findall(
+        r"[а-яА-ЯёЁ]+",
+        text.lower()
+    )
+
+
+    result = []
+
+
+    for word in words:
+
+      lemma = self.lemmatizer.normalize(word)
+
+
+      for bad_word in self.dictionary.words:
+
+        if lemma.startswith(bad_word):
+
+          result.append(word)
+
+          break
+
+
+    return result