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'],