Update app.py

app.py

@@ -423,62 +423,151 @@ def optimize_model_selection(lyrics, genre):
     return model_path, model_config[model_path], params
 
 def main():
-    with gr.Blocks() as demo:
-        with gr.Column():
-            gr.Markdown("# Open SUNO: Full-Song Generation (Multi-Language Support)")
-            
+    # 테마 설정
+    theme = gr.themes.Soft(
+        primary_hue="indigo",
+        secondary_hue="purple",
+        neutral_hue="slate",
+        font=["Arial", "sans-serif"]
+    )
+    
+    # CSS 스타일 정의
+    custom_css = """
+    #main-container {
+        max-width: 1200px;
+        margin: auto;
+        padding: 20px;
+    }
+    #header {
+        text-align: center;
+        margin-bottom: 30px;
+        background: linear-gradient(135deg, #6366f1, #a855f7);
+        padding: 20px;
+        border-radius: 15px;
+        color: white;
+    }
+    .input-section {
+        background: #f8fafc;
+        padding: 20px;
+        border-radius: 15px;
+        margin-bottom: 20px;
+        box-shadow: 0 4px 6px -1px rgb(0 0 0 / 0.1);
+    }
+    .output-section {
+        background: #f0f9ff;
+        padding: 20px;
+        border-radius: 15px;
+        margin-bottom: 20px;
+        box-shadow: 0 4px 6px -1px rgb(0 0 0 / 0.1);
+    }
+    #generate-btn {
+        background: linear-gradient(135deg, #6366f1, #a855f7);
+        border: none;
+        padding: 15px 30px;
+        border-radius: 10px;
+        color: white;
+        font-weight: bold;
+        cursor: pointer;
+        transition: all 0.3s ease;
+    }
+    #generate-btn:hover {
+        transform: translateY(-2px);
+        box-shadow: 0 4px 12px rgba(0,0,0,0.15);
+    }
+    .info-box {
+        background: #fff;
+        padding: 15px;
+        border-radius: 10px;
+        border: 1px solid #e2e8f0;
+        margin: 10px 0;
+    }
+    .examples-section {
+        background: #f8fafc;
+        padding: 20px;
+        border-radius: 15px;
+        margin-top: 20px;
+    }
+    """
+
+    with gr.Blocks(theme=theme, css=custom_css) as demo:
+        with gr.Column(elem_id="main-container"):
+            # 헤더 섹션
+            with gr.Row(elem_id="header"):
+                gr.Markdown(
+                    """
+                    # 🎵 AI Song Creator
+                    ### Transform Your Lyrics into Complete Songs with Music
+                    Create professional songs from your lyrics in multiple languages
+                    """
+                )
+
+            # 메인 컨텐츠
             with gr.Row():
-                with gr.Column():
+                # 입력 섹션
+                with gr.Column(scale=1, elem_classes="input-section"):
+                    gr.Markdown("### 📝 Input Your Song Details")
                     genre_txt = gr.Textbox(
-                        label="Genre",
-                        placeholder="Enter music genre and style descriptions..."
+                        label="🎸 Music Genre & Style",
+                        placeholder="e.g., K-pop bright energetic synth dance electronic...",
+                        elem_id="genre-input"
                     )
                     lyrics_txt = gr.Textbox(
-                        label="Lyrics (Supports English, Korean, Japanese, Chinese)",
-                        placeholder="Enter song lyrics with [verse], [chorus], [bridge] tags...",
-                        lines=10
+                        label="📝 Lyrics",
+                        placeholder="Enter lyrics with section tags: [verse], [chorus], [bridge]...",
+                        lines=10,
+                        elem_id="lyrics-input"
                     )
                     
-                with gr.Column():
-                    num_segments = gr.Number(
-                        label="Number of Song Segments (Auto-adjusted based on lyrics)",
-                        value=2,
-                        minimum=1,
-                        maximum=4,
-                        step=1,
-                        interactive=False
-                    )
-                    max_new_tokens = gr.Slider(
-                        label="Max New Tokens (Auto-adjusted based on lyrics)",
-                        minimum=500,
-                        maximum=32000,
-                        step=500,
-                        value=4000,
-                        interactive=False
-                    )
+                    # 정보 표시 섹션
                     with gr.Row():
