Update TextGen/router.py

TextGen/router.py

@@ -45,7 +45,9 @@ class CreateNewMusic(BaseModel):
         ..., description="tags to describe the new music"
     )
 
-
+class SongRequest(BaseModel):
+    prompt: str | None  = None
+    tags: List[str] | None = None
 
 class Message(BaseModel):
     npc: str | None  = None
@@ -212,9 +214,9 @@ def generate_voice_coqui(message: VoiceMessage = None):
 
     return StreamingResponse(audio_stream(),media_type="audio/mpeg")   
 @app.get("/generate_song")
-async def generate_song(text: str):
+async def generate_song(text: SongRequest):
     song_lyrics=generate_lyrics({
-        "prompt": f"{text}",
+        "prompt": f"{text.prompt}",
         })
     data = custom_generate_audio({
         "prompt": song_lyrics['text'],