Use gr.Error

app.py

@@ -1,8 +1,8 @@
 import gradio as gr
 import lemminflect
 import spacy
-from transformers import pipeline
 import wikipedia
+from transformers import pipeline
 
 nlp = spacy.load("en_core_web_lg")
 sentiment_analyzer = pipeline(
@@ -23,30 +23,42 @@ def make_dystopian(term, text):
   doc = nlp(text)
   if is_positive(term):
     return "".join([make_past_tense(token) for token in doc])
-  return doc.text_with_ws
+  return doc.text
+
+def get_dystopian_summary(term):
+  if term == "":
+    return term
 
-def get_summary(term):
-  if not term:
-    return ""
   try:
-    results = wikipedia.search(term)
+    results = wikipedia.search(term, results=1)
   except wikipedia.exceptions.DisambiguationError as e:
-    return e.error
-  if len(results) > 0:
-    summary = wikipedia.summary(results[0], sentences=1, auto_suggest=False, redirect=True)
-    return make_dystopian(term, summary)
-  return "Could not find an article on the term provided."
+    raise gr.Error(e.error)
+
+  if len(results) == 0:
+    raise gr.Error(
+      f'Could not find an article on the term "{term}". '
+      'Try searching for a different topic.'
+    )
+
+  summary = wikipedia.summary(
+    results[0],
+    sentences=1,
+    auto_suggest=False,
+    redirect=True
+  )
+  return make_dystopian(term, summary)
 
-def launch_demo():
+def launch_demo(**kwargs):
   title = "Dystopedia"
   description = (
-    "Make any Wikipedia topic dystopian. Inspired by [this Tweet](https://twitter.com/lbcyber/status/1115015586243862528). "
+    "Make any Wikipedia topic dystopian. Inspired by "
+    "[this Tweet](https://twitter.com/lbcyber/status/1115015586243862528). "
     "Dystopedia uses [DistilBERT base uncased finetuned SST-2](https://huggingface.co/distilbert-base-uncased-finetuned-sst-2-english) "
     "for sentiment analysis and is subject to its limitations and biases."
   )
   examples = ["joy", "hope", "peace", "Earth", "water", "food"]
   gr.Interface(
-    fn=get_summary,
+    fn=get_dystopian_summary,
     inputs=gr.Textbox(label="term", placeholder="Enter a term...", max_lines=1),
     outputs=gr.Textbox(label="description"),
     title=title,
@@ -54,6 +66,6 @@ def launch_demo():
     examples=examples,
     cache_examples=True,
     allow_flagging="never",
-  ).launch()
+  ).launch(**kwargs)
 
-launch_demo()
+launch_demo(show_error=True)