Update app.py

app.py

@@ -51,58 +51,53 @@ def analyze_lyrics(lyrics, repeat_chorus=2):
         'total_lines': len(lines)
     }
     
+    # 기본값 설정
     current_section = None
+    last_section_start = 0
+
     section_lines = {
         'verse': [],
         'chorus': [],
         'bridge': []
     }
-    last_section = None
-
-    for i, line in enumerate(lines):
-        if '[verse]' in line.lower() or '[chorus]' in line.lower() or '[bridge]' in line.lower():
-            last_section = i
 
+    # 루프를 돌며 태그가 등장하면 해당 태그 전까지의 라인을 이전 섹션에 저장
     for i, line in enumerate(lines):
         lower_line = line.lower()
-        
+
         if '[verse]' in lower_line:
-            if current_section:  
+            if current_section is not None:
                 section_lines[current_section].extend(lines[last_section_start:i])
             current_section = 'verse'
             sections['verse'] += 1
             last_section_start = i + 1
-            continue
+
         elif '[chorus]' in lower_line:
-            if current_section:
+            if current_section is not None:
                 section_lines[current_section].extend(lines[last_section_start:i])
             current_section = 'chorus'
             sections['chorus'] += 1
             last_section_start = i + 1
-            continue
+
         elif '[bridge]' in lower_line:
-            if current_section:
+            if current_section is not None:
                 section_lines[current_section].extend(lines[last_section_start:i])
             current_section = 'bridge'
             sections['bridge'] += 1
             last_section_start = i + 1
-            continue
 
-    if current_section and 'last_section_start' in locals() and last_section_start < len(lines):
+    # 마지막 남은 라인들을 현재 섹션에 추가
+    if current_section is not None:
         section_lines[current_section].extend(lines[last_section_start:])
 
+    # 코러스 반복 등 추가 처리
     if sections['chorus'] > 0 and repeat_chorus > 1:
-        original_chorus = section_lines['chorus'][:]
+        original_chorus = list(section_lines['chorus'])
         for _ in range(repeat_chorus - 1):
             section_lines['chorus'].extend(original_chorus)
 
-    logging.info(
-        f"Section line counts - Verse: {len(section_lines['verse'])}, "
-        f"Chorus: {len(section_lines['chorus'])}, "
-        f"Bridge: {len(section_lines['bridge'])}"
-    )
+    return sections, sum(sections[sec] for sec in ('verse','chorus','bridge')), len(lines), section_lines
 
-    return sections, (sections['verse'] + sections['chorus'] + sections['bridge']), len(lines), section_lines
 
 def calculate_generation_params(lyrics):
     sections, total_sections, total_lines, section_lines = analyze_lyrics(lyrics)