Update app.py

app.py

@@ -81,14 +81,16 @@ def tts_post(request: TTSRequest): # Removed 'async' to prevent event loop block
         wav_buffer = io.BytesIO()
         
         with wave.open(wav_buffer, "wb") as wav_file:
-            wav_file.setnchannels(1) 
-            wav_file.setsampwidth(2) 
+            wav_file.setnchannels(1)
+            wav_file.setsampwidth(2)
             wav_file.setframerate(voice.config.sample_rate)
             
-            # Use the generator directly
             for chunk in voice.synthesize(request.text):
-                wav_file.writeframes(chunk)
-
+                if hasattr(chunk, "__iter__") and not isinstance(chunk, (bytes, bytearray)):
+                    import struct
+                    wav_file.writeframes(struct.pack(f"{len(chunk)}h", *chunk))
+                else:
+                    wav_file.writeframes(chunk)        
         wav_buffer.seek(0)
         return Response(content=wav_buffer.getvalue(), media_type="audio/wav")
 
@@ -103,4 +105,14 @@ def home():
 
 @app.get("/")
 def home():
-    return {"languages": VOICE_MAP}
+    # List all files in the models directory
+    try:
+        files = os.listdir(MODEL_DIR)
+    except Exception as e:
+        files = [f"Error reading directory: {str(e)}"]
+
+    return {
+        "message": "Piper TTS API is running",
+        "models_in_folder": files,
+        "supported_languages": VOICE_MAP
+    }