Update app.py

app.py

@@ -1,27 +1,30 @@
 import gradio as gr
 from gtts import gTTS
+import librosa
 import os
 
 def text_to_speech(prompt):
-    # 입력된 Bulgarian 텍스트를 음성으로 변환
+    # gTTS를 사용하여 Bulgarian 텍스트를 음성으로 변환
     tts = gTTS(text=prompt, lang="bg")  # 'bg'는 Bulgarian 언어 코드입니다.
     audio_file = "output.mp3"
     tts.save(audio_file)
     
-    # 파일 경로 대신 바이너리 데이터를 반환합니다.
-    with open(audio_file, "rb") as f:
-        audio_binary = f.read()
-    # 임시 파일 삭제
+    # librosa를 사용하여 mp3 파일을 numpy 배열과 샘플레이트로 로드합니다.
+    # sr=None로 설정하면 원본 샘플레이트를 유지합니다.
+    audio_array, sample_rate = librosa.load(audio_file, sr=None)
+    
+    # 임시 mp3 파일 삭제
     os.remove(audio_file)
     
-    return audio_binary
+    # gr.Audio(type="numpy")는 (numpy array, sample_rate) 튜플을 기대합니다.
+    return audio_array, sample_rate
 
 with gr.Blocks() as demo:
     gr.Markdown("## Bulgarian Text-to-Speech (TTS)")
     with gr.Row():
         input_prompt = gr.Textbox(label="Enter a prompt in Bulgarian:")
-        # gr.Audio의 type을 "binary"로 지정하여 파일 경로 대신 바이너리 데이터를 사용합니다.
-        output_audio = gr.Audio(label="Generated Speech", type="binary")
+        # type을 "numpy"로 설정하여 numpy 배열을 사용합니다.
+        output_audio = gr.Audio(label="Generated Speech", type="numpy")
     generate_button = gr.Button("Generate Speech")
     
     generate_button.click(text_to_speech, inputs=input_prompt, outputs=output_audio)