Update Dockerfile

Dockerfile

@@ -4,11 +4,12 @@ FROM python:3.11-slim
 # Set working directory
 WORKDIR /app
 
-# Install system dependencies
+# Install system dependencies including wget for health check
 RUN apt-get update && apt-get install -y \
     gcc \
     g++ \
     make \
+    wget \
     && rm -rf /var/lib/apt/lists/*
 
 # Copy requirements first for better caching
@@ -23,21 +24,23 @@ COPY virtual_gpu/ ./virtual_gpu/
 # Copy the application source code
 COPY src/ ./src/
 
+# Copy the main application file
+COPY main.py .
+
 # Create necessary directories
-RUN mkdir -p /app/src/static /app/src/templates
+RUN mkdir -p /app/static /app/database /app/src/static /app/src/templates
 
 # Set environment variables
 ENV PYTHONPATH=/app
-ENV FLASK_APP=src.main
 ENV FLASK_ENV=production
+ENV PORT=5000
 
 # Expose port 5000
 EXPOSE 5000
 
-# Health check
+# Health check using wget instead of curl
 HEALTHCHECK --interval=30s --timeout=10s --start-period=5s --retries=3 \
-    CMD curl -f http://localhost:5000/api/gpu-status || exit 1
+    CMD wget --no-verbose --tries=1 --spider http://localhost:5000/health || exit 1
 
 # Run the application
-CMD ["python", "src/main.py"]
-
+CMD ["python", "main.py"]