Update app.py

app.py

@@ -0,0 +1,56 @@
+import gradio as gr
+from mars5 import Mars5TTS
+import torch
+import numpy as np
+
+# MARS5 TTS 모델 로드 (Hugging Face Hub에서)
+model = Mars5TTS.from_pretrained("camb-ai/mars5-tts")
+
+def clone_with_prosody(text, ref_audio, enhance_prosody=True):
+    """
+    입력 텍스트와 참조 오디오를 받아 말투를 클론하여 오디오 출력
+    :param text: 변환할 텍스트
+    :param ref_audio: 말투를 복제할 오디오 파일 (3-5초 이상 권장)
+    :param enhance_prosody: Prosody(억양/리듬) 강조 여부
+    :return: 출력 오디오 파일 경로
+    """
+    try:
+        # 참조 오디오 처리 (Gradio에서 업로드된 오디오: numpy 배열 또는 파일 경로)
+        if isinstance(ref_audio, np.ndarray):
+            # Gradio에서 제공하는 오디오 데이터 처리
+            audio_data = ref_audio
+        else:
+            # 파일 경로일 경우
+            audio_data = ref_audio
+
+        # MARS5 TTS로 클로닝 (prosody 강조)
+        output_audio = model.tts(
+            text=text,
+            ref_audio=audio_data,
+            prosody_enhance=enhance_prosody,  # 억양/리듬 복제 강화
+            language="ko"  # 한국어 설정
+        )
+        
+        # 출력 오디오를 파일로 저장
+        output_path = "output_cloned_audio.wav"
+        output_audio.save(output_path)
+        return output_path
+    except Exception as e:
+        return f"Error: {str(e)}"
+
+# Gradio 인터페이스 설정
+interface = gr.Interface(
+    fn=clone_with_prosody,
+    inputs=[
+        gr.Textbox(label="Text to Convert", placeholder="Enter text to convert to speech"),
+        gr.Audio(label="Reference Audio (Your Voice)", type="filepath", source="upload"),
+        gr.Checkbox(label="Enhance Prosody (Intonation/Rhythm)", value=True)
+    ],
+    outputs=gr.Audio(label="Cloned Voice Output"),
+    title="MARS5 Voice Cloner with Prosody",
+    description="Upload a 3-5 second audio of your voice and enter text to clone your voice with prosody (intonation, rhythm, emotion).",
+    allow_flagging="never"
+)
+
+# 앱 실행
+interface.launch()