Add application file

Dockerfile

@@ -1,27 +1,27 @@
-FROM python:3.9
+FROM python:3.9-slim
 
+# Create and use a non-root user for security
 RUN useradd -m -u 1000 user
 USER user
 ENV PATH="/home/user/.local/bin:$PATH"
 
 WORKDIR /app
 
+# Copy necessary files
 COPY --chown=user ./requirements.txt requirements.txt
-
 COPY --chown=user ./lib/aicloudlibs-0.1.0-py3-none-any.whl /lib/
-
 COPY --chown=user ./lib/better_profanity-2.0.0-py3-none-any.whl /lib/
-
 COPY --chown=user ./lib/privacy-1.0.9-py3-none-any.whl /lib/
 
+# Install dependencies
 RUN pip install --no-cache-dir --upgrade -r requirements.txt
 
-COPY --chown=user . /app
-
-# Expose the port (default for Hugging Face is 7860)
+# Expose the port used by Hugging Face Spaces (7860)
 EXPOSE 7860
 
-# CMD to run the FastAPI app with Uvicorn
-CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]
+# Copy the rest of the app
+COPY --chown=user . /app
 
+# Command to run the app (avoid Waitress here)
+CMD ["python", "main.py"]
 

main.py

@@ -8,46 +8,38 @@ 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, UnprocessableEntity, InternalServerError
 from flask import Flask
-from router.router import app  # Assuming 'app' is a Flask blueprint
-from waitress import serve
-import os
+from router.router import app  # Make sure this is a Flask Blueprint
 from flask_swagger_ui import get_swaggerui_blueprint
 from flask_cors import CORS
-from config.logger import CustomLogger, request_id_var
 from dotenv import load_dotenv
+import os
 
 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)
+SWAGGER_URL = '/rai/v1/moderations/docs'
+API_URL = '/static/metadata.json'
 
-# Swagger UI Blueprint setup
 swaggerui_blueprint = get_swaggerui_blueprint(
-    SWAGGER_URL,  # Swagger UI static files will be mapped to '{SWAGGER_URL}/dist/'
-    API_URL,
-    config={  # Swagger UI config overrides
-        'app_name': "Infosys Responsible AI - Moderation"
-    },
+    SWAGGER_URL, API_URL,
+    config={'app_name': "Infosys Responsible AI - Moderation"}
 )
 
 # Create Flask app instance
 app1 = Flask(__name__)
 
-# Register Blueprints (Flask app and Swagger UI)
-app1.register_blueprint(app)  # app is the Flask Blueprint
+# Register blueprints
+app1.register_blueprint(app)
 app1.register_blueprint(swaggerui_blueprint)
 
-# CORS configuration
 CORS(app1)
 
-# Error Handlers
+# 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,
@@ -58,7 +50,6 @@ def handle_exception(e):
 
 @app1.errorhandler(UnprocessableEntity)
 def validation_error_handler(exc):
-    """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])
@@ -72,7 +63,6 @@ def validation_error_handler(exc):
 
 @app1.errorhandler(InternalServerError)
 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,
@@ -81,12 +71,10 @@ def internal_server_error_handler(exc):
     response.content_type = "application/json"
     return response
 
-# Start the server using waitress
+# Start Flask server
 if __name__ == "__main__":
     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)))
+    app1.run(host='0.0.0.0', port=int(os.getenv("PORT", 7860)), debug=True)
+
 
 

requirements.txt

@@ -15,6 +15,8 @@ setuptools
 flask
 Gunicorn
 flask[async]
+werkzeug==2.2.3
+python-dotenv==1.0.0
 flask-cors
 azure-ai-translation-text==1.0.0b1
 waitress