Update ui/components.py

ui/components.py

@@ -452,14 +452,16 @@ def create_text2music_ui(
                         choices=list(QUALITY_PRESETS.keys()),
                         value="Standard",
                         label="품질 프리셋",
-                        scale=2
+                        scale=2,
+                        interactive=True
                     )
                     multi_seed_mode = gr.Dropdown(
                         choices=list(MULTI_SEED_OPTIONS.keys()),
                         value="Single",
                         label="다중 생성 모드",
                         scale=2,
-                        info="여러 번 생성하여 최고 품질 선택"
+                        info="여러 번 생성하여 최고 품질 선택",
+                        interactive=True
                     )
                 
                 preset_description = gr.Textbox(
@@ -519,8 +521,8 @@ def create_text2music_ui(
 
             def toggle_ref_audio_visibility(is_checked):
                 return (
-                    gr.update(visible=is_checked, elem_id="ref_audio_input"),
-                    gr.update(visible=is_checked, elem_id="ref_audio_strength"),
+                    gr.update(visible=is_checked),
+                    gr.update(visible=is_checked),
                 )
 
             audio2audio_enable.change(
@@ -540,14 +542,14 @@ def create_text2music_ui(
                             value="Custom",
                             label="장르 프리셋",
                             scale=1,
-                            interactive=True  # 명시적으로 interactive 추가
+                            interactive=True
                         )
                         song_style = gr.Dropdown(
                             choices=list(SONG_STYLES.keys()),
                             value="듀엣 (남녀 혼성)",
                             label="곡 스타일",
                             scale=1,
-                            interactive=True  # 명시적으로 interactive 추가
+                            interactive=True
                         )
                         enable_smart_enhancement = gr.Checkbox(
                             label="스마트 향상",
@@ -562,6 +564,7 @@ def create_text2music_ui(
                         max_lines=4,
                         value=TAG_DEFAULT,
                         placeholder="콤마로 구분된 태그들...",
+                        interactive=True
                     )
 
             with gr.Group():
@@ -572,7 +575,8 @@ def create_text2music_ui(
                     lyric_prompt = gr.Textbox(
                         label="작사 주제",
                         placeholder="예: 첫사랑의 설렘, 이별의 아픔, 희망찬 내일...",
-                        scale=3
+                        scale=3,
+                        interactive=True
                     )
                     generate_lyrics_btn = gr.Button("🤖 AI 작사", variant="secondary", scale=1)
                 
@@ -581,7 +585,8 @@ def create_text2music_ui(
                     label="Lyrics",
                     max_lines=13,
                     value=LYRIC_DEFAULT,
-                    placeholder="가사를 입력하세요. [verse], [chorus] 등의 구조 태그 사용을 권장합니다."
+                    placeholder="가사를 입력하세요. [verse], [chorus] 등의 구조 태그 사용을 권장합니다.",
+                    interactive=True
                 )
 
             with gr.Accordion("Basic Settings", open=False):
@@ -1222,9 +1227,8 @@ def create_text2music_ui(
                     outputs=extend_outputs + [extend_input_params_json],
                 )
 
-    # ===== 모든 UI 요소가 정의된 후 이벤트 핸들러 설정 =====
+    # ===== 이벤트 핸들러 함수 정의 =====
     
-    # 태그 업데이트 함수
     def update_tags_from_preset(preset_name, current_tags, song_style):
         """장르 프리셋 선택시 태그 업데이트"""
         print(f"🎵 update_tags_from_preset called: preset={preset_name}, style={song_style}")
@@ -1244,18 +1248,19 @@ def create_text2music_ui(
         print(f"✅ Updated tags: {result[:50]}...")
         return result
     
-    # AI 작사 버튼 이벤트
-    def generate_ai_lyrics_handler(lyric_prompt, genre_preset, song_style):
+    def generate_ai_lyrics_wrapper(lyric_prompt, genre_preset, song_style):
+        """AI 작사 버튼 이벤트 핸들러"""
         print(f"🎤 AI 작사 버튼 클릭: prompt='{lyric_prompt}'")
         if not lyric_prompt or lyric_prompt.strip() == "":
             print("⚠️ 빈 프롬프트")
-            return lyrics.value  # 현재 가사 유지
+            gr.Warning("작사 주제를 입력해주세요!")
+            return gr.update()  # 현재 가사 유지
         
         generated = generate_lyrics_with_ai(lyric_prompt, genre_preset, song_style)
         return generated
     
-    # 랜덤 데이터 생성 함수
     def generate_random_music_data(genre_preset, song_style):
+        """랜덤 데이터 생성"""
         print("🎲 Random 버튼 클릭")
         # 랜덤 장르 선택
         if genre_preset == "Custom":
@@ -1307,7 +1312,6 @@ def create_text2music_ui(
             None,      # ref_audio_input
         )
     
-    # 실시간 프리뷰 기능
     def generate_preview(prompt, lyrics, genre_preset, song_style):
         """10초 프리뷰 생성"""
         print("🎵 Preview 버튼 클릭")
@@ -1323,26 +1327,25 @@ def create_text2music_ui(
         enhanced_prompt = enhance_prompt_with_genre(prompt, genre_preset, song_style)
         
         try:
-            # 실제 구현에서는 빠른 생성 모드 사용
             result = enhanced_process_func(
                 preview_params["audio_duration"],
                 enhanced_prompt,
-                lyrics[:200],  # 가사 일부만 사용
+                lyrics[:200],
                 preview_params["infer_step"],
                 preview_params["guidance_scale"],
                 preview_params["scheduler_type"],
                 preview_params["cfg_type"],
                 preview_params["omega_scale"],
-                None,  # manual_seeds
-                0.5,   # guidance_interval
-                0.0,   # guidance_interval_decay
-                3.0,   # min_guidance_scale
-                True,  # use_erg_tag
-                False, # use_erg_lyric
-                True,  # use_erg_diffusion
-                None,  # oss_steps
-                0.0,   # guidance_scale_text
-                0.0,   # guidance_scale_lyric
+                None,
+                0.5,
+                0.0,
+                3.0,
+                True,
+                False,
+                True,
+                None,
+                0.0,
+                0.0,
                 multi_seed_mode="Single",
                 song_style=song_style
             )
@@ -1350,39 +1353,37 @@ def create_text2music_ui(
         except Exception as e:
             return f"프리뷰 생성 실패: {str(e)}"
     
-    # 이벤트 핸들러 연결
+    # ===== 이벤트 핸들러 연결 =====
     print("🔗 이벤트 핸들러 연결 중...")
     
     # 장르 프리셋 변경 이벤트
-    genre_preset.change(
+    genre_preset.select(
         fn=update_tags_from_preset,
         inputs=[genre_preset, prompt, song_style],
         outputs=[prompt]
     )
     
     # 곡 스타일 변경 이벤트
-    song_style.change(
+    song_style.select(
         fn=update_tags_from_preset,
         inputs=[genre_preset, prompt, song_style],
         outputs=[prompt]
     )
     
     # 품질 프리셋 변경 이벤트
-    quality_preset.change(
+    quality_preset.select(
         fn=lambda x: QUALITY_PRESETS.get(x, {}).get("description", ""),
         inputs=[quality_preset],
         outputs=[preset_description]
-    )
-
-    quality_preset.change(
-        fn=update_quality_preset,  # 전역 함수 직접 사용
+    ).then(
+        fn=update_quality_preset,
         inputs=[quality_preset],
         outputs=[infer_step, guidance_scale, scheduler_type, omega_scale, use_erg_diffusion, use_erg_tag]
     )
     
     # AI 작사 버튼 이벤트
     generate_lyrics_btn.click(
-        fn=generate_ai_lyrics_handler,
+        fn=generate_ai_lyrics_wrapper,
         inputs=[lyric_prompt, genre_preset, song_style],
         outputs=[lyrics]
     )