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):