Update app.py

app.py

@@ -169,20 +169,20 @@ from transformers import MBartForConditionalGeneration, MBart50Tokenizer
 #tokenizer = MBart50Tokenizer.from_pretrained(model_name)
 
 import re, string, unicodedata
-# import wikipedia as wk #pip install wikipedia
+import wikipedia as wk #pip install wikipedia
 from collections import defaultdict
 
-# def wikipedia_data(input_text):
-#     reg_ex = re.search('from wikipedia (.*)', input_text)#tell me about 
-#     try:
-#         if reg_ex:
-#             topic = reg_ex.group(1)
-#             wiki = wk.summary(topic, sentences = 3)
-#             return wiki
-#         else:
-#             print("My apology, Can you please rephrase your query?")
-#     except Exception as e:
-#             print("I do not understand...Please rephrase")
+def wikipedia_data(input_text):
+    reg_ex = re.search('from wikipedia (.*)', input_text)#tell me about 
+    try:
+        if reg_ex:
+            topic = reg_ex.group(1)
+            wiki = wk.summary(topic, sentences = 3)
+            return wiki
+        else:
+            print("My apology, Can you please rephrase your query?")
+    except Exception as e:
+            print("I do not understand...Please rephrase")
 
 
 def get_response(input_text):
@@ -208,9 +208,9 @@ def get_response(input_text):
             if tag == intent["tag"]:
                 return random.choice(intent['responses'])#responses
     else:
-        #if "From Wikipedia" in sentence:
-        #if sentence:
-        robo_response = wikipedia_data(input_text)
+        if "From Wikipedia" in sentence:
+            if sentence:
+                robo_response = wikipedia_data(input_text)
         return robo_response