Update ui/components.py

ui/components.py

@@ -1227,238 +1227,139 @@ def create_text2music_ui(
                     outputs=extend_outputs + [extend_input_params_json],
                 )
 
-    # ===== 이벤트 핸들러 함수 정의 =====
+    # ===== 간단하고 직접적인 이벤트 핸들러 =====
+    print("🔗 이벤트 핸들러 연결 중...")
     
-    def update_tags_from_preset(preset_name, current_tags, song_style):
-        """장르 프리셋 선택시 태그 업데이트"""
-        print(f"🎵 update_tags_from_preset called: preset={preset_name}, style={song_style}")
-        
-        if preset_name == "Custom":
-            return current_tags
-        
-        base_tags = GENRE_PRESETS.get(preset_name, "")
-        
-        # 스타일 태그 추가
-        if song_style in SONG_STYLES:
-            style_tags = SONG_STYLES[song_style]
-            result = f"{base_tags}, {style_tags}"
-        else:
-            result = base_tags
-            
-        print(f"✅ Updated tags: {result[:50]}...")
-        return result
+    # 1. 장르 프리셋 변경
+    def on_genre_change(genre, style):
+        print(f"🎵 Genre changed to: {genre}")
+        if genre == "Custom":
+            return TAG_DEFAULT
+        tags = GENRE_PRESETS.get(genre, TAG_DEFAULT)
+        if style and style in SONG_STYLES:
+            tags = f"{tags}, {SONG_STYLES[style]}"
+        return tags
     
-    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("⚠️ 빈 프롬프트")
-            gr.Warning("작사 주제를 입력해주세요!")
-            return gr.update()  # 현재 가사 유지
-        
-        generated = generate_lyrics_with_ai(lyric_prompt, genre_preset, song_style)
-        return generated
+    genre_preset.change(
+        fn=on_genre_change,
+        inputs=[genre_preset, song_style],
+        outputs=prompt
+    )
     
-    def generate_random_music_data(genre_preset, song_style):
-        """랜덤 데이터 생성"""
-        print("🎲 Random 버튼 클릭")
-        # 랜덤 장르 선택
-        if genre_preset == "Custom":
-            genre = random.choice(list(GENRE_PRESETS.keys()))
+    # 2. 스타일 변경
+    def on_style_change(genre, style):
+        print(f"🎤 Style changed to: {style}")
+        if genre == "Custom":
+            base_tags = TAG_DEFAULT
         else:
-            genre = genre_preset
-        
-        # 랜덤 주제 리스트
-        themes = [
-            "도시의 밤", "첫사랑의 추억", "여름날의 해변", "가을의 정취",
-            "희망찬 내일", "자유로운 영혼", "별빛 아래 춤", "청춘의 열정",
-            "비 오는 날의 감성", "꿈을 향한 도전", "이별 후의 성장", "새로운 시작"
-        ]
-        
-        # 랜덤 설정
-        duration = random.choice([30, 60, 90, 120, 180])
-        theme = random.choice(themes)
-        
-        # AI로 가사 생성
-        lyrics_text = generate_lyrics_with_ai(theme, genre, song_style)
-        
-        # 태그 생성
-        tags = GENRE_PRESETS.get(genre, "")
-        if song_style in SONG_STYLES:
-            tags = f"{tags}, {SONG_STYLES[song_style]}"
-        
-        # 랜덤 파라미터 설정
-        return (
-            duration,  # audio_duration
-            tags,      # prompt
-            lyrics_text,    # lyrics
-            150,       # infer_step
-            15.0,      # guidance_scale
-            "euler",   # scheduler_type
-            "apg",     # cfg_type
-            10.0,      # omega_scale
-            str(random.randint(1, 10000)),  # 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
-            False,     # audio2audio_enable
-            0.5,       # ref_audio_strength
-            None,      # ref_audio_input
-        )
-    
-    def generate_preview(prompt, lyrics, genre_preset, song_style):
-        """10초 프리뷰 생성"""
-        print("🎵 Preview 버튼 클릭")
-        preview_params = {
-            "audio_duration": 10,
-            "infer_step": 50,
-            "guidance_scale": 12.0,
-            "scheduler_type": "euler",
-            "cfg_type": "apg",
-            "omega_scale": 5.0,
-        }
-        
-        enhanced_prompt = enhance_prompt_with_genre(prompt, genre_preset, song_style)
+            base_tags = GENRE_PRESETS.get(genre, TAG_DEFAULT)
         