-                        duration_info = gr.Label(label="Estimated Duration")
-                        sections_info = gr.Label(label="Section Information")
-                    submit_btn = gr.Button("Generate Music", variant="primary")
-                    music_out = gr.Audio(label="Generated Audio")
-
-            gr.Examples(
-                examples=[
-                    [
-                        "female blues airy vocal bright vocal piano sad romantic guitar jazz",
-                        """[verse]
-In the quiet of the evening, shadows start to fall
-Whispers of the night wind echo through the hall
-Lost within the silence, I hear your gentle voice
-Guiding me back homeward, making my heart rejoice
-
-[chorus]
-Don't let this moment fade, hold me close tonight
-With you here beside me, everything's alright
-Can't imagine life alone, don't want to let you go
-Stay with me forever, let our love just flow
+                        with gr.Column(scale=1):
+                            duration_info = gr.Label(
+                                label="⏱️ Estimated Duration",
+                                elem_classes="info-box"
+                            )
+                        with gr.Column(scale=1):
+                            sections_info = gr.Label(
+                                label="📊 Section Analysis",
+                                elem_classes="info-box"
+                            )
+                    
+                    submit_btn = gr.Button(
+                        "🎼 Generate Music",
+                        variant="primary",
+                        elem_id="generate-btn"
+                    )
 
-[verse]
+                # 출력 섹션
+                with gr.Column(scale=1, elem_classes="output-section"):
+                    gr.Markdown("### 🎵 Generated Music")
+                    music_out = gr.Audio(
+                        label="Generated Song",
+                        elem_id="music-output"
+                    )
+                    
+                    # 진행 상태
+                    with gr.Box():
+                        gr.Markdown("### 🔄 Generation Status")
+                        num_segments = gr.Number(
+                            label="Song Segments",
+                            value=2,
+                            interactive=False,
+                            visible=False
+                        )
+                        max_new_tokens = gr.Number(
+                            label="Tokens",
+                            value=4000,
+                            interactive=False,
+                            visible=False
+                        )
+
+            # 예제 섹션
+            with gr.Accordion("📖 Examples", open=False, elem_classes="examples-section"):
+                gr.Examples(
+                    examples=[
+                        [
+                            "female blues airy vocal bright vocal piano sad romantic guitar jazz",
+                            """[verse]
 In the quiet of the evening, shadows start to fall
 Whispers of the night wind echo through the hall
 Lost within the silence, I hear your gentle voice
@@ -489,10 +578,10 @@ Don't let this moment fade, hold me close tonight
 With you here beside me, everything's alright
 Can't imagine life alone, don't want to let you go
 Stay with me forever, let our love just flow"""
-                    ],
-                    [
-                        "K-pop bright energetic synth dance electronic",
-                        """[verse]
+                        ],
+                        [
+                            "K-pop bright energetic synth dance electronic",
+                            """[verse]
 언젠가 마주한 눈빛 속에서
 
 [chorus]
@@ -502,26 +591,36 @@ Stay with me forever, let our love just flow"""
 어두운 밤을 지날 때마다
 
 [chorus]
-다시 한 번 내게 말해줘
-"""
-                    ]
-                ],
-                inputs=[genre_txt, lyrics_txt]
-            )
-
-        initialize_system()
-
-        def update_info(lyrics):
-            if not lyrics:
-                return "No lyrics entered", "No sections detected"
-            params = calculate_generation_params(lyrics)
-            duration = params['estimated_duration']
-            sections = params['sections']
-            return (
-                f"Estimated duration: {duration:.1f} seconds",
-                f"Verses: {sections['verse']}, Chorus: {sections['chorus']} (Expected full length including chorus)"
-            )
-
+다시 한 번 내게 말해줘"""
+                        ]
+                    ],
+                    inputs=[genre_txt, lyrics_txt]
+                )
+
+            # 도움말 섹션
+            with gr.Accordion("ℹ️ Help & Information", open=False):
+                gr.Markdown(
+                    """
+                    ### 🎵 How to Use
+                    1. **Enter Genre & Style**: Describe the musical style you want
+                    2. **Input Lyrics**: Write your lyrics using section tags
+                    3. **Generate**: Click the Generate button and wait for your music!
+
+                    ### 🌏 Supported Languages
+                    - English
+                    - Korean (한국어)
+                    - Japanese (日本語)
+                    - Chinese (中文)
+
+                    ### ⚡ Tips for Best Results
+                    - Be specific with genre descriptions
+                    - Include emotion and instrument preferences
+                    - Properly tag your lyrics sections
+                    - Include both verse and chorus sections
+                    """
+                )
+
+        # 이벤트 핸들러 설정
         lyrics_txt.change(
             fn=update_info,
             inputs=[lyrics_txt],