Post process output

app.py

@@ -130,11 +130,11 @@ def chat():
     return {"message": response}
 
 def post_process(output):
-    
+    output = str(output)
     lines = output.split("\n")
     for line in lines:
         for word in word_remove_sentence:
-            if word in line.lower():
+            if string_found(word, line.lower()):
                 lines.remove(line)
                 break
     
@@ -142,8 +142,12 @@ def post_process(output):
     for item in map_words.keys():
         output = re.sub(item, map_words[item], output, re.I)
         
-    return output
-    
+    return output.strip()
+
+def string_found(string1, string2):
+    if re.search(r"\b" + re.escape(string1) + r"\b", string2):
+        return True
+    return False
 
 
 subprocess.Popen(["autossh", "-M", "0", "-o", "StrictHostKeyChecking=no",