Update app.py

app.py

@@ -154,11 +154,26 @@ def getadline(text_inp):
 
 def getadvertisement(topic):
     if topic!='':
-        input_keyword=topic
+        mainstring=topic
+        foundnsfw=0
+        for each_word in nsfwlowerlist:
+            raw_search_string = r"\b" + each_word + r"\b"
+            match_output = re.search(raw_search_string, mainstring)
+            no_match_was_found = ( match_output is None )
+            if no_match_was_found:
+              foundnsfw=0
+            else:
+              foundnsfw=1
+              print(each_word)
+              break
+        if foundnsfw==1:
+            topic="Unsafe content found. Please try again with different prompts."
+            print(topic)
+            input_keyword=topic
     else:
         input_keyword=getadline(random.choice('abcdefghijklmnopqrstuvwxyz'))
-        if 'Unsafe content found' in input_keyword:
-            input_keyword='Abstract art with a splash of colors'
+    if 'Unsafe content found' in input_keyword:
+        input_keyword='Abstract art with a splash of colors'
     print(input_keyword)
     print(datetime.today().strftime("%d-%m-%Y"))
     prompt_SD=input_keyword+','+prompt_image