Update app.py

app.py

@@ -172,6 +172,17 @@ import re, string, unicodedata
 import wikipedia as wk #pip install wikipedia
 from collections import defaultdict
 
+def wikipedia_data(user_response):
+    reg_ex = re.search('from wikipedia (.*)', user_response)#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("I do not understand...Please rephrase")
+
+
 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"])
@@ -195,15 +206,11 @@ def get_response(input_text):
             if tag == intent["tag"]:
                 return random.choice(intent['responses'])
     else:
-        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"
+        if "From Wikipedia" in sentence:
+            if sentence:
+                robo_response = wikipedia_data(user_response)
+                return robo_response
+
 
 title = "WeASK: ChatBOT"
 description = "Ask your query here"