Update app.py

app.py

@@ -41,8 +41,25 @@ titleslist=prompt.split('SEPERATOR')[2].split(',')
 def getblog(text_inp1,text_inp2):
   print(text_inp1,text_inp2)
   print(datetime.today().strftime("%d-%m-%Y"))
+  
+  response_nsfw = requests.get('https://github.com/coffee-and-fun/google-profanity-words/raw/main/data/list.txt')
+  data_nsfw = response_nsfw.text
+  nsfwlist=data_nsfw.split('\n')
+  nsfwlowerlist=[]
+  for each in nsfwlist:
+      if each!='':
+        nsfwlowerlist.append(each.lower())
+  
+  foundnsfw=0
+  for each_word in nsfwlowerlist:
+    if each_word in text_inp1:
+        foundnsfw=1
+  if foundnsfw==1:
+    error_msg="Unsafe content found. Please try again with different prompts."
+    print(error_msg)
+    return(error_msg)
+      
   if text_inp2!='':
-      #openai.api_key = text_inp2
       template_tmp=template
       template_tmp=template_tmp.replace("{topic}",text_inp1).strip()
       llm = OpenAI(temperature=0,openai_api_key=text_inp2)
@@ -75,7 +92,6 @@ def getblog(text_inp1,text_inp2):
       for each in nsfwlist:
           if each!='':
             nsfwlowerlist.append(each.lower())
-      nsfwlowerlist.extend(['bra','gay','lesbian',])
       
       foundnsfw=0
       for each_word in nsfwlowerlist: