Update Dockerfile

Dockerfile

@@ -1,41 +1,33 @@
-# Use NVIDIA's CUDA base image with Python
-FROM nvidia/cuda:11.8.0-runtime-ubuntu20.04
+# Use an official Python runtime as a parent image
+FROM python:3.9-slim
 
-# Set environment variables
-ENV LANG=C.UTF-8 LC_ALL=C.UTF-8
-ENV PATH="/home/user/.local/bin:$PATH"
+# Set environment variables for Python
+ENV PYTHONDONTWRITEBYTECODE 1
+ENV PYTHONUNBUFFERED 1
 
-# Install Python, pip, and other dependencies
-RUN apt-get update && apt-get install -y \
-    python3 \
-    python3-pip \
-    python3-dev \
-    libgl1-mesa-glx \
-    libglib2.0-0 \  # Required for GLib
-    && apt-get clean && rm -rf /var/lib/apt/lists/*
-
-# Create a non-root user
-RUN useradd -m user
-
-# Set the user to the non-root user
-USER user
-
-# Set the working directory to /app
+# Set the working directory inside the container
 WORKDIR /app
 
-# Copy the requirements.txt file and install dependencies
-COPY --chown=user requirements.txt /app/requirements.txt
-RUN pip3 install --no-cache-dir --upgrade -r /app/requirements.txt
+# Copy the requirements file into the container at /app
+COPY requirements.txt /app/
 
-# Copy the application code into the container
-COPY --chown=user . /app
+# Install the dependencies specified in requirements.txt
+RUN pip install --no-cache-dir -r requirements.txt
 
-# Create the folders for uploads and results with appropriate permissions
-RUN mkdir -p /app/static/uploads /app/static/results && \
-    chmod -R 755 /app/static/uploads /app/static/results
+# Create the necessary directories and ensure they have proper permissions
+RUN mkdir -p /app/uploads /app/static/uploads /app/static/results /tmp/flask_sessions /app/flask_sessions && \
+    chmod -R 777 /app/uploads /app/static/uploads /app/static/results /tmp/flask_sessions /app/flask_sessions
 
-# Expose the port the app will run on
+# Copy the entire application code to /app
+COPY . /app/
+
+# Expose the port used by the Flask app
 EXPOSE 7860
 
-# Run the Flask app using uvicorn
-CMD ["python3", "-m", "uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]
+# Set environment variables for Flask
+ENV FLASK_APP=app.py
+ENV FLASK_ENV=production
+ENV SESSION_FILE_DIR=/app/flask_sessions
+
+# Run the application with Gunicorn, setting workers and timeout
+CMD ["gunicorn", "-w", "1", "-b", "0.0.0.0:7860", "--timeout", "120", "app:app"]