Mask profanity based on characters

app.py

@@ -122,7 +122,7 @@ def predict(text):
     try:
         output = query(preprocess(text))[0]
     except Exception:
-        print(output)
+        return "Loading", "The model is loading please try again later."
 
     output = [tuple(i.values()) for i in output]
     output = dict((x, y) for x, y in output)
@@ -131,9 +131,12 @@ def predict(text):
 
     if predicted_label == 'Abusive':
         output_text = text
-        for i in profanities:
-            compiled = re.compile(re.escape(i), re.IGNORECASE)
-            output_text = compiled.sub('****', output_text)
+        for profanity in profanities:
+            compiled = re.compile(re.escape(profanity), re.IGNORECASE)
+            mask = ""
+            for i in profanity:
+                mask += "*" if i != " " else " "
+            output_text = compiled.sub(mask, output_text)
         return output, output_text
     else:
         return output, text