dockerise

.dockerignore

@@ -0,0 +1,12 @@
+.git
+.gitignore
+.venv
+__pycache__/
+.ipynb_checkpoints/
+*.py[cod]
+*.ipynb
+.DS_Store
+debug_frames/
+force_06/
+out*.mp4
+output.png

Dockerfile

@@ -0,0 +1,23 @@
+FROM python:3.11-slim
+
+ENV PYTHONDONTWRITEBYTECODE=1 \
+    PYTHONUNBUFFERED=1 \
+    PIP_NO_CACHE_DIR=1 \
+    GRADIO_SERVER_NAME=0.0.0.0 \
+    GRADIO_SERVER_PORT=7860
+
+WORKDIR /app
+
+RUN apt-get update && apt-get install -y --no-install-recommends \
+    ffmpeg \
+    libgomp1 \
+    && rm -rf /var/lib/apt/lists/*
+
+COPY requirements.txt .
+RUN pip install --upgrade pip && pip install -r requirements.txt
+
+COPY . .
+
+EXPOSE 7860
+
+CMD ["python", "app.py"]

Makefile

@@ -0,0 +1,21 @@
+COMPOSE := docker compose
+
+.PHONY: run run-bg stop down logs build ps
+
+run:
+	$(COMPOSE) up --build -d
+
+stop:
+	$(COMPOSE) stop
+
+down:
+	$(COMPOSE) down
+
+logs:
+	$(COMPOSE) logs -f app
+
+build:
+	$(COMPOSE) build
+
+ps:
+	$(COMPOSE) ps

README.md

@@ -35,6 +35,22 @@ python app.py
 Gradio will print a local URL (for example, `http://127.0.0.1:7860`). Open it
 in your browser, upload an MP4, and click "Detect".
 
+## Docker Compose + Make
+Run with:
+```bash
+make run
+```
+
+Other commands:
+```bash
+make logs
+make stop
+make down
+```
+
+Then open `http://127.0.0.1:7860` in your browser.
+
 ## Notes
 - The first run downloads the wildfire detection model from Hugging Face.
-- OpenCV is required for video decoding and frame sampling.
+- `ffmpeg`/`ffprobe` are required for frame extraction.
+- OpenCV is used for motion features and image processing.

app.py

@@ -646,4 +646,7 @@ with gr.Blocks() as demo:
 
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch(
+        server_name=os.getenv("GRADIO_SERVER_NAME", "127.0.0.1"),
+        server_port=int(os.getenv("GRADIO_SERVER_PORT", "7860")),
+    )

docker-compose.yml

@@ -0,0 +1,12 @@
+services:
+  app:
+    build:
+      context: .
+      dockerfile: Dockerfile
+    container_name: pyronear-wildfire-detection
+    ports:
+      - "7860:7860"
+    environment:
+      GRADIO_SERVER_NAME: "0.0.0.0"
+      GRADIO_SERVER_PORT: "7860"
+    restart: unless-stopped