Update app.py

app.py

@@ -7,6 +7,7 @@ import os
 import datetime
 from datetime import datetime
 import openai
+import re
 
 #print('for update')
 
@@ -50,12 +51,21 @@ def getblog(text_inp1,text_inp2):
       if each!='':
         nsfwlowerlist.append(each.lower())
   
+  mainstring=text_inp1
   foundnsfw=0
   for each_word in nsfwlowerlist:
-    if each_word in text_inp1:
+      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
-  if foundnsfw==1:
-    error_msg="Unsafe content found. Please try again with different prompts."
+        print(each_word)
+        break
+  
+    if foundnsfw==1:
+    error_msg="Unsafe content found. Please try again with different keywords."
     print(error_msg)
     return(error_msg)
       
@@ -85,20 +95,21 @@ def getblog(text_inp1,text_inp2):
           blog+=temp
       print(blog)
 
-      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())
-      
+      mainstring=blog
       foundnsfw=0
       for each_word in nsfwlowerlist:
-        if each_word in blog.lower() or each_word in text_inp1:
+          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:
-        error_msg="Unsafe content found. Please try again with different prompts."
+        error_msg="Unsafe content found. Please try again with different keywords."
         print(error_msg)
         return(error_msg)
       else: