Update app/main.py

app/main.py

@@ -29,25 +29,26 @@ print("[BOOT] Model loaded")
 app = FastAPI(title="LFM2.5 Speech-to-Speech", version="3.0")
 
 
-def wav_header(sr=24000, ch=1, bits=16):
-    byte_rate = sr * ch * bits // 8
-    block_align = ch * bits // 8
+def wav_header(sample_rate: int, channels: int = 1, bits: int = 16) -> bytes:
+    byte_rate = sample_rate * channels * bits // 8
+    block_align = channels * bits // 8
     return (
         b"RIFF"
-        + b"\xff\xff\xff\xff"
+        + (b"\xff\xff\xff\xff")
         + b"WAVEfmt "
         + (16).to_bytes(4, "little")
         + (1).to_bytes(2, "little")
-        + ch.to_bytes(2, "little")
-        + sr.to_bytes(4, "little")
+        + channels.to_bytes(2, "little")
+        + sample_rate.to_bytes(4, "little")
         + byte_rate.to_bytes(4, "little")
         + block_align.to_bytes(2, "little")
         + bits.to_bytes(2, "little")
         + b"data"
-        + b"\xff\xff\xff\xff"
+        + (b"\xff\xff\xff\xff")
     )
 
 
+
 async def generate_response(websocket: WebSocket, audio_np: np.ndarray):
 
     chat = ChatState(processor)