Update app.py

app.py

@@ -17,25 +17,29 @@ import matplotlib.pyplot as plt
 
 
 def opendomain(text,wikipedia_language="en"):
-  question_words = STOPWORDS.union(set(['likes','play','.',',','like',"don't",'?','use','choose','important','better','?']))
-  lower_text = text.lower()
-  lower_text = word_tokenize(lower_text)
-  new_text = [i for i in lower_text if i not in question_words]
-  new_txt = "".join(new_text)
-  if wikipedia_language:
-    wikipedia.set_lang(wikipedia_language)
+  try:
+    question_words = STOPWORDS.union(
+      set(['likes', 'play', '.', ',', 'like', "don't", '?', 'use', 'choose', 'important', 'better', '?']))
+    lower_text = text.lower()
+    lower_text = word_tokenize(lower_text)
+    new_text = [i for i in lower_text if i not in question_words]
+    new_txt = "".join(new_text)
+    if wikipedia_language:
+      wikipedia.set_lang(wikipedia_language)
 
-  et_page = wikipedia.page(new_txt.replace(" ", ""))
-  title = et_page.title
-  content = et_page.content
-  page_url = et_page.url
-  linked_pages = et_page.links
+    et_page = wikipedia.page(new_txt.replace(" ", ""))
+    title = et_page.title
+    content = et_page.content
+    page_url = et_page.url
+    linked_pages = et_page.links
 
-  text = content
-  wordcloud = WordCloud(font_path="HelveticaWorld-Regular.ttf").generate(text)
+    text = content
+    wordcloud = WordCloud(font_path="HelveticaWorld-Regular.ttf").generate(text)
 
-  plt.imshow(wordcloud, interpolation='bilinear')
-  plt.axis("off")
+    plt.imshow(wordcloud, interpolation='bilinear')
+    plt.axis("off")
+  except:
+    return "Please write correct wikipedia article name OR question"
 
 
   final_out = re.sub(r'\=.+\=', '', text)