Update routers/profanity.py

routers/profanity.py

@@ -1,53 +1,46 @@
 # routers/profanity.py
 
-from fastapi import APIRouter, Query
+from fastapi import APIRouter, Query, HTTPException
 from better_profanity import profanity
 
-# Cria um roteador para a API de palavrões
+# Create a router for the profanity API
 router = APIRouter()
 
-# Define a rota para verificar o texto
 @router.get("/profanity/check/")
 def check_profanity(text: str = Query(..., description="Text to be checked for profanity")):
     """
-    Checks if the text contains profanity and returns details in English.
+    Check if the text contains profanity and return the percentage of offensive words.
     """
-    # Load default profanity words
-    profanity.load_censor_words()
-
-    # Detects profanity
-    contains_profanity = profanity.contains_profanity(text)
-    
-    # Censor the text
-    censored_text = profanity.censor(text)
-    
-    # Split text into words
-    words = text.split()
-    total_word_count = len(words)
-    
-    # Identify offensive words
-    offensive_words = [
-        word for word in words
-        if profanity.contains_profanity(word)
-    ]
-    offensive_word_count = len(offensive_words)
-
-    # Calculate offensive percentage
-    offensive_percentage = (offensive_word_count / total_word_count * 100) if total_word_count > 0 else 0
-
-    # Determine severity level
-    if offensive_percentage == 0:
-        severity = "mild"
-    elif offensive_percentage <= 30:
-        severity = "moderate"
-    else:
-        severity = "severe"
-
-    return {
-        "contains_profanity": contains_profanity,
-        "offensive_words": offensive_words,
-        "offensive_word_count": offensive_word_count,
-        "total_word_count": total_word_count,
-        "offensive_percentage": round(offensive_percentage, 2),
-        "severity": severity
-    }
+    try:
+        # Validate input
+        if not text.strip():
+            raise HTTPException(status_code=400, detail="The text must not be empty.")
+
+        # Load default dictionary of offensive words
+        profanity.load_censor_words()
+
+        # Split the text into words
+        words = text.split()
+        total_word_count = len(words)
+
+        # Validate word count
+        if total_word_count == 0:
+            raise HTTPException(status_code=400, detail="The text must contain at least one word.")
+
+        # Identify offensive words
+        offensive_words = [
+            word for word in words if profanity.contains_profanity(word)
+        ]
+        offensive_word_count = len(offensive_words)
+
+        # Calculate the percentage of offensive words
+        offensive_percentage = (offensive_word_count / total_word_count * 100)
+
+        # Return the percentage rounded to 2 decimal places
+        return {
+            "offensive_percentage": round(offensive_percentage, 2)
+        }
+
+    except Exception as e:
+        # Catch unexpected errors and return a generic error message
+        raise HTTPException(status_code=500, detail="An unexpected error occurred.") from e