-        try:
-            result = enhanced_process_func(
-                preview_params["audio_duration"],
-                enhanced_prompt,
-                lyrics[:200],
-                preview_params["infer_step"],
-                preview_params["guidance_scale"],
-                preview_params["scheduler_type"],
-                preview_params["cfg_type"],
-                preview_params["omega_scale"],
-                None,
-                0.5,
-                0.0,
-                3.0,
-                True,
-                False,
-                True,
-                None,
-                0.0,
-                0.0,
-                multi_seed_mode="Single",
-                song_style=song_style
-            )
-            return result[0] if result else None
-        except Exception as e:
-            return f"프리뷰 생성 실패: {str(e)}"
-    
-    # ===== 이벤트 핸들러 연결 =====
-    print("🔗 이벤트 핸들러 연결 중...")
+        if style and style in SONG_STYLES:
+            return f"{base_tags}, {SONG_STYLES[style]}"
+        return base_tags
     
-    # 장르 프리셋 변경 이벤트
-    genre_preset.select(
-        fn=update_tags_from_preset,
-        inputs=[genre_preset, prompt, song_style],
-        outputs=[prompt]
+    song_style.change(
+        fn=on_style_change,
+        inputs=[genre_preset, song_style],
+        outputs=prompt
     )
     
-    # 곡 스타일 변경 이벤트
-    song_style.select(
-        fn=update_tags_from_preset,
-        inputs=[genre_preset, prompt, song_style],
-        outputs=[prompt]
-    )
+    # 3. 품질 프리셋
+    def on_quality_change(preset):
+        print(f"⚡ Quality preset changed to: {preset}")
+        if preset in QUALITY_PRESETS:
+            p = QUALITY_PRESETS[preset]
+            return (
+                p["description"],
+                p["infer_step"],
+                p["guidance_scale"],
+                p["scheduler_type"],
+                p["omega_scale"],
+                p["use_erg_diffusion"],
+                p["use_erg_tag"]
+            )
+        return ("", 150, 15.0, "euler", 10.0, True, True)
     
-    # 품질 프리셋 변경 이벤트
-    quality_preset.select(
-        fn=lambda x: QUALITY_PRESETS.get(x, {}).get("description", ""),
-        inputs=[quality_preset],
-        outputs=[preset_description]
-    ).then(
-        fn=update_quality_preset,
-        inputs=[quality_preset],
-        outputs=[infer_step, guidance_scale, scheduler_type, omega_scale, use_erg_diffusion, use_erg_tag]
+    quality_preset.change(
+        fn=on_quality_change,
+        inputs=quality_preset,
+        outputs=[preset_description, infer_step, guidance_scale, scheduler_type, omega_scale, use_erg_diffusion, use_erg_tag]
     )
     
-    # AI 작사 버튼 이벤트
+    # 4. AI 작사
+    def generate_lyrics_click(prompt_text, genre, style):
+        print(f"🤖 AI 작사 버튼 클릭! Prompt: '{prompt_text}'")
+        if not prompt_text:
+            return "작사 주제를 입력해주세요!"
+        return generate_lyrics_with_ai(prompt_text, genre, style)
+    
     generate_lyrics_btn.click(
-        fn=generate_ai_lyrics_wrapper,
+        fn=generate_lyrics_click,
         inputs=[lyric_prompt, genre_preset, song_style],
-        outputs=[lyrics]
+        outputs=lyrics
     )
     
-    # Random 버튼 이벤트
+    # 5. Random 버튼
+    def random_click(genre, style):
+        print("🎲 Random 버튼 클릭!")
+        
+        if genre == "Custom":
+            genre = random.choice(list(GENRE_PRESETS.keys()))
+        
+        themes = ["도시의 밤", "첫사랑", "여름 해변", "가을 정취", "희망", "자유", "별빛", "청춘"]
+        theme = random.choice(themes)
+        duration = random.choice([30, 60, 90, 120])
+        
+        # 태그
+        tags = GENRE_PRESETS.get(genre, TAG_DEFAULT)
+        if style in SONG_STYLES:
+            tags = f"{tags}, {SONG_STYLES[style]}"
+        
+        # 가사 생성
+        new_lyrics = generate_lyrics_with_ai(theme, genre, style)
+        
+        return [
+            duration, tags, new_lyrics, 150, 15.0, "euler", "apg", 10.0,
+            str(random.randint(1, 10000)), 0.5, 0.0, 3.0, True, False, True,
+            None, 0.0, 0.0, False, 0.5, None
+        ]
+    
     random_bnt.click(
-        fn=generate_random_music_data,
+        fn=random_click,
         inputs=[genre_preset, song_style],
         outputs=[
-            audio_duration,
-            prompt,
-            lyrics,
-            infer_step,
-            guidance_scale,
-            scheduler_type,
-            cfg_type,
-            omega_scale,
-            manual_seeds,
-            guidance_interval,
-            guidance_interval_decay,
-            min_guidance_scale,
-            use_erg_tag,
-            use_erg_lyric,
-            use_erg_diffusion,
-            oss_steps,
-            guidance_scale_text,
-            guidance_scale_lyric,
-            audio2audio_enable,
-            ref_audio_strength,
-            ref_audio_input,
-        ],
+            audio_duration, prompt, lyrics, infer_step, guidance_scale,
+            scheduler_type, cfg_type, omega_scale, manual_seeds,
+            guidance_interval, guidance_interval_decay, min_guidance_scale,
+            use_erg_tag, use_erg_lyric, use_erg_diffusion, oss_steps,
+            guidance_scale_text, guidance_scale_lyric, audio2audio_enable,
+            ref_audio_strength, ref_audio_input
+        ]
     )
     
-    # Preview 버튼 이벤트
+    # 6. Preview 버튼
     preview_bnt.click(
-        fn=generate_preview,
+        fn=lambda p, l, g, s: print(f"🎵 Preview clicked! Genre: {g}, Style: {s}"),
         inputs=[prompt, lyrics, genre_preset, song_style],
-        outputs=[outputs[0]]
+        outputs=None
     )
     
-    # 메인 생성 버튼 이벤트
+    # 7. 메인 생성 버튼
     text2music_bnt.click(
         fn=enhanced_process_func,
         inputs=[
-            audio_duration,
-            prompt,
-            lyrics,
-            infer_step,
-            guidance_scale,
-            scheduler_type,
-            cfg_type,
-            omega_scale,
-            manual_seeds,
-            guidance_interval,
-            guidance_interval_decay,
-            min_guidance_scale,
-            use_erg_tag,
-            use_erg_lyric,
-            use_erg_diffusion,
-            oss_steps,
-            guidance_scale_text,
-            guidance_scale_lyric,
-            audio2audio_enable,
-            ref_audio_strength,
-            ref_audio_input,
-            lora_name_or_path,
-            multi_seed_mode,
-            enable_smart_enhancement,
-            genre_preset,
-            song_style
+            audio_duration, prompt, lyrics, infer_step, guidance_scale,
+            scheduler_type, cfg_type, omega_scale, manual_seeds,
+            guidance_interval, guidance_interval_decay, min_guidance_scale,
+            use_erg_tag, use_erg_lyric, use_erg_diffusion, oss_steps,
+            guidance_scale_text, guidance_scale_lyric, audio2audio_enable,
+            ref_audio_strength, ref_audio_input, lora_name_or_path,
+            multi_seed_mode, enable_smart_enhancement, genre_preset, song_style
         ],
-        outputs=outputs + [input_params_json],
+        outputs=outputs + [input_params_json]
     )
     
-    print("✅ 모든 이벤트 핸들러 연결 완료")
+    print("✅ 모든 이벤트 핸들러 연결 완료!")
 
 
 def create_main_demo_ui(