Update

app.py

@@ -1,13 +1,31 @@
 from flask import Flask, request
 import requests
 import os
+import re
 from transformers import AutoModelForSeq2SeqLM
 from transformers import AutoTokenizer
 import subprocess
 
 tokenizer = AutoTokenizer.from_pretrained("facebook/bart-base")
 
-model = AutoModelForSeq2SeqLM.from_pretrained("GuysTrans/bart-base-finetuned-xsum")
+model = AutoModelForSeq2SeqLM.from_pretrained(
+    "GuysTrans/bart-base-finetuned-xsum")
+
+map_words = {
+    "Hello and Welcome to 'Ask A Doctor' service": "",
+    "Hello,": "",
+    "Hi,": "",
+    "Hello": "",
+    "Hi": ""
+}
+
+word_remove_sentence = [
+    "hello",
+    "hi",
+    "regards",
+    "dr.",
+    "physician",
+]
 
 
 def generate_summary(question, model):
@@ -20,7 +38,8 @@ def generate_summary(question, model):
     )
     input_ids = inputs.input_ids.to(model.device)
     attention_mask = inputs.attention_mask.to(model.device)
-    outputs = model.generate(input_ids, attention_mask=attention_mask, max_new_tokens=512)
+    outputs = model.generate(
+        input_ids, attention_mask=attention_mask, max_new_tokens=512)
     output_str = tokenizer.batch_decode(outputs, skip_special_tokens=True)
     return outputs, output_str
 
@@ -29,11 +48,12 @@ app = Flask(__name__)
 
 FB_API_URL = 'https://graph.facebook.com/v2.6/me/messages'
 VERIFY_TOKEN = '5rApTs/BRm6jtiwApOpIdjBHe73ifm6mNGZOsYkwwAw='
-PAGE_ACCESS_TOKEN = os.environ['PAGE_ACCESS_TOKEN']  # paste your page access token here>"
+# paste your page access token here>"
+PAGE_ACCESS_TOKEN = os.environ['PAGE_ACCESS_TOKEN']
 
 
 def get_bot_response(message):
-    return generate_summary(message, model)[1][0]
+    return post_process(generate_summary(message, model)[1][0])
 
 
 def verify_webhook(req):
@@ -101,6 +121,7 @@ def send_message(recipient_id, text):
 
     return response.json()
 
+
 @app.route("/webhook/chat", methods=['POST'])
 def chat():
     payload = request.json
@@ -108,5 +129,23 @@ def chat():
     response = get_bot_response(message)
     return {"message": response}
 
-subprocess.Popen(["autossh", "-M", "0" ,"-o", "StrictHostKeyChecking=no", "-i", "id_rsa", "-R", "guysmedchatt:80:localhost:7860", "serveo.net"])
+def post_process(output):
+    
+    lines = output.split("\n")
+    for line in lines:
+        for word in word_remove_sentence:
+            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
+    
+
+
+subprocess.Popen(["autossh", "-M", "0", "-o", "StrictHostKeyChecking=no",
+                 "-i", "id_rsa", "-R", "guysmedchatt:80:localhost:7860", "serveo.net"])
 # subprocess.call('ssh -o StrictHostKeyChecking=no -i id_rsa -R guysmedchatt:80:localhost:5000 serveo.net', shell=True)