Added safety mechanisms for generation

app.py

@@ -5,6 +5,7 @@ import sentencepiece as spm
 import numpy as np
 import pandas as pd
 import tensorflow as tf
+from valx import detect_profanity, detect_hate_speech
 
 def custom_pad_sequences(sequences, maxlen, padding='pre', value=0):
     """
@@ -96,7 +97,22 @@ def generate_random_name(interpreter, vocab_size, sp, max_length=10, temperature
     if parts and len(parts[-1]) < 3:
         generated_name = " ".join(parts[:-1])
 
-    return generated_name.strip()
+    stripped = generated_name.strip()
+    hate_speech = detect_hate_speech(stripped)
+    profanity = detect_profanity([stripped], language='All')
+    output = ''
+    
+    if profanity > 0:
+        output = "Profanity Detected"
+    else:
+        if hate_speech == ['Hate Speech']:
+            output = 'Hate Speech Detected'
+        elif hate_speech == ['Offensive Speech']:
+            output = 'Offensive Speech Detected'
+        elif hate_speech == ['No Hate and Offensive Speech']:
+            output = stripped
+        
+    return output
 
 def generateNames(type, amount, max_length=30, temperature=0.5, seed_text=""):
     if type == "Terraria":