Update audio.py

routes/audio.py

@@ -1,12 +1,21 @@
 from fastapi import APIRouter, UploadFile, File, HTTPException
 from fastapi.responses import StreamingResponse
 import io
-import mimetypes
 from services.tts_service import generate_tts
 from services.stt_service import speech_to_text
 
 router = APIRouter(prefix="/audio", tags=["Audio"])
 
+# Allowed MIME types
+ALLOWED_AUDIO_TYPES = {
+    "audio/wav",
+    "audio/x-wav",
+    "audio/mpeg",    # mp3
+    "audio/mp3",     # mp3
+    "audio/mp4",     # sometimes m4a
+    "audio/m4a"      # m4a
+}
+
 # ------------------------
 # Text to Speech
 # ------------------------
@@ -15,6 +24,12 @@ async def tts(text: str):
     """
     Convert text to speech and return audio.
     """
+    if not text or len(text) > 500:
+        raise HTTPException(
+            status_code=400,
+            detail="Text must be between 1 and 500 characters"
+        )
+    
     try:
         audio_bytes = await generate_tts(text)
     except Exception as e:
@@ -26,17 +41,6 @@ async def tts(text: str):
 # ------------------------
 # Speech to Text
 # ------------------------
-
-# Allowed MIME types
-ALLOWED_AUDIO_TYPES = {
-    "audio/wav",
-    "audio/x-wav",
-    "audio/mpeg",    # mp3
-    "audio/mp3",     # mp3
-    "audio/mp4",     # sometimes m4a
-    "audio/m4a"      # m4a
-}
-
 @router.post("/stt")
 async def stt(file: UploadFile = File(...)):
     """
@@ -51,7 +55,7 @@ async def stt(file: UploadFile = File(...)):
 
     try:
         audio_bytes = await file.read()
-        text = await speech_to_text(audio_bytes, file.filename)  # Pass filename for correct MIME
+        text = await speech_to_text(audio_bytes, file.filename)
     except Exception as e:
         raise HTTPException(status_code=500, detail=str(e))
     
@@ -84,4 +88,4 @@ async def chatbot(file: UploadFile = File(...)):
     except Exception as e:
         raise HTTPException(status_code=500, detail=str(e))
 
-    return StreamingResponse(io.BytesIO(audio_response), media_type="audio/wav")
+    return StreamingResponse(io.BytesIO(audio_response), media_type="audio/wav")