image generation added

docker/Dockerfile

@@ -31,10 +31,13 @@ ENV PATH=/root/.local/bin:$PATH
 COPY src/ ./src/
 COPY api/ ./api/
 COPY web/ ./web/
+COPY generate_images.py ./
 
 # Create directories for data
 RUN mkdir -p saved_mazes output_images
 
+RUN python generate_images.py
+
 # Expose port
 EXPOSE 5000