Update post-process response

app.py

@@ -2,6 +2,7 @@ from flask import Flask, request
 import requests
 import os
 import re
+import textwrap
 from transformers import AutoModelForSeq2SeqLM
 from transformers import AutoTokenizer
 from langdetect import detect
@@ -24,7 +25,6 @@ map_words = {
 }
 
 word_remove_sentence = [
-    "welcome to",
     "hello",
     "hi",
     "regards",
@@ -139,24 +139,21 @@ def chat():
     return {"message": response}
 
 def post_process(output):
-    output = str(output)
+    
+    output = textwrap.fill(textwrap.dedent(output).strip(), width=120)
     lines = output.split("\n")
     for line in lines:
         for word in word_remove_sentence:
-            if string_found(word, line.lower()):
+            if word in line.lower():
                 lines.remove(line)
                 break
     
     output = "\n".join(lines)
     for item in map_words.keys():
         output = re.sub(item, map_words[item], output, re.I)
-        
-    return output.strip()
-
-def string_found(string1, string2):
-    if re.search(r"\b" + re.escape(string1) + r"\b", string2):
-        return True
-    return False
+    
+    return textwrap.fill(textwrap.dedent(output).strip(), width=120)
+    
 
 
 subprocess.Popen(["autossh", "-M", "0", "-tt", "-o", "StrictHostKeyChecking=no",