Update app.py

app.py

@@ -1,7 +1,6 @@
 import streamlit as st
 import requests
 import streamlit.components.v1 as components
-from googletrans import Translator, LANGUAGES
 from gtts import gTTS
 from gtts.lang import tts_langs
 from io import BytesIO
@@ -40,37 +39,41 @@ 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_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:
+                # 선택된 언어로 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")
+            except Exception as e:
+                st.error(f"Error: {e}")
+        else:
+            st.warning("Please enter some text to synthesize.")
 
-translator = Translator()
-
-if 'translated_text' not in st.session_state:
-    st.session_state['translated_text'] = ""
-
-# 번역할 텍스트 입력
-text_to_translate = st.text_area("Enter text to translate and synthesize", "")
-
-# 타겟 언어 선택
-target_language = st.selectbox("Select target language for translation", list(LANGUAGES.values()))
-
-# 번역 버튼
-if st.button("Translate"):
-    translation = translator.translate(text_to_translate, dest=target_language)
-    st.session_state['translated_text'] = translation.text
-    st.write(f"Translated Text: {translation.text}")
-
-# TTS 생성 버튼
-if st.button("Synthesize Speech"):
-    if st.session_state['translated_text']:
-        try:
-            tts = gTTS(text=st.session_state['translated_text'], lang='en', slow=False)  # 예시에서는 'en'을 사용
-            audio_file = BytesIO()
-            tts.write_to_fp(audio_file)
-            audio_file.seek(0)
-            st.audio(audio_file, format="audio/mp3")
-        except Exception as e:
-            st.error(f"Error: {e}")
-    else:
-        st.warning("Please translate some text first.")
             
             
 # Pexels API 키 설정