Update app.py

app.py

@@ -1,6 +1,7 @@
 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
@@ -39,41 +40,38 @@ else:
     st.session_state['current_sub_menu'] = ''
 
 
+
+
+translator = Translator()
+
 # 'Sound' 메뉴의 'TTS(Voice)' 선택 시
 if selected_menu == "Sound" and selected_sub_menu == "TTS(Voice)":
-    st.header("Text-to-Speech")
+    st.header("Text-to-Speech with Translation")
+    
     # 텍스트 입력
-    text = st.text_area("Enter text to synthesize", "Hello, welcome to ViDraft TTS service.")
+    text_to_translate = st.text_area("Enter text to translate and 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'을 기본 언어로 설정
-    )
+    # 타겟 언어 선택
+    target_language = st.selectbox("Choose Target Language for Translation", options=list(LANGUAGES.values()), index=list(LANGUAGES.values()).index('English'))
     
-    # '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.")
-
+    # 번역 버튼
+    if st.button("Translate"):
+        # 번역 실행
+        translation = translator.translate(text_to_translate, dest=list(LANGUAGES.keys())[list(LANGUAGES.values()).index(target_language)])
+        st.text_area("Translated Text", translation.text, height=100)
+    
+    # 음성 생성 버튼
+    if st.button("Synthesize Voice"):
+        try:
+            # 번역된 텍스트를 음성으로 변환
+            tts = gTTS(text=translation.text, lang=list(LANGUAGES.keys())[list(LANGUAGES.values()).index(target_language)], 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}")
             
             
 # Pexels API 키 설정