diff --git a/public/script.js b/public/script.js
index f62df6f89..8b9a489ab 100644
--- a/public/script.js
+++ b/public/script.js
@@ -1583,8 +1583,9 @@ $(document).ready(function () {
                                 getMessage = data.results[0].text;
                             } else if (main_api == 'textgenerationwebui') {
                                 getMessage = data.data[0];
-                                if (getMessage == null) {
-                                    runGenerate("");
+                                if (getMessage == null || data.error) {
+                                    popup_type = 'default';
+                                    callPopup('<h3>Got empty response from Text generation web UI. Try restarting the API with recommended options.</h3>');
                                     return;
                                 }
                                 getMessage = getMessage.substring(finalPromt.length);