Add application file

main.py

@@ -10,36 +10,25 @@ THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLI
 import json
 from werkzeug.exceptions import HTTPException, UnprocessableEntity, InternalServerError
 from flask import Flask
-from router.router import app  # Ensure this is a Flask Blueprint
 from flask_swagger_ui import get_swaggerui_blueprint
 from flask_cors import CORS
-from dotenv import load_dotenv
+from waitress import serve
 import os
-
+from dotenv import load_dotenv
 load_dotenv()
 
-# Initialize request_id_var properly, if necessary
-# For example, you can set it as a simple string or initialize your logger here.
-# request_id_var.set("StartUp")   # Comment out or initialize properly
-
-SWAGGER_URL = '/rai/v1/moderations/docs'
-API_URL = '/static/metadata.json'
-
-swaggerui_blueprint = get_swaggerui_blueprint(
-    SWAGGER_URL, API_URL,
-    config={'app_name': "Infosys Responsible AI - Moderation"}
-)
-
-# Create Flask app instance
+# Flask app setup
 app1 = Flask(__name__)
 
-# Register blueprints
-app1.register_blueprint(app)
+# Swagger UI setup
+SWAGGER_URL = '/rai/v1/moderations/docs'  
+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)
 
+# CORS and error handling setup
 CORS(app1)
 
-# Error handlers
 @app1.errorhandler(HTTPException)
 def handle_exception(e):
     response = e.get_response()
@@ -73,10 +62,10 @@ def internal_server_error_handler(exc):
     response.content_type = "application/json"
     return response
 
-# Start Flask server
+# Use Waitress for production server
 if __name__ == "__main__":
-    # Make sure the logger is not trying to write to non-writable directories
-    app1.run(host='0.0.0.0', port=int(os.getenv("PORT", 7860)), debug=False)
+    serve(app1, host="0.0.0.0", port=7860)
+