Update TextGen/router.py

TextGen/router.py

@@ -169,9 +169,9 @@ async def generate_wav(message: VoiceMessage):
 
 
 
-@app.get("/generate_voice", response_class=StreamingResponse)
-@app.post("/generate_voice", response_class=StreamingResponse)
-def generate_voice(message: VoiceMessage = None):
+@app.get("/generate_voice_eleven", response_class=StreamingResponse)
+@app.post("/generate_voice_eleven", response_class=StreamingResponse)
+def generate_voice_eleven(message: VoiceMessage = None):
  global Last_message # Declare Last_message as global
  if message is None:
  message = Last_message
@@ -184,25 +184,44 @@ def generate_voice(message: VoiceMessage = None):
  yield chunk
 
  return StreamingResponse(audio_stream(), media_type="audio/mpeg")
- 
+@app.get("/generate_voice_coqui", response_class=StreamingResponse)
+@app.post("/generate_voice_coqui", response_class=StreamingResponse)
+def generate_voice_coqui(message: VoiceMessage = None):
+ global Last_message 
+ if message is None:
+ message = Last_message
+ else:
+ Last_message = message
+
+ def audio_stream():
+ voice = determine_vocie_from_npc(message.npc, message.genre)
+ result = predict(
+ prompt=message.input,
+ language=message.language,
+ audio_file_pth=voice,
+ mic_file_path=None,
+ use_mic=False,
+ voice_cleanup=False,
+ no_lang_auto_detect=False,
+ agree=True,
+ )
+ # Generate the audio stream from ElevenLabs
+ for chunk in result:
+ print("received : ",chunk)
+ yield chunk
+
+ return StreamingResponse(audio_stream(),media_type="audio/mpeg") 
 @app.get("/generate_song")
 async def generate_song(text: str):
- try:
- data = custom_generate_audio({
- "prompt": f"{text}",
- "make_instrumental": False,
- "wait_audio": False
+ song_lyrics=generate_lyrics({
+ "prompt": f"{text}",
  })
- ids = f"{data[0]['id']},{data[1]['id']}"
- print(f"ids: {ids}")
-
- for _ in range(60):
- data = get_audio_information(ids)
- if data[0]["status"] == 'streaming':
- print(f"{data[0]['id']} ==> {data[0]['audio_url']}")
- print(f"{data[1]['id']} ==> {data[1]['audio_url']}")
- break
- # sleep 5s
- time.sleep(5)
- except:
- print("Error")
+ data = custom_generate_audio({
+ "prompt": song_lyrics['text'],
+ "tags": "male bard",
+ "title":"Everchangin_Quest_song",
+ "wait_audio":True,
+ 
+ })
+ infos=get_audio_information(f"{data[0]['id']},{data[1]['id']}")
+ return infos