Update app.py

app.py

@@ -16,7 +16,7 @@ st.sidebar.title("ViDraft")
 menus = {
     "Home": [],
     "Free Stock": ["Template Video", "Search Video", "Search Image"],
-    "Image": ["Remove Background", "Compositing"],
+    "Image": ["Face Swap", "Remove Background", "Compositing"],
     "Video": ["Generation", "Talking Face", "Remove Background", "Compositing"],
     "Sound": ["TTS(Voice)", "Video SFX", "Video Music","Image SFX", "Image Music"],
     "Scripts": []
@@ -135,7 +135,11 @@ if selected_menu == "Image" and st.session_state['current_sub_menu'] == "Composi
     # iframe을 사용하여 외부 URL 임베드
     components.iframe("https://arxivgpt-vidistudio.hf.space", width=None, height=1800, scrolling=True)
      
-
+# 'Image' 메뉴에서 'Face Swap' 선택 시
+if selected_menu == "Image" and st.session_state['current_sub_menu'] == "Face Swap":
+    st.header("Face Swap")
+    # iframe을 사용하여 외부 URL 임베드
+    components.iframe("https://arxivgpt-vidifs.hf.space", width=None, height=1800, scrolling=True)
 
 # 'video' 메뉴에서 'Compositing' 선택 시
 if selected_menu == "Video" and st.session_state['current_sub_menu'] == "Compositing":
@@ -148,6 +152,8 @@ if selected_menu == "Video" and st.session_state['current_sub_menu'] == "Remove
     st.header("Remove Background")
     # iframe을 사용하여 외부 URL 임베드
     components.iframe("https://arxivgpt-vidvback.hf.space", width=None, height=1800, scrolling=True)
+
+
     
 # 'Home' 페이지 표시
 if selected_menu == "Home":
@@ -203,9 +209,6 @@ if selected_menu == "Home":
                 col.markdown(f"<div style='text-align: center; font-weight: bold; font-size: 20px;'>{image_texts[i + idx]}</div>", unsafe_allow_html=True)
 
 
-
-
-
 elif selected_menu == "Free Stock":
     # 'Free Stock' 메뉴 선택 시 처리 로직
     if st.session_state['current_sub_menu'] == "Template Video":
@@ -220,6 +223,9 @@ elif selected_menu == "Free Stock":
                 st.video(video_file)
 
 
+
+                
+
 # 'Video' 메뉴 선택 시 처리 로직
 elif selected_menu == "Video":
     if st.session_state['current_sub_menu'] == "Generation":
@@ -245,6 +251,7 @@ elif selected_menu == "Sound":
         st.header("Image Music")
         st.write("Associate music tracks with your images.")
 
+        
 # 'Scripts' 메뉴 선택 시 처리 로직
 elif selected_menu == "Scripts":
     st.header("Scripts")