futurespyhi commited on
Commit
7b364b8
·
1 Parent(s): 1fba260

Fix lyrics generation function call and response formatting

Browse files

- Correct parameter order for generate_structured_lyrics() call in respond() function
- Add format_lyrics import to properly format SongStructure objects for display
- Fix 'takes 4 positional arguments but 5 were given' error in chat interface

Files changed (1) hide show
  1. app.py +7 -5
app.py CHANGED
@@ -198,7 +198,7 @@ if project_root not in sys.path:
198
 
199
  from tools.groq_client import client as groq_client
200
  from openai import OpenAI
201
- from tools.generate_lyrics import generate_structured_lyrics, format_lyrics_for_yue
202
 
203
  # Apply patches after all imports are set up
204
  patch_applied = apply_transformers_patch()
@@ -372,14 +372,16 @@ def respond(message, state):
372
  state.conversation.append({"role": "user", "content": message})
373
 
374
  # Generate response using your existing lyrics generation logic
375
- response = generate_structured_lyrics(
376
- message,
377
  state.genre,
378
  state.mood,
379
- state.theme,
380
- state.conversation
381
  )
382
 
 
 
 
383
  # Add assistant response
384
  state.conversation.append({"role": "assistant", "content": response})
385
 
 
198
 
199
  from tools.groq_client import client as groq_client
200
  from openai import OpenAI
201
+ from tools.generate_lyrics import generate_structured_lyrics, format_lyrics_for_yue, format_lyrics
202
 
203
  # Apply patches after all imports are set up
204
  patch_applied = apply_transformers_patch()
 
372
  state.conversation.append({"role": "user", "content": message})
373
 
374
  # Generate response using your existing lyrics generation logic
375
+ song_structure = generate_structured_lyrics(
376
+ state.conversation,
377
  state.genre,
378
  state.mood,
379
+ state.theme
 
380
  )
381
 
382
+ # Format the structured lyrics for display
383
+ response = format_lyrics(song_structure)
384
+
385
  # Add assistant response
386
  state.conversation.append({"role": "assistant", "content": response})
387