Update TextGen/router.py

TextGen/router.py

@@ -166,13 +166,23 @@ async def generate_wav(message: VoiceMessage):
 
  except Exception as e:
  raise HTTPException(status_code=500, detail=str(e))
-@app.post("/generate_voice")
-def generate_voice(message: VoiceMessage):
- def audio_stream():
- for chunk in Eleven_client.generate(text=message.input, stream=True):
- yield chunk
+ 
+@app.post("/generate_voice", response_class=StreamingResponse)
+@app.get("/generate_voice", response_class=StreamingResponse)
+def generate_voice(message: VoiceMessage = None):
+ if message is None:
+ # If GET request without message input, return a default message
+ message = VoiceMessage(input="Default test message")
 
- return StreamingResponse(audio_stream(), media_type="audio/wav")
+ def audio_stream():
+ try:
+ # Generate the audio stream from ElevenLabs
+ for chunk in client.generate(text=message.input, stream=True):
+ yield chunk
+ except Exception as e:
+ raise HTTPException(status_code=500, detail=str(e))
+
+ return StreamingResponse(audio_stream(), media_type="audio/mpeg")
  
 @app.get("/generate_song")
 async def generate_song(text: str):