Suno

# Suno 스타일 프롬프트 구성 예제 (개념적 코드)
# 실제 Suno는 웹 인터페이스를 통해 사용하며,
# API 자동화는 비공식 도구(suno-api 등)를 활용

class SunoPromptBuilder:
    """Suno 프롬프트를 구조화하여 생성하는 헬퍼 클래스"""

    def __init__(self):
        self.style_tags = []
        self.lyrics = ""
        self.title = ""

    def add_genre(self, genre: str):
        """장르 태그 추가 (pop, rock, hip-hop, k-pop, jazz 등)"""
        self.style_tags.append(genre)
        return self

    def add_mood(self, mood: str):
        """분위기 태그 (upbeat, melancholic, energetic, chill 등)"""
        self.style_tags.append(mood)
        return self

    def add_vocal_style(self, style: str):
        """보컬 스타일 (male vocal, female vocal, choir 등)"""
        self.style_tags.append(style)
        return self

    def add_instruments(self, instruments: list):
        """악기 지정 (acoustic guitar, piano, synth, drums 등)"""
        self.style_tags.extend(instruments)
        return self

    def set_lyrics(self, lyrics: str):
        """
        가사 설정 - Suno 형식 지원
        [Verse], [Chorus], [Bridge], [Outro] 등의 태그 사용
        """
        self.lyrics = lyrics
        return self

    def build(self) -> dict:
        """최종 프롬프트 생성"""
        return {
            "style": ", ".join(self.style_tags),
            "lyrics": self.lyrics,
            "title": self.title
        }


# 실제 사용 예제
prompt_builder = SunoPromptBuilder()

# K-Pop 스타일 노래 프롬프트 구성
kpop_prompt = (
    prompt_builder
    .add_genre("k-pop")
    .add_mood("upbeat, energetic")
    .add_vocal_style("female vocal, harmony")
    .add_instruments(["synth", "electronic drums", "bass"])
    .set_lyrics("""
[Verse 1]
새벽빛이 물드는 거리
너를 만나러 가는 길
설레는 맘 감출 수 없어
오늘도 빛나는 하루

[Chorus]
빛나는 별처럼 shine
우리의 이야기 tonight
함께라면 두렵지 않아
영원히 빛날 거야
    """)
    .build()
)

print("Style Tags:", kpop_prompt["style"])
print("Lyrics:", kpop_prompt["lyrics"])


# 프롬프트 품질 체크리스트
class SunoQualityChecker:
    """생성 품질을 높이기 위한 프롬프트 검증"""

    @staticmethod
    def check_lyrics_structure(lyrics: str) -> list:
        """가사 구조 검증"""
        issues = []

        # 섹션 태그 확인
        required_sections = ["[Verse", "[Chorus"]
        for section in required_sections:
            if section not in lyrics:
                issues.append(f"Missing {section} section")

        # 가사 길이 확인 (너무 길면 잘림)
        if len(lyrics) > 3000:
            issues.append("Lyrics too long, may be truncated")

        # 빈 줄로 섹션 구분 확인
        if "\n\n" not in lyrics:
            issues.append("Consider adding blank lines between sections")

        return issues

    @staticmethod
    def suggest_style_improvements(style_tags: list) -> list:
        """스타일 태그 개선 제안"""
        suggestions = []

        # 장르 태그 확인
        common_genres = ["pop", "rock", "hip-hop", "jazz", "r&b",
                         "electronic", "classical", "folk", "k-pop"]
        has_genre = any(g in " ".join(style_tags).lower()
                        for g in common_genres)
        if not has_genre:
            suggestions.append("Add a specific genre tag for better results")

        # BPM 힌트
        if len(style_tags) < 3:
            suggestions.append("Add mood/tempo tags (e.g., 'upbeat', '120 bpm')")

        return suggestions


# 검증 실행
checker = SunoQualityChecker()
issues = checker.check_lyrics_structure(kpop_prompt["lyrics"])
suggestions = checker.suggest_style_improvements(kpop_prompt["style"].split(", "))

print("\n--- Quality Check ---")
print("Issues:", issues if issues else "None")
print("Suggestions:", suggestions if suggestions else "All good!")