Update app.py

app.py

@@ -2,6 +2,7 @@ import streamlit as st
 import requests
 import streamlit.components.v1 as components
 from gtts import gTTS
+from gtts.lang import tts_langs
 from io import BytesIO
 
 # 페이지 설정
@@ -37,65 +38,37 @@ if selected_menu in menus:
 else:
     st.session_state['current_sub_menu'] = ''
 
+
 # 'Sound' 메뉴의 'TTS(Voice)' 선택 시
 if selected_menu == "Sound" and selected_sub_menu == "TTS(Voice)":
     st.header("Text-to-Speech")
     # 텍스트 입력
     text = st.text_area("Enter text to synthesize", "Hello, welcome to ViDraft TTS service.")
     
-    # 언어 선택 - 더 많은 언어 추가
-    languages = [
-        ("Arabic", "ar-XA"),
-        ("Bengali", "bn-IN"),
-        ("Chinese, Mandarin (Simplified)", "cmn-CN"),
-        ("Chinese, Mandarin (Traditional)", "cmn-TW"),
-        ("Danish", "da-DK"),
-        ("Dutch", "nl-NL"),
-        ("English, Australia", "en-AU"),
-        ("English, India", "en-IN"),
-        ("English, UK", "en-GB"),
-        ("English, US", "en-US"),  # 기본값으로 설정될 언어
-        ("French", "fr-FR"),
-        ("French, Canada", "fr-CA"),
-        ("German", "de-DE"),
-        ("Gujarati", "gu-IN"),
-        ("Hindi", "hi-IN"),
-        ("Indonesian", "id-ID"),
-        ("Italian", "it-IT"),
-        ("Japanese", "ja-JP"),
-        ("Kannada", "kn-IN"),
-        ("Korean", "ko-KR"),
-        ("Malayalam", "ml-IN"),
-        ("Marathi", "mr-IN"),
-        ("Norwegian", "nb-NO"),
-        ("Polish", "pl-PL"),
-        ("Portuguese, Brazil", "pt-BR"),
-        ("Portuguese, Portugal", "pt-PT"),
-        ("Russian", "ru-RU"),
-        ("Spanish, Spain", "es-ES"),
-        ("Spanish, Mexico", "es-MX"),
-        ("Swedish", "sv-SE"),
-        ("Tamil", "ta-IN"),
-        ("Telugu", "te-IN"),
-        ("Thai", "th-TH"),
-        ("Turkish", "tr-TR"),
-        ("Ukrainian", "uk-UA"),
-        ("Vietnamese", "vi-VN"),
-    ]
-    # 'English, US'를 기본 언어로 설정
-    default_lang_index = languages.index(("English, US", "en-US"))
-    language = st.selectbox("Choose Language", languages, format_func=lambda x: x[0], index=default_lang_index, key='language_select')
+    # 지원되는 언어 목록을 불러옵니다.
+    languages_dict = tts_langs()
+    # ISO 639-1 표준에 따라 두 글자 코드를 가진 언어만 필터링
+    two_letter_languages = {code: lang for code, lang in languages_dict.items() if len(code) == 2}
+
+    # 언어 선택을 위한 selectbox를 생성합니다.
+    selected_language_code = st.selectbox(
+        "Choose Language", 
+        options=list(two_letter_languages.keys()), 
+        format_func=lambda x: f"{two_letter_languages[x]} ({x})",
+        index=list(two_letter_languages.keys()).index('en')  # 'en'을 기본 언어로 설정
+    )
     
     # 'Synthesize' 버튼
     if st.button("Synthesize"):
         if text:
             try:
-                tts_language = dict(languages)[language]
-                tts = gTTS(text=text, lang=tts_language, slow=False)
+                # 선택된 언어로 gTTS 객체 생성
+                tts = gTTS(text=text, lang=selected_language_code, slow=False)
                 audio_file = BytesIO()
                 tts.write_to_fp(audio_file)
                 audio_file.seek(0)
-                st.audio(audio_file, format='audio/mp3')
+                # 생성된 오디오 파일을 재생
+                st.audio(audio_file, format="audio/mp3")
             except Exception as e:
                 st.error(f"Error: {e}")
         else: