Update app.py

app.py

@@ -1,6 +1,9 @@
 import streamlit as st
 import requests
-
+import streamlit.components.v1 as components
+from gtts import gTTS
+import os
+from io import BytesIO
 
 # 페이지 설정
 st.set_page_config(page_title="ViDraft", layout="wide")
@@ -24,19 +27,20 @@ if 'current_menu' not in st.session_state:
 if 'current_sub_menu' not in st.session_state:
     st.session_state['current_sub_menu'] = ''
 
-# 메인 메뉴 선택
-selected_menu = st.sidebar.selectbox("Menu", list(menus.keys()))
+# 메인 메뉴 선택 - 중복 제거 후 하나만 남김
+selected_menu = st.sidebar.selectbox("Menu", list(menus.keys()), key='main_menu')
 st.session_state['current_menu'] = selected_menu
 
-# 여기에서 `selected_menu` 변수가 이미 정의되었습니다. 이제 하위 메뉴 선택 이후 사용이 시작됩니다.
-
 # 하위 메뉴 선택 (해당되는 경우)
 if selected_menu in menus:
-    selected_sub_menu = st.sidebar.selectbox("Sub Menu", [""] + menus[selected_menu])
+    # `key` 인자에 `selected_menu` 값을 사용하여 고유한 키를 생성
+    selected_sub_menu = st.sidebar.selectbox("Sub Menu", [""] + menus[selected_menu], key=f'sub_menu_{selected_menu}')
     st.session_state['current_sub_menu'] = selected_sub_menu
 else:
     st.session_state['current_sub_menu'] = ''
 
+
+
 # 여기서부터 `selected_menu` 변수를 기반으로 조건문을 사용하는 나머지 코드가 올바르게 작동할 것입니다.
     
 # Pexels API 키 설정
@@ -84,7 +88,37 @@ if selected_menu == "Free Stock" and st.session_state['current_sub_menu'] == "Se
             with cols[idx % 3]:
                 st.video(video)
 
+# 'Image' 메뉴에서 'Remove Background' 선택 시
+if selected_menu == "Image" and st.session_state['current_sub_menu'] == "Remove Background":
+    st.header("Remove Background")
+    # iframe을 사용하여 외부 URL 임베드
+    components.iframe("https://seawolf2357-gttstest.hf.space", width=None, height=600, scrolling=True)
+
 
+# 'Sound' 메뉴의 'TTS(Voice)' 선택 시
+if st.sidebar.selectbox("Menu", list(menus.keys())) == "Sound":
+    if st.sidebar.selectbox("Sub Menu", menus["Sound"]) == "TTS(Voice)":
+        st.header("Text-to-Speech")
+        # 텍스트 입력
+        text = st.text_area("Enter text to synthesize", "Hello, welcome to ViDraft TTS service.")
+        # 언어 선택
+        language = st.selectbox("Choose Language", ["en", "es", "de", "fr", "it", "pt", "ru", "ja", "ko", "zh-cn", "zh-tw"])
+        
+        # 'Synthesize' 버튼
+        if st.button("Synthesize"):
+            if text:
+                try:
+                    tts = gTTS(text=text, lang=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}")
+            else:
+                st.warning("Please enter some text to synthesize.")
+                
+    
 # 'Home' 페이지 표시
 if selected_menu == "Home":
     st.header("Welcome to ViDraft")
@@ -114,9 +148,7 @@ elif selected_menu == "Image":
     elif st.session_state['current_sub_menu'] == "Inpainting":
         st.header("Inpainting")
         st.write("Fill in missing parts of images.")
-    elif st.session_state['current_sub_menu'] == "Remove Background":
-        st.header("Remove Background")
-        st.write("Easily remove backgrounds from images.")
+
     elif st.session_state['current_sub_menu'] == "Studio":
         st.header("Image Studio")
         st.write("Access a suite of image editing tools.")
@@ -144,9 +176,6 @@ elif selected_menu == "Sound":
     elif st.session_state['current_sub_menu'] == "Video Music":
         st.header("Video Music")
         st.write("Find the perfect music for your video content.")
-    elif st.session_state['current_sub_menu'] == "TTS(Voice)":
-        st.header("Text-to-Speech")
-        st.write("Convert your text into natural-sounding voice.")
     elif st.session_state['current_sub_menu'] == "Image SFX":
         st.header("Image Sound Effects")
         st.write("Add sound effects to your images.")