Update app.py

app.py

@@ -41,27 +41,42 @@ def optimize_gpu_settings():
         if 'L40S' in torch.cuda.get_device_name(0):
             torch.cuda.set_per_process_memory_fraction(0.95)
 
+import logging
+
 def analyze_lyrics(lyrics, repeat_chorus=2):
-    lines = [line.strip() for line in lyrics.split('\n') if line.strip()]
+    # 먼저 라인별로 분리하고, 공백 줄 제거
+    lines = [line.strip() for line in lyrics.split('\n')]
+    lines = [line for line in lines if line]
     
+    # 만약 전체가 비어있다면 강제로 '.' 한 줄 추가
+    if not lines:
+        lines = ['.']
+    else:
+        # 마지막 줄이 [verse], [chorus], [bridge] 태그로만 끝나면
+        # 임의로 '.' 한 줄을 추가하여 실제 가사 라인이 되도록 처리
+        last_line_lower = lines[-1].lower()
+        if last_line_lower in ['[verse]', '[chorus]', '[bridge]']:
+            lines.append('.')
+
+    # 기본 섹션 정보
     sections = {
         'verse': 0,
         'chorus': 0,
         'bridge': 0,
         'total_lines': len(lines)
     }
-    
-    # 기본값 설정
-    current_section = None
-    last_section_start = 0
 
+    # 섹션 라인들을 담을 딕셔너리
     section_lines = {
         'verse': [],
         'chorus': [],
         'bridge': []
     }
 
-    # 루프를 돌며 태그가 등장하면 해당 태그 전까지의 라인을 이전 섹션에 저장
+    current_section = None
+    last_section_start = 0
+
+    # [verse], [chorus], [bridge] 태그가 나오면 섹션을 구분하여 라인을 저장
     for i, line in enumerate(lines):
         lower_line = line.lower()
 
@@ -86,17 +101,26 @@ def analyze_lyrics(lyrics, repeat_chorus=2):
             sections['bridge'] += 1
             last_section_start = i + 1
 
-    # 마지막 남은 라인들을 현재 섹션에 추가
-    if current_section is not None:
+    # 마지막 섹션에 남아 있는 라인들을 추가
+    if current_section is not None and last_section_start < len(lines):
         section_lines[current_section].extend(lines[last_section_start:])
 
-    # 코러스 반복 등 추가 처리
+    # 코러스 반복 처리
     if sections['chorus'] > 0 and repeat_chorus > 1:
         original_chorus = list(section_lines['chorus'])
         for _ in range(repeat_chorus - 1):
             section_lines['chorus'].extend(original_chorus)
 
-    return sections, sum(sections[sec] for sec in ('verse','chorus','bridge')), len(lines), section_lines
+    # 섹션별 라인수 로깅
+    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, (sections['verse'] + sections['chorus'] + sections['bridge']), len(lines), section_lines
+
 
 
 def calculate_generation_params(lyrics):