Update main.py

main.py

@@ -549,26 +549,14 @@ async def text_to_speech_v2(request: AudioAPI.TTSRequest, api_key: str = Depends
             model=request.model,
             voice=request.voice,
             input=request.input,
-            response_format="mp3"  # Always set to MP3
+            response_format=request.output_format
         )
         
-        return StreamingResponse(io.BytesIO(response.content), media_type="audio/mp3")
-    
-    except Exception as e:
-        raise HTTPException(status_code=500, detail=str(e))
-    # try:
-    #     response = openai_client.audio.speech.create(
-    #         model=request.model,
-    #         voice=request.voice,
-    #         input=request.input,
-    #         response_format=request.output_format
-    #     )
-        
-    #     content_type = f"audio/{request.output_format.value}"
-    #     if request.output_format == OpenaiTTSModels.OutputFormat.pcm:
-    #         content_type = "audio/pcm"
+        content_type = f"audio/{request.output_format.value}"
+        if request.output_format == OpenaiTTSModels.OutputFormat.pcm:
+            content_type = "audio/pcm"
         
-    #     return StreamingResponse(io.BytesIO(response.content), media_type=content_type)
+        return StreamingResponse(io.BytesIO(response.content), media_type=content_type)
     
     except Exception as e:
         raise HTTPException(status_code=500, detail=str(e))