Aau/#9/refacto ray

.gitignore

@@ -175,3 +175,6 @@ cython_debug/
 
 # PyPI configuration file
 .pypirc
+
+# Others
+.trash

Makefile

@@ -0,0 +1,10 @@
+lint:
+	black .
+	isort .
+	flake8 .
+	mypy .
+
+test:
+	pytest
+
+check: lint test

pyproject.toml

@@ -0,0 +1,31 @@
+[tool.black]
+line-length = 88
+target-version = ["py310"]
+
+[tool.isort]
+profile = "black"
+line_length = 88
+multi_line_output = 3
+include_trailing_comma = true
+skip_gitignore = true
+
+[tool.flake8]
+max-line-length = 88
+extend-ignore = []
+exclude = """
+  __pycache__
+  venv
+  .venv
+"""
+
+[tool.mypy]
+python_version = "3.10"
+ignore_missing_imports = true
+strict_optional = true
+warn_unused_ignores = true
+warn_return_any = true
+warn_unused_configs = true
+
+[tool.pytest.ini_options]
+testpaths = ["tests"]
+python_files = ["test_*.py"]

requirements.txt

@@ -1,3 +1,10 @@
+black
+isort
+mypy
+flake8
+flake8-toml-config
+pytest
+
 deepfilternet
 sounddevice
 soundfile

src/client.py

@@ -0,0 +1,45 @@
+import asyncio
+
+import numpy as np
+import sounddevice as sd
+import websockets
+
+SERVER_URL = "ws://localhost:8000/session"
+CHUNK_DURATION = 1
+SAMPLE_RATE = 16000
+
+
+async def stream_audio():
+
+    async with websockets.connect(SERVER_URL) as ws:
+        print("Connected to server")
+
+        async def receive(ws: websockets.ClientConnection):
+            while True:
+                text = await ws.recv()
+                print("received:", text)
+
+        async def send(ws: websockets.ClientConnection):
+            loop = asyncio.get_running_loop()
+
+            queue: asyncio.Queue = asyncio.Queue()
+
+            def callback(indata: np.ndarray, frames, time, status):
+                loop.call_soon_threadsafe(queue.put_nowait, indata.copy())
+
+            with sd.InputStream(
+                samplerate=SAMPLE_RATE,
+                channels=1,
+                dtype="int16",
+                callback=callback,
+                blocksize=int(CHUNK_DURATION * SAMPLE_RATE),
+            ):
+                while True:
+                    chunk = await queue.get()
+                    await ws.send(chunk.tobytes())
+
+        await asyncio.gather(receive(ws), send(ws))
+
+
+if __name__ == "__main__":
+    asyncio.run(stream_audio())