Update app.py

app.py

@@ -1,49 +1,39 @@
 import gradio as gr
 from gtts import gTTS
 from pydub import AudioSegment
-import numpy as np
-import os
+from io import BytesIO
 
-def text_to_speech(prompt):
-    try:
-        # 1) gTTS로 mp3 생성
-        tts = gTTS(text=prompt, lang="bg")
-        audio_file = "output.mp3"
-        tts.save(audio_file)
-    except Exception as e:
-        print("gTTS 생성 오류:", e)
-        raise e  # 에러를 다시 발생시켜 Gradio에서 감지하도록 함
+def bulgarian_tts(bulgarian_text):
+    # gTTS를 이용해 불가리아어 텍스트를 음성으로 변환
+    tts = gTTS(text=bulgarian_text, lang="bg")
     
-    try:
-        # 2) mp3 -> numpy 변환 (pydub 사용)
-        sound = AudioSegment.from_mp3(audio_file)
-        samples = np.array(sound.get_array_of_samples())
-        
-        # 스테레오면 모노로 변환
-        if sound.channels > 1:
-            samples = samples.reshape((-1, sound.channels))
-            samples = samples.mean(axis=1)
-        
-        # int16 -> float32 정규화
-        samples = samples.astype(np.float32) / 32768.0
-        sample_rate = sound.frame_rate
-    except Exception as e:
-        print("pydub 로딩/처리 오류:", e)
-        raise e
-    finally:
-        if os.path.exists(audio_file):
-            os.remove(audio_file)
+    # 메모리 버퍼에 mp3로 저장 후, pydub 오디오 세그먼트로 로딩
+    audio_buffer = BytesIO()
+    tts.write_to_fp(audio_buffer)
+    audio_buffer.seek(0)
+    tts_audio = AudioSegment.from_file(audio_buffer, format="mp3")
     
-    return samples, sample_rate
+    # mp3 파일로 내보내기
+    output_file = "bulgarian_output.mp3"
+    tts_audio.export(output_file, format="mp3")
+    
+    return output_file
 
 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:")
-        output_audio = gr.Audio(label="Generated Speech", type="numpy")
+    
+    # 단일 불가리아어 입력만 받도록 구성
+    bulgarian_input = gr.Textbox(label="Enter Bulgarian Text:", placeholder="Здравейте")
+    output_audio = gr.Audio(label="Generated Speech", type="filepath")
+    
     generate_button = gr.Button("Generate Speech")
     
-    generate_button.click(text_to_speech, inputs=input_prompt, outputs=output_audio)
+    # 불가리아어 텍스트만 받아 TTS 오디오 파일 생성
+    generate_button.click(
+        bulgarian_tts, 
+        inputs=bulgarian_input, 
+        outputs=output_audio
+    )
 
 if __name__ == "__main__":
     demo.launch()