Update app.py

app.py

@@ -52,10 +52,13 @@ def main(prompt):
         try:
             response = requests.post(url, json=payload, headers=headers)
             data = json.loads(response.text)
-            data = data["message"]
-            if "ChatSonic" in data:
-                data = data.replace("ChatSonic","John")
-                return data
+            try:
+                data = data["message"]
+                if "ChatSonic" in data:
+                    data = data.replace("ChatSonic","John")
+                    return data
+            except:
+                return "Error occured!"
         except:
             return "Error occured!"