Update routers/profanity.py

routers/profanity.py

@@ -4,46 +4,45 @@ from better_profanity import profanity
 # Create a router for the profanity API
 router = APIRouter()
 
-# Define the route to check the text
 @router.get("/profanity/check/")
-def check_profanity(text: str = Query(..., description="Text to be analyzed")):
+def check_profanity(text: str = Query(..., description="Text to be checked")):
     """
-    Checks if the text contains profanity and returns details.
+    Check if the text contains profanity and return details.
     """
-    # Ensure text is not empty
-    if not text.strip():
-        raise HTTPException(status_code=400, detail="The input text cannot be empty.")
+    try:
+        # Load default dictionary
+        profanity.load_censor_words()
 
-    # Load default profanity words
-    profanity.load_censor_words()
+        # Detect profanity
+        contains_profanity = profanity.contains_profanity(text)
+        censored_text = profanity.censor(text)
 
-    # Check for profanity
-    contains_profanity = profanity.contains_profanity(text)
-    censored_text = profanity.censor(text)
+        # Extract offensive words
+        words = text.split()
+        offensive_words = [
+            word for word in words if profanity.contains_profanity(word)
+        ]
 
-    # Identify offensive words
-    offensive_words = [
-        word for word in text.split()
-        if profanity.contains_profanity(word)
-    ]
-    offensive_word_count = len(offensive_words)
-    total_words = len(text.split())
+        # Calculate percentage of offensive words
+        offensive_word_count = len(offensive_words)
+        total_word_count = len(words)
+        offensive_percentage = (
+            (offensive_word_count / total_word_count) * 100 if total_word_count > 0 else 0
+        )
 
-    # Calculate percentage of offensive words
-    offensive_percentage = (
-        (offensive_word_count / total_words) * 100 if total_words > 0 else 0
-    )
-
-    return {
-        "status_code": 200,
-        "message": "Profanity analysis successful.",
-        "data": {
+        # Return response
+        return {
             "original_text": text,
             "contains_profanity": contains_profanity,
             "censored_text": censored_text,
             "offensive_words": offensive_words,
             "offensive_word_count": offensive_word_count,
-            "total_word_count": total_words,
-            "offensive_percentage": round(offensive_percentage, 2),
-        },
-    }
+            "total_word_count": total_word_count,
+            "offensive_percentage": offensive_percentage,
+        }
+    except Exception as e:
+        # Handle errors gracefully
+        raise HTTPException(
+            status_code=500,
+            detail=f"An error occurred while processing the text: {str(e)}"
+        )