Update app.py

app.py

@@ -105,20 +105,26 @@ def generate_and_exec_cypher(input_query):
     return run_cypher_on_neo4j(gen_query, gen_key)
 
 def chatbot(input, history=[]):
-    output = str(generate_and_exec_cypher(input))
-    history.append((input, output))
-    return history, history
-
-gr.Interface(fn = chatbot,
-             title = "Mahabharata Chatbot",
-             description = "powered by Neo4j",
-             theme = "soft",
-             chatbot = gr.Chatbot(height=500),
-             undo_btn = None,
-             clear_btn = "\U0001F5D1 Clear chat",
-             examples = ["Who killed Ghatotakach?",
-                         "Who are the parents of Karna?",
-                         "Who are the kids of Kunti?",
-                         "Who are the siblings of Karna?"],
-             inputs = ["text",'state'],
-             outputs = ["chatbot",'state']).launch(debug = True, share=True)
+    
+    try:
+        output = str(generate_and_exec_cypher(input))
+        history.append((input, output))
+        return history, history
+    except:
+        # a bit of protection against exposed error messages
+        # we could log these situations in the backend to revisit later in development
+        return "I'm sorry, there was an error retrieving the information you requested."
+
+gr.ChatInterface(fn = chatbot,
+                 title = "Mahabharata Chatbot",
+                 description = "powered by Neo4j",
+                 theme = "soft",
+                 chatbot = gr.Chatbot(height=500),
+                 undo_btn = None,
+                 clear_btn = "\U0001F5D1 Clear chat",
+                 examples = ["Who killed Ghatotakach?",
+                             "Who are the parents of Karna?",
+                             "Who are the kids of Kunti?",
+                             "Who are the siblings of Karna?"],
+                 inputs = ["text",'state'],
+                 outputs = ["chatbot",'state']).launch(debug = True, share=True)