Spaces:
Sleeping
Sleeping
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
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 |
-
|
| 376 |
-
|
| 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 |
|