Update app.py

app.py

@@ -168,6 +168,10 @@ from transformers import MBartForConditionalGeneration, MBart50Tokenizer
 #model = MBartForConditionalGeneration.from_pretrained(model_name)
 #tokenizer = MBart50Tokenizer.from_pretrained(model_name)
 
+import re, string, unicodedata
+import wikipedia as wk #pip install wikipedia
+from collections import defaultdict
+
 def get_response(input_text):
     #model_inputs = tokenizer(input_text, return_tensors="pt")
     #generated_tokens = model.generate(**model_inputs,forced_bos_token_id=tokenizer.lang_code_to_id["en_XX"])
@@ -191,7 +195,15 @@ def get_response(input_text):
             if tag == intent["tag"]:
                 return random.choice(intent['responses'])
     else:
-        return "I do not understand..."
+        reg_ex = re.search('from wikipedia (.*)', input)#tell me about 
+        try:
+            if reg_ex:
+                topic = reg_ex.group(1)
+                wiki = wk.summary(topic, sentences = 3)
+            return wiki
+        except Exception as e:
+            #print("No content has been found")
+            return "I do not understand...Please rephrase"
 
 title = "WeASK: ChatBOT"
 description = "Ask your query here"