Add application file

main.py

@@ -8,25 +8,24 @@ The above copyright notice and this permission notice shall be included in all c
 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
-from werkzeug.exceptions import HTTPException,BadRequest,UnprocessableEntity,InternalServerError
+from werkzeug.exceptions import HTTPException, BadRequest, UnprocessableEntity, InternalServerError
 from flask import Flask
-from router.router import app
+from router.router import app  # Assuming 'app' is a Flask blueprint
 from waitress import serve
 import os
 from flask_swagger_ui import get_swaggerui_blueprint
 from flask_cors import CORS
-from config.logger import CustomLogger, request_id_var 
-# import os
+from config.logger import CustomLogger, request_id_var
 from dotenv import load_dotenv
+
 load_dotenv()
+
 request_id_var.set("StartUp")
+
 SWAGGER_URL = '/rai/v1/moderations/docs'  # URL for exposing Swagger UI (without trailing '/')
-# API_URL = '/static/metadata.json'  # Our API url (can of course be a local resource)
-# API_URL = 'src/config/swagger/metadata.json'  # Our API url (can of course be a local resource)
 API_URL = '/static/metadata.json'  # Our API url (can of course be a local resource)
 
-
-# Call factory function to create our blueprint
+# Swagger UI Blueprint setup
 swaggerui_blueprint = get_swaggerui_blueprint(
     SWAGGER_URL,  # Swagger UI static files will be mapped to '{SWAGGER_URL}/dist/'
     API_URL,
@@ -34,26 +33,22 @@ swaggerui_blueprint = get_swaggerui_blueprint(
         'app_name': "Infosys Responsible AI - Moderation"
     },
 )
-  
+
+# Create Flask app instance
 app1 = Flask(__name__)
-# app1.register_blueprint(app)
 
-# CORS(app1, origins="*", methods="*", headers="*")
+# Register Blueprints (Flask app and Swagger UI)
+app1.register_blueprint(app)  # app is the Flask Blueprint
+app1.register_blueprint(swaggerui_blueprint)
 
+# CORS configuration
 CORS(app1)
 
-app1.register_blueprint(app)
-
-
-app1.register_blueprint(swaggerui_blueprint)
- 
- 
+# Error Handlers
 @app1.errorhandler(HTTPException)
 def handle_exception(e):
     """Return JSON instead of HTML for HTTP errors."""
-    # start with the correct headers and status code from the error
     response = e.get_response()
-    # replace the body with JSON
     response.data = json.dumps({
         "code": e.code,
         "details": e.description,
@@ -63,38 +58,35 @@ def handle_exception(e):
 
 @app1.errorhandler(UnprocessableEntity)
 def validation_error_handler(exc):
-        """Return JSON instead of HTML for HTTP errors."""
-        # start with the correct headers and status code from the error
-        response = exc.get_response()
-        print(response)
-        # replace the body with JSON
-        exc_code_desc=exc.description.split("-")
-        exc_code=int(exc_code_desc[0])
-        exc_desc=exc_code_desc[1]
-        response.data = json.dumps({
-            "code": exc_code,
-            "details":  exc_desc,
-        })
-        response.content_type = "application/json"
-        return response
+    """Return JSON instead of HTML for validation errors (422)."""
+    response = exc.get_response()
+    exc_code_desc = exc.description.split("-")
+    exc_code = int(exc_code_desc[0])
+    exc_desc = exc_code_desc[1]
+    response.data = json.dumps({
+        "code": exc_code,
+        "details": exc_desc,
+    })
+    response.content_type = "application/json"
+    return response
 
 @app1.errorhandler(InternalServerError)
-def validation_error_handler(exc):
-        """Return JSON instead of HTML for HTTP errors."""
-        # start with the correct headers and status code from the error
-        response = exc.get_response()
-        print(response)
-        # replace the body with JSON
-        response.data = json.dumps({
-            "code": 500,
-            "details": "Some Error Occurred ,Please try Later",
-        })
-        response.content_type = "application/json"
-        return response
-
-    
+def internal_server_error_handler(exc):
+    """Return JSON instead of HTML for internal server errors (500)."""
+    response = exc.get_response()
+    response.data = json.dumps({
+        "code": 500,
+        "details": "Some Error Occurred, Please try later",
+    })
+    response.content_type = "application/json"
+    return response
+
+# Start the server using waitress
 if __name__ == "__main__":
-   request_id_var.set("StartUp")
-   serve(app1, host='0.0.0.0', port=int(os.getenv("PORT")), threads=int(os.getenv('THREADS',6)),connection_limit=int(os.getenv('CONNECTION_LIMIT',500)), channel_timeout=int(os.getenv('CHANNEL_TIMEOUT',120)))
+    request_id_var.set("StartUp")
+    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)))