update space

Dockerfile

@@ -0,0 +1,26 @@
+# File: Dockerfile
+
+# Use an official Python runtime as a parent image
+FROM python:3.10-slim
+
+
+# Create and set the working directory
+WORKDIR /app
+
+# Copy the requirements file into the container
+COPY requirements.txt /app/
+
+# Install any needed packages specified in requirements.txt
+RUN pip install --no-cache-dir -r requirements.txt
+
+# Copy the rest of the application code into the container
+COPY . /app
+
+# Install Gunicorn
+RUN pip install gunicorn
+
+# Expose the port that the app runs on
+EXPOSE 5000
+
+# Command to run the application using Gunicorn
+CMD ["gunicorn", "app:app", "-b", "0.0.0.0:5000", "-w", "2", "--threads", "4", "--timeout", "120"]

app.py

@@ -13,6 +13,8 @@ from flask import Flask, render_template, request, redirect, url_for, session
 from celery import Celery
 import time
 import threading
+import gunicorn.app.base
+from gunicorn.six import iteritems
 
 # Setup logging for better monitoring
 logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
@@ -210,11 +212,31 @@ def chatbot():
     chatbot_interface = build_chatbot_interface({'username': username})
     return chatbot_interface.launch(inline=True, share=True)
 
-# Launch the Flask application
+# Custom Gunicorn application to run Flask safely in a production environment like Hugging Face Space Docker
+class StandaloneApplication(gunicorn.app.base.BaseApplication):
+    def __init__(self, app, options=None):
+        self.application = app
+        self.options = options or {}
+        super().__init__()
+
+    def load_config(self):
+        config = {key: value for key, value in iteritems(self.options) if key in self.cfg.settings and value is not None}
+        for key, value in config.items():
+            self.cfg.set(key.lower(), value)
+
+    def load(self):
+        return self.application
+
+# Launch the Flask application using Gunicorn in a production-safe manner
 if __name__ == "__main__":
     try:
-        logger.info("Launching Flask server for AdminLTE authentication.")
-        app.run(host='0.0.0.0', port=5000, debug=False)
+        options = {
+            'bind': '%s:%s' % ('0.0.0.0', '5000'),
+            'workers': 2,  # Set number of worker processes
+            'threads': 4,  # Set number of threads per worker
+            'timeout': 120,  # Set timeout for workers
+        }
+        StandaloneApplication(app, options).run()
     except Exception as e:
         logger.error(f"Failed to launch Flask server: {e}")
         raise RuntimeError("Could not launch the Flask server. Please check the application setup.")