futurespyhi commited on
Commit
7e55e4f
·
1 Parent(s): 51c37c6

update lyrics representing logic

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -848,8 +848,9 @@ def response_audio(state: AppState, audio: tuple, genre_value, mood_value, theme
848
 
849
  state.conversation.append({"role": "assistant", "content": assistant_message})
850
 
851
- # Update lyrics from conversation
852
- state.lyrics = extract_lyrics_from_conversation(state.conversation)
 
853
 
854
  os.remove(file_name)
855
 
 
848
 
849
  state.conversation.append({"role": "assistant", "content": assistant_message})
850
 
851
+ # Update lyrics using same logic as text input for consistency
852
+ if any(marker in assistant_message.lower() for marker in ["[verse", "[chorus", "[bridge", "**verse", "**chorus", "sectiontype.verse", "verse:"]):
853
+ state.lyrics = assistant_message
854
 
855
  os.remove(file_name)
856