Update ui/components.py

ui/components.py

@@ -18,9 +18,16 @@ from openai import OpenAI
 
 # OpenAI 클라이언트 초기화
 try:
-    client = OpenAI(api_key=os.getenv("LLM_API"))
-except:
+    api_key = os.getenv("LLM_API") or os.getenv("OPENAI_API_KEY")
+    if api_key:
+        client = OpenAI(api_key=api_key)
+        print("OpenAI API client initialized successfully")
+    else:
+        client = None
+        print("Warning: No OpenAI API key found. AI lyrics generation will be disabled.")
+except Exception as e:
     client = None
+    print(f"Warning: Failed to initialize OpenAI client: {e}")
 
 TAG_DEFAULT = "funk, pop, soul, rock, melodic, guitar, drums, bass, keyboard, percussion, 105 BPM, energetic, upbeat, groovy, vibrant, dynamic, duet, male and female vocals"
 LYRIC_DEFAULT = """[verse - male]
@@ -62,7 +69,7 @@ In this moment we take flight
 
 # 확장된 장르 프리셋 (기존 + 개선된 태그)
 GENRE_PRESETS = {
-    "Modern Pop": "pop, synth, drums, guitar, 120 bpm, upbeat, catchy, vibrant, duet vocals, polished vocals, radio-ready, commercial, layered vocals",
+    "Modern Pop": "pop, synth, drums, guitar, 120 bpm, upbeat, catchy, vibrant, polished vocals, radio-ready, commercial, layered vocals",
     "Rock": "rock, electric guitar, drums, bass, 130 bpm, energetic, rebellious, gritty, powerful vocals, raw vocals, power chords, driving rhythm",
     "Hip Hop": "hip hop, 808 bass, hi-hats, synth, 90 bpm, bold, urban, intense, rhythmic vocals, trap beats, punchy drums",
     "Country": "country, acoustic guitar, steel guitar, fiddle, 100 bpm, heartfelt, rustic, warm, twangy vocals, storytelling, americana",
@@ -109,22 +116,29 @@ LYRIC_SYSTEM_PROMPT = """너는 노래 가사를 작사하는 전문가 역할
 ...
 """
 
-def generate_lyrics_with_ai(prompt: str, genre: str, song_style: str, language: str = "auto") -> str:
+def generate_lyrics_with_ai(prompt: str, genre: str, song_style: str) -> str:
     """AI를 사용하여 가사 생성"""
     if not client:
+        print("OpenAI client not available, returning default lyrics")
+        return LYRIC_DEFAULT
+    
+    if not prompt or prompt.strip() == "":
+        print("Empty prompt, returning default lyrics")
         return LYRIC_DEFAULT
     
     try:
         # 언어 감지 및 스타일 정보 추가
         style_info = ""
         if "듀엣" in song_style:
-            style_info = "남녀 듀엣 형식으로 파트를 나누어 작성해주세요."
+            style_info = "남녀 듀엣 형식으로 파트를 나누어 작성해주세요. [verse - male], [verse - female], [chorus - duet] 형식을 사용하세요."
         elif "솔로 (남성)" in song_style:
             style_info = "남성 솔로 가수를 위한 가사를 작성해주세요."
         elif "솔로 (여성)" in song_style:
             style_info = "여성 솔로 가수를 위한 가사를 작성해주세요."
         elif "그룹" in song_style:
             style_info = "그룹이 부르는 형식으로 파트를 나누어 작성해주세요."
+        elif "인스트루멘탈" in song_style:
+            return "[instrumental]\n\n[inst]\n\n[instrumental break]\n\n[inst]"
         
         user_prompt = f"""
 주제: {prompt}
