Add ffmpeg and better error logging

Dockerfile

@@ -5,6 +5,8 @@ WORKDIR /app
 # Install system dependencies
 RUN apt-get update && apt-get install -y \
     git \
+    build-essential \
+    ffmpeg \
     && rm -rf /var/lib/apt/lists/*
 
 # Install uv for fast package management

pocket_tts/main.py

@@ -111,19 +111,26 @@ def write_to_queue(queue, text_to_generate, model_state, **kwargs):
 def generate_data_with_state(text_to_generate: str, model_state: dict, **kwargs):
     queue = Queue()
 
+    def worker():
+        try:
+            write_to_queue(queue, text_to_generate, model_state, **kwargs)
+        except Exception as e:
+            logger.error(f"Error in generation worker: {e}", exc_info=True)
+            queue.put(e)  # Send exception to main thread
+
     # Run your function in a thread
-    thread = threading.Thread(
-        target=write_to_queue, args=(queue, text_to_generate, model_state), kwargs=kwargs
-    )
+    thread = threading.Thread(target=worker)
     thread.start()
 
     # Yield data as it becomes available
-    i = 0
     while True:
         data = queue.get()
         if data is None:
             break
-        i += 1
+        if isinstance(data, Exception):
+            # Re-raise the exception to abort the stream (will cause 500 on client if headers not sent, or truncated stream)
+            logger.error("Re-raising exception from worker in generator")
+            raise data
         yield data
 
     thread.join()