Update app.py

app.py

@@ -41,37 +41,36 @@ else:
 
 
 
-
 translator = Translator()
 
-# 'Sound' 메뉴의 'TTS(Voice)' 선택 시
-if selected_menu == "Sound" and selected_sub_menu == "TTS(Voice)":
-    st.header("Text-to-Speech with Translation")
-    
-    # 텍스트 입력
-    text_to_translate = st.text_area("Enter text to translate and synthesize", "Hello, welcome to ViDraft TTS service.")
-    
-    # 타겟 언어 선택
-    target_language = st.selectbox("Choose Target Language for Translation", options=list(LANGUAGES.values()), index=list(LANGUAGES.values()).index('English'))
-    
-    # 번역 버튼
-    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"):
+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=translation.text, lang=list(LANGUAGES.keys())[list(LANGUAGES.values()).index(target_language)], slow=False)
+            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 키 설정