app.py

import gradio as gr
import requests

def generate_audio(text, voice_name):
    
    MAX_CHARS = 100
    # 입력된 텍스트 길이 검증
    if len(text) > MAX_CHARS:
        text = text[:MAX_CHARS]  # 최대 글자수로 텍스트 자르기
        
    voices = {
        "남성AI": "6uQD2RqeSgIey9LVBafG",
        "여성AI": "jeuEgCxtyEYkC3zsQODy",  
        "트럼프": "nP4xQZkEwzlvgutdc3xM",         
        "차승원": "SKwm0HLYsVDCM2ruvw2p",
        "김정은": "YPWL3nQPzBN1XaiZF4aj",
        "노무현": "6JdivX36i1eJ3LqLmvwH",
        "궁예": "4kFrgJPCTjA6DyPM5Gr5",        
        "윤석열": "Qrf03Q67kRgpFlj9fto3"        
    }
    voice_id = voices[voice_name]
    url = f"https://api.elevenlabs.io/v1/text-to-speech/{voice_id}"
    model_id = "eleven_multilingual_v2"
    payload = {
        "model_id": model_id,
        "text": text,
        "voice_settings": {
            "similarity_boost": 0.75,
            "stability": 0.5,
            "style": 0,
            "use_speaker_boost": True
        }
    }
    headers = {
        "Content-Type": "application/json",
        "xi-api-key": "c5fb99a2b25402f104d246379bcf819a"
    }

    response = requests.post(url, json=payload, headers=headers, stream=True)

    if response.status_code == 200:
        chunks = []
        for chunk in response.iter_content(chunk_size=CHUNK_SIZE):
            chunks.append(chunk)
        return b''.join(chunks)
    else:
        raise Exception(f"오류 발생. 상태 코드: {response.status_code}")

CHUNK_SIZE = 1024


# Gradio 인터페이스 정의
with gr.Blocks() as demo:
    gr.Markdown("FakeVoice: AI가 복제한 유명인의 음성으로, 텍스트만 입력하면 자연스럽게 해당 인물의 음성으로 생성합니다.")    
    gr.Markdown("지금 커뮤니티에 참여하시면 더욱 많은 정보와 다양한 기회를 얻으실 것입니다. 커뮤니티 참여 링크 https://open.kakao.com/o/gE6hK9Vf ")
    gr.Markdown("생성된 오디오를 다운로드 할 수 있습니다. 다운로드 받으신 후 다운로드 받은 파일의 확장자를 mp3로 변경하시면 됩니다. ")    
    text_input = gr.Textbox(label="음성으로 생성할 텍스트를 입력하세요.(100글자 이내 제한 설정되어, 초과시 음성 생성이 제한됩니다.)")
    voice_choice = gr.Dropdown(choices=["남성AI", "여성AI", "트럼프", "차승원", "김정은", "노무현", "궁예", "윤석열"], label="음성 선택")
    submit_button = gr.Button("생성")
    audio_output = gr.Audio(label="생성된 오디오")

    submit_button.click(
        fn=generate_audio,
        inputs=[text_input, voice_choice],
        outputs=audio_output
    )


if __name__ == "__main__":
    demo.launch()