stephan
528a185d3b
This commit introduces a wide range of improvements to the application, focusing on stability, developer experience (DX), and documentation.
Key changes include:
- **Fix Application Startup:** Resolved a critical bug where the FastAPI application instance was not correctly exposed, preventing Uvicorn from starting ().
- **Simplify Docker Compose:** Removed the integrated Traefik setup from the default to support users with existing reverse proxies and simplify the local development environment.
- **Improve Makefile:**
- Implemented a robust, automatic port-finding mechanism for Starting development environment on port 8001...
#1 [internal] load local bake definitions
#1 reading from stdin 534B done
#1 DONE 0.0s
#2 [internal] load build definition from Dockerfile
#2 transferring dockerfile: 1.22kB done
#2 WARN: FromAsCasing: 'as' and 'FROM' keywords' casing do not match (line 2)
#2 DONE 0.0s
#3 [internal] load metadata for docker.io/library/python:3.11
#3 DONE 0.7s
#4 [internal] load metadata for docker.io/library/python:3.11-slim
#4 DONE 0.7s
#5 [internal] load .dockerignore
#5 transferring context: 385B done
#5 DONE 0.0s
#6 [builder 1/4] FROM docker.io/library/python:3.11@sha256:bf2d36b8fb1b4a0b590b36736cdd8a6b5175b411bf135c42694ecd68ab8fed02
#6 DONE 0.0s
#7 [stage-1 1/6] FROM docker.io/library/python:3.11-slim@sha256:193fdd0bbcb3d2ae612bd6cc3548d2f7c78d65b549fcaa8af75624c47474444d
#7 DONE 0.0s
#8 [internal] load build context
#8 transferring context: 4.90kB done
#8 DONE 0.0s
#9 [builder 2/4] WORKDIR /opt/venv
#9 CACHED
#10 [stage-1 4/6] WORKDIR /home/appuser
#10 CACHED
#11 [stage-1 3/6] RUN useradd --create-home --shell /bin/bash appuser
#11 CACHED
#12 [stage-1 2/6] RUN apt-get update && apt-get install -y --no-install-recommends ffmpeg && rm -rf /var/lib/apt/lists/*
#12 CACHED
#13 [stage-1 5/6] COPY --from=builder /opt/venv /opt/venv
#13 CACHED
#14 [builder 4/4] RUN python -m venv . && . /opt/venv/bin/activate && pip install --no-cache-dir -r requirements.txt
#14 CACHED
#15 [builder 3/4] COPY requirements.txt .
#15 CACHED
#16 [stage-1 6/6] COPY app/ ./app
#16 CACHED
#17 exporting to image
#17 exporting layers done
#17 writing image sha256:6cac7caac7fda2808672ad2f3d117d46d38c1d93013867858543ec74917857b7 done
#17 naming to docker.io/library/audioenginehub-app done
#17 DONE 0.0s
#18 resolving provenance for metadata file
#18 DONE 0.0s and Using host port 8000 for single app container
8a1c69e868e7f13b4c8c9948e81921b48efd9536f326d200b9e912fb12ff66e3 to prevent port conflicts.
- Added a target (Running health check on running container...
App container is running on port 8001.
Waiting for app to initialize...
ERROR: Failed to decode JSON from health endpoint.) to run post-deployment sanity checks against the running container's endpoint.
- Recommended using Starting development environment on port 8002...
#1 [internal] load local bake definitions
#1 reading from stdin 534B done
#1 DONE 0.0s
#2 [internal] load build definition from Dockerfile
#2 transferring dockerfile: 1.22kB done
#2 WARN: FromAsCasing: 'as' and 'FROM' keywords' casing do not match (line 2)
#2 DONE 0.0s
#3 [internal] load metadata for docker.io/library/python:3.11-slim
#3 DONE 0.1s
#4 [internal] load metadata for docker.io/library/python:3.11
#4 DONE 0.2s
#5 [internal] load .dockerignore
#5 transferring context: 385B done
#5 DONE 0.0s
#6 [builder 1/4] FROM docker.io/library/python:3.11@sha256:bf2d36b8fb1b4a0b590b36736cdd8a6b5175b411bf135c42694ecd68ab8fed02
#6 DONE 0.0s
#7 [stage-1 1/6] FROM docker.io/library/python:3.11-slim@sha256:193fdd0bbcb3d2ae612bd6cc3548d2f7c78d65b549fcaa8af75624c47474444d
#7 DONE 0.0s
#8 [internal] load build context
#8 transferring context: 1.09GB 5.1s
#8 transferring context: 1.66GB 7.9s done
#8 DONE 8.0s
#9 [builder 3/4] COPY requirements.txt .
#9 CACHED
#10 [builder 4/4] RUN python -m venv . && . /opt/venv/bin/activate && pip install --no-cache-dir -r requirements.txt
#10 CACHED
#11 [stage-1 4/6] WORKDIR /home/appuser
#11 CACHED
#12 [stage-1 3/6] RUN useradd --create-home --shell /bin/bash appuser
#12 CACHED
#13 [builder 2/4] WORKDIR /opt/venv
#13 CACHED
#14 [stage-1 2/6] RUN apt-get update && apt-get install -y --no-install-recommends ffmpeg && rm -rf /var/lib/apt/lists/*
#14 CACHED
#15 [stage-1 5/6] COPY --from=builder /opt/venv /opt/venv
#15 CACHED
#16 [stage-1 6/6] COPY app/ ./app
#16 CACHED
#17 exporting to image
#17 exporting layers done
#17 writing image sha256:6cac7caac7fda2808672ad2f3d117d46d38c1d93013867858543ec74917857b7 done
#17 naming to docker.io/library/audioenginehub-app done
#17 DONE 0.0s
#18 resolving provenance for metadata file
#18 DONE 0.0s for reliable port detection.
- **Update Documentation:**
- Replaced the outdated (which contained old source code) with a comprehensive guide covering setup, usage, and commands.
- Added a note to to clarify that it describes an older, more advanced setup, pointing readers to the new for the current recommended workflow.
These changes address the service startup failures and significantly improve the project's usability and maintainability.
70 lines
3.1 KiB
Python
70 lines
3.1 KiB
Python
import unittest
|
|
import os
|
|
import shutil
|
|
import asyncio
|
|
from importlib.resources import files
|
|
from app.engines.f5_tts import F5TTSEngine
|
|
|
|
class TestF5TTSEngine(unittest.TestCase):
|
|
async def asyncSetUp(self):
|
|
self.engine = F5TTSEngine()
|
|
self.voices_dir = "app/models/f5-tts-voices"
|
|
self.test_speaker_name = "test_speaker"
|
|
self.test_speaker_wav = os.path.join(self.voices_dir, f"{self.test_speaker_name}.wav")
|
|
self.test_speaker_txt = os.path.join(self.voices_dir, f"{self.test_speaker_name}.txt")
|
|
|
|
# Create a dummy speaker for testing
|
|
if not await asyncio.to_thread(os.path.exists, self.test_speaker_wav):
|
|
default_wav_path = str(files("f5_tts").joinpath("infer/examples/basic/basic_ref_en.wav"))
|
|
await asyncio.to_thread(shutil.copy, default_wav_path, self.test_speaker_wav)
|
|
if not await asyncio.to_thread(os.path.exists, self.test_speaker_txt):
|
|
await asyncio.to_thread(lambda: open(self.test_speaker_txt, "w").write("Some call me nature, others call me mother nature."))
|
|
|
|
# Reload speakers to include the new test speaker
|
|
self.engine._load_speakers()
|
|
|
|
async def asyncTearDown(self):
|
|
# Clean up the dummy speaker files
|
|
if await asyncio.to_thread(os.path.exists, self.test_speaker_wav):
|
|
await asyncio.to_thread(os.remove, self.test_speaker_wav)
|
|
if await asyncio.to_thread(os.path.exists, self.test_speaker_txt):
|
|
await asyncio.to_thread(os.remove, self.test_speaker_txt)
|
|
|
|
async def test_synthesize_default_wav(self):
|
|
if not self.engine.model:
|
|
self.skipTest("F5-TTS model not initialized.")
|
|
|
|
text = "Hello, this is a test with the default voice."
|
|
audio_file = await self.engine.synthesize(text, fmt="wav")
|
|
|
|
self.assertTrue(await asyncio.to_thread(os.path.exists, audio_file))
|
|
self.assertTrue(await asyncio.to_thread(os.path.getsize, audio_file) > 0)
|
|
|
|
await asyncio.to_thread(os.remove, audio_file)
|
|
|
|
async def test_synthesize_custom_speaker_wav(self):
|
|
if not self.engine.model:
|
|
self.skipTest("F5-TTS model not initialized.")
|
|
|
|
text = "Hello, this is a test with a custom voice."
|
|
audio_file = await self.engine.synthesize(text, speaker=self.test_speaker_name, fmt="wav")
|
|
|
|
self.assertTrue(await asyncio.to_thread(os.path.exists, audio_file))
|
|
self.assertTrue(await asyncio.to_thread(os.path.getsize, audio_file) > 0)
|
|
|
|
await asyncio.to_thread(os.remove, audio_file)
|
|
|
|
async def test_synthesize_ogg(self):
|
|
if not self.engine.model:
|
|
self.skipTest("F5-TTS model not initialized.")
|
|
|
|
text = "Hello, this is a test in ogg format."
|
|
audio_file = await self.engine.synthesize(text, fmt="ogg")
|
|
|
|
self.assertTrue(await asyncio.to_thread(os.path.exists, audio_file))
|
|
self.assertTrue(await asyncio.to_thread(os.path.getsize, audio_file) > 0)
|
|
|
|
await asyncio.to_thread(os.remove, audio_file)
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main() |