Update TextGen/router.py

TextGen/router.py

@@ -26,7 +26,7 @@ Eleven_client = ElevenLabs(
 )
 
 
-
+Last_message=None
 class PlayLastMusic(BaseModel):
     '''plays the lastest created music '''
     Desicion: str = Field(
@@ -167,19 +167,23 @@ async def generate_wav(message: VoiceMessage):
     except Exception as e:
         raise HTTPException(status_code=500, detail=str(e))
 
-def audio_stream(message):
-
-    # Generate the audio stream from ElevenLabs
-    for chunk in Eleven_client.generate(text=message.input, stream=True):
-        yield chunk
 
+@app.get("/generate_voice", response_class=StreamingResponse)
 @app.post("/generate_voice", response_class=StreamingResponse)
 def generate_voice(message: VoiceMessage = None):
     print("message",message)
+    if message is None:
+        message=Last_message
+    else:
+        Last_message=message
+    def audio_stream():
 
+        # Generate the audio stream from ElevenLabs
+        for chunk in Eleven_client.generate(text=message.input, stream=True):
+            yield chunk
    
 
-    return StreamingResponse(audio_stream(message), media_type="audio/mpeg")
+    return StreamingResponse(audio_stream(), media_type="audio/mpeg")
     
 @app.get("/generate_song")
 async def generate_song(text: str):