From ec2d0987aca9d21a9f5f58bcfdfaa027adb2ccb4 Mon Sep 17 00:00:00 2001
From: Ryan Voots <simcop2387@simcop2387.info>
Date: Tue, 23 Apr 2024 09:50:43 -0400
Subject: [PATCH] Switch to wyoming and this for the docker container

---
 Dockerfile       | 16 +++++++++++-----
 requirements.txt |  7 ++-----
 2 files changed, 13 insertions(+), 10 deletions(-)

diff --git a/Dockerfile b/Dockerfile
index 0545123..4f34681 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -1,12 +1,18 @@
 FROM nvidia/cuda:12.4.1-devel-ubuntu22.04
 
 RUN apt-get update && apt-get install -y \
-    git-lfs python3 build-essential python3-pip \
+    git-lfs python3 build-essential python3-pip python3-venv \
     && rm -rf /var/lib/apt/lists/*
 
 WORKDIR /app
-COPY requirements.txt .
-RUN pip3 install -r requirements.txt
-COPY detect.py .
+RUN git clone https://github.com/rhasspy/wyoming-openwakeword --depth 1 --single-branch --recurse-submodules -b v1.10.0
+
+WORKDIR /app/wyoming-openwakeword/
+RUN python3 script/setup
+
+WORKDIR /app
+COPY requirements.txt .
+RUN /app/wyoming-openwakeword/.venv/bin/pip install -r requirements.txt
+
+ENTRYPOINT ["/app/wyoming-openwakeword/.venv/bin/python", "/app/wyoming-openwakeword/script/run"]
 
-CMD [ "python3", "-u", "detect.py", "-c", "/config/config.yaml"]
diff --git a/requirements.txt b/requirements.txt
index 7bd8090..e01702c 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,5 +1,2 @@
---find-links https://download.pytorch.org/whl/torch_stable.html
-openwakeword @ git+https://github.com/dscripka/openWakeWord@v0.6.0
-paho-mqtt==1.6.1
-pyyaml==6.0
-torch==2.1.0+cu121
+tensorflow[and-cuda]
+