Update app.py

app.py

@@ -92,22 +92,67 @@ def chat():
         token = request_data.get("token") if request_data else None
         message = request_data.get("message") if request_data else None
 
+        # Error Handling for Missing Parameters
         if not token:
-            return jsonify({"error": "No token provided in the request."}), 400
+            return jsonify({"error": "Token Not Provided", "details": "Please provide a valid token in the request body."}), 400
         if not message:
-            return jsonify({"error": "No message provided in the request."}), 400
+            return jsonify({"error": "Message Not Provided", "details": "Please provide a message in the request body."}), 400
 
         # Check if the token is valid
         if not is_token_valid(token):
-            return jsonify({"error": "You must subscribe to continue."}), 403
+            # Enhanced Error Response for Token Validation
+            if token_expired(token):
+                return jsonify({"error": "Token Expired", "details": "Your token has expired. Please renew your subscription to continue."}), 403
+            elif token_not_found(token):
+                return jsonify({"error": "Token Not Found", "details": "The provided token does not exist. Please check your token or subscribe to the service."}), 403
+            else:
+                return jsonify({"error": "Invalid Token", "details": "The provided token is invalid. Please check your token or contact support."}), 403
 
         # Generate AI response using Brain Inc API
         ai_response = generate_ai_response(message)
-        return jsonify({"response": ai_response})
+        if ai_response == "Error generating response":
+            return jsonify({"error": "AI Response Generation Failed", "details": "Failed to generate a response from the AI service. Please try again later."}), 500
+        else:
+            return jsonify({"response": ai_response})
+
+    except requests.exceptions.RequestException as e:
+        # Catch API Request Exceptions
+        return jsonify({"error": "API Request Error", "details": str(e)}), 500
+    except Exception as e:
+        # Catch Any Unexpected Errors
+        return jsonify({"error": "An Unexpected Error Occurred", "details": str(e)}), 500
+
 
+# Helper functions for more specific token error handling
+def token_expired(token):
+    try:
+        response = requests.get(SEPTIMIUS_JSON_URL)
+        if response.status_code == 200:
+            json_data = response.json()
+            if json_data.get("success"):
+                for entry in json_data["data"]:
+                    if entry.get("token") == token:
+                        expired_date = entry.get("expired_date")
+                        if expired_date and datetime.fromisoformat(expired_date) <= datetime.now():
+                            return True
+    except Exception as e:
+        print(f"Error checking token expiration: {e}")
+    return False
+
+
+def token_not_found(token):
+    try:
+        response = requests.get(SEPTIMIUS_JSON_URL)
+        if response.status_code == 200:
+            json_data = response.json()
+            if json_data.get("success"):
+                for entry in json_data["data"]:
+                    if entry.get("token") == token:
+                        return False  # Token found
+            return True  # Token not found in the loop
     except Exception as e:
-        # Catch any unexpected errors
-        return jsonify({"error": "An error occurred", "details": str(e)}), 500
+        print(f"Error checking token existence: {e}")
+    return True  # Assume not found if an error occurs
 
 
 if __name__ == "__main__":