Update app.py

app.py

@@ -16,8 +16,8 @@ st.sidebar.title("ViDraft")
 menus = {
     "Home": [],
     "Free Stock": ["Template Video", "Search Video", "Search Image"],
-    "Image": ["Face Swap", "Remove Background", "Compositing"],
-    "Video": ["Generation", "Talking Face", "Remove Background", "Compositing"],
+    "Image": ["Face Swap", "SFX", "Remove Background", "Compositing"],
+    "Video": ["Generation", "Talking Face", "SFX", "Remove Background", "Compositing"],
     "Sound": ["TTS(Voice)", "Video SFX", "Video Music","Image SFX", "Image Music"],
     "Scripts": []
 }
@@ -153,7 +153,19 @@ if selected_menu == "Video" and st.session_state['current_sub_menu'] == "Remove
     # iframe을 사용하여 외부 URL 임베드
     components.iframe("https://arxivgpt-vidvback.hf.space", width=None, height=1800, scrolling=True)
 
+    
+# 'Image' 메뉴에서 'SFX' 선택 시
+if selected_menu == "Image" and st.session_state['current_sub_menu'] == "SFX":
+    st.header("Image SFX")
+    # iframe을 사용하여 외부 URL 임베드
+    components.iframe("https://arxivgpt-vidisfx.hf.space", width=None, height=1800, scrolling=True)
 
+# 'video' 메뉴에서 'SFX' 선택 시
+if selected_menu == "Video" and st.session_state['current_sub_menu'] == "SFX":
+    st.header("Video SFX")
+    # iframe을 사용하여 외부 URL 임베드
+    components.iframe("https://arxivgpt-vidvsfx.hf.space", width=None, height=1800, scrolling=True)
+    
     
 # 'Home' 페이지 표시
 if selected_menu == "Home":
@@ -240,15 +252,9 @@ elif selected_menu == "Video":
 
 # 'Sound' 메뉴 선택 시 처리 로직
 elif selected_menu == "Sound":
-    if st.session_state['current_sub_menu'] == "Video SFX":
-        st.header("Video Sound Effects")
-        st.write("Enhance your videos with sound effects.")
     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'] == "Image SFX":
-        st.header("Image Sound Effects")
-        st.write("Add sound effects to your images.")
     elif st.session_state['current_sub_menu'] == "Image Music":
         st.header("Image Music")
         st.write("Associate music tracks with your images.")