Update app.py

app.py

@@ -462,7 +462,7 @@ def main():
             with gr.Row(elem_id="header"):
                 gr.Markdown(
                     """
-                    # 🎵 Open SUNO: AI Music Generator
+                    # 🎵 AI Song Creator: Transform Your Lyrics into Complete Songs
                     ### Create complete songs from your lyrics in multiple languages
                     """,
                     elem_id="title"
@@ -526,10 +526,12 @@ def main():
                     with gr.Row():
                         history_container = gr.HTML("""
                             <div id="history-container" style="width: 100%; padding: 10px;">
-                                <h3>Generation History</h3>
+                                <h3>🎵 Generation History</h3>
                                 <div id="history-list"></div>
                             </div>
                         """)
+                    
+                    # 히스토리 상태 저장
                     history_state = gr.State([])
 
             # 예제 섹션
@@ -588,6 +590,10 @@ Don't let this moment fade, hold me close tonight
                     """
                 )
 
+        # 숨겨진 상태 변수들
+        num_segments = gr.State(value=2)  # 기본값 2
+        max_new_tokens = gr.State(value=4000)  # 기본값 4000
+
         # 시스템 초기화
         initialize_system()
 
@@ -634,10 +640,10 @@ Don't let this moment fade, hold me close tonight
             outputs=[duration_info, sections_info]
         )
         
-        def generate_with_progress(genre, lyrics, num_segments, max_tokens, history):
+        def generate_with_progress(genre, lyrics, segments, tokens, history):
             progress.update(value="🎵 Starting generation...")
             try:
-                result = infer(genre, lyrics, num_segments, max_tokens)
+                result = infer(genre, lyrics, segments, tokens)
                 if result:
                     progress.update(value="✅ Generation complete!")
                     new_history, history_html = update_history(result, genre, lyrics, history)