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