@@ -134,8 +148,10 @@ def generate_lyrics_with_ai(prompt: str, genre: str, song_style: str, language:
 위 정보를 바탕으로 노래 가사를 작성해주세요. 입력된 언어와 동일한 언어로 작성하고, 구조 태그를 반드시 포함해주세요.
 """
         
+        print(f"Generating lyrics with prompt: {prompt}")
+        
         response = client.chat.completions.create(
-            model="gpt-4-mini",
+            model="gpt-4o-mini",  # 모델명 수정
             messages=[
                 {"role": "system", "content": LYRIC_SYSTEM_PROMPT},
                 {"role": "user", "content": user_prompt}
@@ -144,7 +160,10 @@ def generate_lyrics_with_ai(prompt: str, genre: str, song_style: str, language:
             max_tokens=1000
         )
         
-        return response.choices[0].message.content
+        generated_lyrics = response.choices[0].message.content
+        print(f"Generated lyrics successfully")
+        return generated_lyrics
+        
     except Exception as e:
         print(f"AI 가사 생성 오류: {e}")
         return LYRIC_DEFAULT
@@ -226,9 +245,9 @@ generation_cache = MusicGenerationCache()
 
 def enhance_prompt_with_genre(base_prompt: str, genre: str, song_style: str) -> str:
     """장르와 스타일에 따른 스마트 프롬프트 확장"""
-    if genre == "Custom" or not genre:
-        enhanced_prompt = base_prompt
-    else:
+    enhanced_prompt = base_prompt
+    
+    if genre != "Custom" and genre:
         # 장르별 추가 개선 태그
         genre_enhancements = {
             "Modern Pop": ["polished production", "mainstream appeal", "hook-driven"],
@@ -248,8 +267,6 @@ def enhance_prompt_with_genre(base_prompt: str, genre: str, song_style: str) ->
         if genre in genre_enhancements:
             additional_tags = ", ".join(genre_enhancements[genre])
             enhanced_prompt = f"{base_prompt}, {additional_tags}"
-        else:
-            enhanced_prompt = base_prompt
     
     # 스타일 태그 추가
     if song_style in SONG_STYLES:
@@ -278,10 +295,19 @@ def calculate_quality_score(audio_path: str) -> float:
     except:
         return 50.0  # 기본값
 
-def update_tags_from_preset(preset_name):
+def update_tags_from_preset(preset_name, current_tags, song_style):
+    """장르 프리셋 선택시 태그 업데이트"""
     if preset_name == "Custom":
-        return ""
-    return GENRE_PRESETS.get(preset_name, "")
+        return current_tags
+    
+    base_tags = GENRE_PRESETS.get(preset_name, "")
+    
+    # 스타일 태그 추가
+    if song_style in SONG_STYLES:
+        style_tags = SONG_STYLES[song_style]
+        return f"{base_tags}, {style_tags}"
+    
+    return base_tags
 
 def update_quality_preset(preset_name):
     """품질 프리셋 적용"""
@@ -690,9 +716,12 @@ def create_text2music_ui(
 
             # AI 작사 버튼 이벤트
             def generate_ai_lyrics(lyric_prompt, genre_preset, song_style):
-                if not lyric_prompt:
-                    return "작사 주제를 입력해주세요."
-                return generate_lyrics_with_ai(lyric_prompt, genre_preset, song_style)
+                if not lyric_prompt or lyric_prompt.strip() == "":
+                    return lyrics.value  # 현재 가사 유지
+                
+                print(f"Generating lyrics with: prompt={lyric_prompt}, genre={genre_preset}, style={song_style}")
+                generated = generate_lyrics_with_ai(lyric_prompt, genre_preset, song_style)
+                return generated
             
             generate_lyrics_btn.click(
                 fn=generate_ai_lyrics,
@@ -752,10 +781,20 @@ def create_text2music_ui(
                     None,      # ref_audio_input
                 )
 
-            # 모든 UI 요소가 정의된 후 이벤트 핸들러 설정
+            # 장르 프리셋 변경 시 태그 업데이트
+            def on_genre_change(genre_preset, current_tags, song_style):
+                return update_tags_from_preset(genre_preset, current_tags, song_style)
+            
             genre_preset.change(
-                fn=update_tags_from_preset,
-                inputs=[genre_preset],
+                fn=on_genre_change,
+                inputs=[genre_preset, prompt, song_style],
+                outputs=[prompt]
+            )
+            
+            # 곡 스타일 변경 시 태그 업데이트
+            song_style.change(
+                fn=on_genre_change,
+                inputs=[genre_preset, prompt, song_style],
                 outputs=[prompt]
             )
             
@@ -1570,6 +1609,14 @@ def create_main_demo_ui(
                 - **장르 특화**: 장르 프리셋 선택 후 "스마트 향상" 체크
                 - **가사 구조**: [verse], [chorus], [bridge] 태그 적극 활용
                 - **다국어 지원**: 한국어로 주제를 입력하면 한국어 가사가 생성됩니다
+                
+                ### ⚠️ OpenAI API 설정
+                AI 작사 기능을 사용하려면 환경변수에 OpenAI API 키를 설정해야 합니다:
+                ```bash
+                export LLM_API="your-openai-api-key"
+                # 또는
+                export OPENAI_API_KEY="your-openai-api-key"
+                ```
                 """)
             
             with gr.Tab("🎵 Enhanced Text2Music", elem_classes="gr-tab"):