Update main.py

main.py

@@ -247,6 +247,24 @@ async def llm_agent(query: LLMAgentQueryModel, background_tasks: BackgroundTasks
 
     return StreamingResponse(process_response(), media_type="text/event-stream")
 
+import edge_tts
+import io
+
+
+@app.get("/tts")
+async def text_to_speech(
+    text: str = Query(..., description="Text to convert to speech"),
+    voice: str = Query(default="en-GB-SoniaNeural", description="Voice to use for speech")
+):
+    communicate = edge_tts.Communicate(text, voice)
+
+    async def generate():
+        async for chunk in communicate.stream():
+            if chunk["type"] == "audio":
+                yield chunk["data"]
+
+    return StreamingResponse(generate(), media_type="audio/mpeg")
+
 # ============================================================================
 # Main Execution
 # ============================================================================