Add application file

main.py

@@ -70,14 +70,18 @@ if __name__ == "__main__":
 THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
 '''
 import json
+import os
+import logging
 from werkzeug.exceptions import HTTPException, UnprocessableEntity, InternalServerError
 from flask import Flask
 from flask_swagger_ui import get_swaggerui_blueprint
 from flask_cors import CORS
 from waitress import serve
-import os
 from dotenv import load_dotenv
-from router.router import app  # Importing the original blueprint
+import spacy
+from spacy.cli import download
+
+# Load environment variables
 load_dotenv()
 
 # Flask app setup
@@ -89,14 +93,32 @@ API_URL = '/static/metadata.json'
 swaggerui_blueprint = get_swaggerui_blueprint(SWAGGER_URL, API_URL, config={'app_name': "Infosys Responsible AI - Moderation"})
 app1.register_blueprint(swaggerui_blueprint)
 
-# Register the app blueprint for the '/rai/v1/moderations' route
-app1.register_blueprint(app)  # Registering the blueprint that contains the route
-
-# CORS and error handling setup
+# CORS setup
 CORS(app1)
 
+# Ensure Spacy model is installed and loaded
+def load_spacy_model():
+    try:
+        # Attempt to load the model
+        nlp = spacy.load('en_core_web_lg')
+    except OSError:
+        # If model is not found, download it
+        download('en_core_web_lg')
+        nlp = spacy.load('en_core_web_lg')
+    return nlp
+
+# Attempt to load Spacy model
+try:
+    nlp = load_spacy_model()
+    logging.info("Spacy model loaded successfully.")
+except Exception as e:
+    logging.error(f"Failed to load Spacy model: {e}")
+    raise
+
+# Error Handlers
 @app1.errorhandler(HTTPException)
 def handle_exception(e):
+    """Return JSON instead of HTML for HTTP errors."""
     response = e.get_response()
     response.data = json.dumps({
         "code": e.code,
@@ -107,6 +129,7 @@ def handle_exception(e):
 
 @app1.errorhandler(UnprocessableEntity)
 def validation_error_handler(exc):
+    """Return JSON instead of HTML for HTTP errors."""
     response = exc.get_response()
     exc_code_desc = exc.description.split("-")
     exc_code = int(exc_code_desc[0])
@@ -120,6 +143,7 @@ def validation_error_handler(exc):
 
 @app1.errorhandler(InternalServerError)
 def internal_server_error_handler(exc):
+    """Return JSON instead of HTML for HTTP errors."""
     response = exc.get_response()
     response.data = json.dumps({
         "code": 500,
@@ -128,9 +152,31 @@ def internal_server_error_handler(exc):
     response.content_type = "application/json"
     return response
 
-# Use Waitress for production server
+# Ensure that log directories exist and are writable
+log_dir = '/path/to/logs'  # Update this path as needed
+if not os.path.exists(log_dir):
+    os.makedirs(log_dir, exist_ok=True)
+
+# Configure basic logging
+logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
+
+# Debugging: log environment variables
+logging.info(f"PORT: {os.getenv('PORT')}")
+logging.info(f"THREADS: {os.getenv('THREADS')}")
+logging.info(f"CONNECTION_LIMIT: {os.getenv('CONNECTION_LIMIT')}")
+logging.info(f"CHANNEL_TIMEOUT: {os.getenv('CHANNEL_TIMEOUT')}")
+
+# Use Waitress for production server, or Flask for development
 if __name__ == "__main__":
-    serve(app1, host="0.0.0.0", port=int(os.getenv("PORT", 7860)))  # Ensure correct port is used
+    try:
+        # Start the Flask application
+        logging.info("Starting the application...")
+        serve(app1, host="0.0.0.0", port=int(os.getenv("PORT", 7860)), threads=int(os.getenv('THREADS', 6)),
+              connection_limit=int(os.getenv('CONNECTION_LIMIT', 500)), channel_timeout=int(os.getenv('CHANNEL_TIMEOUT', 120)))
+    except Exception as e:
+        logging.error(f"Error starting the application: {e}")
+        raise
+