Update app.py

app.py

@@ -86,12 +86,14 @@ def plot_chart():
 db_chain = get_db_chain()
 json_chain = get_json_chain()
 
-
 def from_gpt(query: str, plot: bool):
 
-    chains = [db_chain, json_chain] if plot else [db_chain]
-    main_chain = SimpleSequentialChain(chains=chains, verbose=True)
-    ans = main_chain.run(query)
+    try:
+        chains = [db_chain, json_chain] if plot else [db_chain]
+        main_chain = SimpleSequentialChain(chains=chains, verbose=True)
+        ans = main_chain.run(query)
+    except Exception e:
+        return "Sorry, I didn't get that" 
 
     # Save data as json if plot
     st.session_state[JSON_DATA_LABEL] = ans if plot else {}