First

Dockerfile

@@ -4,10 +4,14 @@ RUN useradd -m -u 1000 user
 USER user
 ENV PATH="/home/user/.local/bin:$PATH"
 
+# Install Xvfb for virtual display
+RUN apt-get update && apt-get install -y xvfb
+
 WORKDIR /app
 
 COPY --chown=user ./requirements.txt requirements.txt
 RUN pip install --no-cache-dir --upgrade -r requirements.txt
 
 COPY --chown=user . /app
-CMD ["uvicorn", "api:app", "--host", "0.0.0.0", "--port", "7860"]
+# Start Xvfb and run the application
+CMD sh -c "Xvfb :99 -screen 0 1024x768x24 > /dev/null 2>&1 & export DISPLAY=:99 && uvicorn api:app --host 0.0.0.0 --port 7860"

api.py

@@ -31,10 +31,6 @@ if platform.system() == 'Linux':
     logger = logging.getLogger(__name__)
     logger.info("Started virtual display for Linux environment")
 
-# --- Logging Setup ---
-# Configure logging to output to both console and a file named 'app.log'
-log_file_path = "app.log"
-# The 'force=True' argument is used to reconfigure the root logger
 logging.basicConfig(
     level=logging.INFO,
     format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',