Update app.py

app.py

@@ -88,16 +88,19 @@ def calculate_generation_params(lyrics):
     }
     
     # 각 섹션별 예상 시간 계산
-    section_durations = {
-        'verse': section_lines['verse'] * time_per_line['verse'],
-        'chorus': section_lines['chorus'] * time_per_line['chorus'],
-        'bridge': section_lines['bridge'] * time_per_line['bridge']
-    }
+    section_durations = {}
+    for section_type in ['verse', 'chorus', 'bridge']:
+        # 각 섹션의 라인 수에 해당 섹션의 시간을 곱함
+        if isinstance(section_lines[section_type], list):
+            section_durations[section_type] = len(section_lines[section_type]) * time_per_line[section_type]
+        else:
+            section_durations[section_type] = section_lines[section_type] * time_per_line[section_type]
     
-    total_duration = sum(section_durations.values())
+    # 전체 시간 계산
+    total_duration = sum(duration for duration in section_durations.values())
     total_duration = max(60, total_duration)  # 최소 60초
     
-    # 토큰 계산 (더 보수적인 값 사용)
+    # 토큰 계산
     base_tokens = 3000  # 기본 토큰 수
     tokens_per_line = 200  # 줄당 토큰 수