Update lyrics with improved format recognition - extract only segments

.gitignore

@@ -110,6 +110,7 @@ packages.txt.backup
 requirements.txt.backup
 download_model.py.backup
 decode_vocode.txt
+backup/
 
 # test files
 tests/

app.py

@@ -558,9 +558,9 @@ def respond(message, state):
         # Add assistant response
         state.conversation.append({"role": "assistant", "content": response})
         
-        # Update lyrics with improved format recognition
+        # Update lyrics with improved format recognition - extract only segments
         if any(marker in response.lower() for marker in ["[verse", "[chorus", "[bridge", "**verse", "**chorus", "sectiontype.verse", "verse:"]):
-            state.lyrics = response
+            state.lyrics = extract_lyrics_segments_only(response)
         
         # Format conversation for display
         return "", [{"role": msg["role"], "content": msg["content"]} for msg in state.conversation], state
@@ -866,9 +866,9 @@ def response_audio(state: AppState, audio: tuple, genre_value, mood_value, theme
 
         state.conversation.append({"role": "assistant", "content": assistant_message})
 
-        # Update lyrics using same logic as text input for consistency
+        # Update lyrics using same logic as text input for consistency - extract only segments
         if any(marker in assistant_message.lower() for marker in ["[verse", "[chorus", "[bridge", "**verse", "**chorus", "sectiontype.verse", "verse:"]):
-            state.lyrics = assistant_message
+            state.lyrics = extract_lyrics_segments_only(assistant_message)
 
         os.remove(file_name)
 
@@ -879,6 +879,44 @@ def response_audio(state: AppState, audio: tuple, genre_value, mood_value, theme
     
     return state, conversation_display, state.lyrics
 
+def extract_lyrics_segments_only(content):
+    """
+    Extract only the lyrics segments (VERSE, CHORUS, etc.) from AI response,
+    removing any AI commentary or explanation text.
+    """
+    import re
+
+    if not content:
+        return ""
+
+    lines = content.split('\n')
+    lyrics_lines = []
+    in_lyrics_section = False
+
+    for line in lines:
+        line = line.strip()
+
+        # Check if this line is a section header (VERSE, CHORUS, etc.)
+        if re.match(r'^\*\*(VERSE|CHORUS|BRIDGE|OUTRO).*\*\*$', line) or re.match(r'^\[(VERSE|CHORUS|BRIDGE|OUTRO).*\]$', line):
+            in_lyrics_section = True
+            lyrics_lines.append(line)
+            continue
+
+        # If we're in a lyrics section
+        if in_lyrics_section:
+            # Stop if we hit AI commentary
+            if line and any(phrase in line.lower() for phrase in [
+                'how do you like', 'would you like', 'let me know',
+                'what do you think', 'any changes', 'take a look',
+                'here are the lyrics', 'i\'ve created', 'feel free to'
+            ]):
+                break
+
+            # Add lyrics line (including empty lines for formatting)
+            lyrics_lines.append(line)
+
+    return '\n'.join(lyrics_lines).strip()
+
 def extract_lyrics_from_conversation(conversation):
     """
     Extract lyrics from conversation history with cross-platform compatibility.
@@ -889,11 +927,11 @@ def extract_lyrics_from_conversation(conversation):
             content_lower = message["content"].lower()
             # 先尝试严格匹配（保持原逻辑）
             if "verse" in content_lower and "chorus" in content_lower:
-                lyrics = message["content"]
+                lyrics = extract_lyrics_segments_only(message["content"])
                 break
             # 如果没找到，再用宽泛匹配（兼容性备选）
             elif any(marker in content_lower for marker in ["[verse", "[chorus", "**verse", "**chorus"]):
-                lyrics = message["content"]
+                lyrics = extract_lyrics_segments_only(message["content"])
                 break
     return lyrics