Update TextGen/router.py

TextGen/router.py

@@ -8,6 +8,7 @@ from fastapi.middleware.cors import CORSMiddleware
 from langchain.chains import LLMChain
 from langchain.prompts import PromptTemplate
 from TextGen.suno import custom_generate_audio, get_audio_information
+from TextGen.coqui import predict
 from langchain_google_genai import (
     ChatGoogleGenerativeAI,
     HarmBlockThreshold,
@@ -156,7 +157,33 @@ async def generate_wav(message: VoiceMessage):
 
     except Exception as e:
         raise HTTPException(status_code=500, detail=str(e))
+@app.post("/generate_voice")
+async def generate_voice(message: VoiceMessage):
+    try:
+        voice = determine_vocie_from_npc(message.npc, message.genre)
+        audio_file_pth = handle_file(voice)
 
+        # Generator function to yield audio chunks
+        async def audio_stream():
+            result = predict(
+                prompt=message.input,
+                language=message.language,
+                audio_file_pth=audio_file_pth,
+                mic_file_path=None,
+                use_mic=False,
+                voice_cleanup=False,
+                no_lang_auto_detect=False,
+                agree=True,
+            )
+            for sampling_rate, audio_chunk in result:
+                yield audio_chunk.tobytes()
+                await asyncio.sleep(0)  # Yield control to the event loop
+
+        # Return the generated audio as a streaming response
+        return StreamingResponse(audio_stream(), media_type="audio/wav")
+
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=str(e))
 
 @app.get("/generate_song")
 async def generate_song(text: str):