initial docker

Dockerfile

@@ -1,16 +1,15 @@
-FROM python:3.7-slim-stretch
+FROM python:3.8
+LABEL maintainer="Robin Cole @robmarkcole"
 
-RUN apt-get update && apt-get install -y git python3-dev gcc \
-    && rm -rf /var/lib/apt/lists/*
+EXPOSE 8501
 
 COPY requirements.txt .
+RUN pip install --upgrade pip
+RUN pip install -r requirements.txt
 
-RUN pip install --upgrade -r requirements.txt
+RUN mkdir -p /app
+COPY app /app
+WORKDIR /app
 
-COPY app app/
-
-RUN python app/server.py
-
-EXPOSE 5000
-
-CMD ["python", "app/server.py", "serve"]
+ENTRYPOINT [ "streamlit", "run"]
+CMD ["streamlit-ui.py"]

README.md

@@ -21,6 +21,17 @@ venv/bin/streamlit run streamlit-ui.py
 <img src="https://github.com/robmarkcole/deepstack-ui/blob/master/usage.png" width="800">
 </p>
 
+## Run with Docker
+From the root dir, build from source:
+```
+    docker build -t deepstack-ui .
+    docker run -p 8501:8501 -e DEEPSTACK_IP='192.168.1.133' deepstack-ui:latest
+```
+
+Alternatively run the pre-built image: `docker run -p 8501:8501 -e DEEPSTACK_IP='192.168.1.133' robmarkcole/deepstack-ui:latest`
+
+Then visit [localhost:8501](http://localhost:8501/)
+
 ## Development
 Use venv:
 ```

streamlit-ui.py → app/streamlit-ui.py

@@ -10,7 +10,7 @@
 import const
 
 ## Depstack setup
-DEEPSTACK_IP_ADDRESS = "localhost"
+DEEPSTACK_IP = os.getenv("DEEPSTACK_IP", "set-your-deepstack-ip")
 DEEPSTACK_PORT = "5000"
 DEEPSTACK_API_KEY = ""
 DEEPSTACK_TIMEOUT = 20  # Default is 10
@@ -46,7 +46,7 @@ def process_image(pil_image, dsobject):
     pil_image = Image.open(TEST_IMAGE)
 
 dsobject = ds.DeepstackObject(
-    DEEPSTACK_IP_ADDRESS, DEEPSTACK_PORT, DEEPSTACK_API_KEY, DEEPSTACK_TIMEOUT
+    DEEPSTACK_IP, DEEPSTACK_PORT, DEEPSTACK_API_KEY, DEEPSTACK_TIMEOUT
 )
 
 predictions, summary = process_image(pil_image, dsobject)