Update app.py

app.py

@@ -1,4 +1,6 @@
 import gradio as gr
+from datasets import load_dataset
+
 import random
 import time
 import requests
@@ -9,6 +11,10 @@ import traceback
 from base64 import b64decode,b64encode
 from io import BytesIO
 
+word_list_dataset = load_dataset("stabilityai/word-list", data_files="list.txt", use_auth_token=True)
+word_list = word_list_dataset["train"]['text']
+
+
     
 with gr.Blocks(theme="darkdefault") as demo:
 
@@ -21,6 +27,11 @@ with gr.Blocks(theme="darkdefault") as demo:
             import uuid
             import os
             print(_prompt,_token)
+            
+            for filter in word_list:
+                if re.search(rf"\b{filter}\b", _prompt):
+                    return '','unsafe','','',''        
+            
             r = requests.post(url='https://xcyborgart-app.herokuapp.com/create',data={"prompt":_prompt,"token":_token})
             all_data = r.json()
             print(all_data.keys())