Update app.py

app.py

@@ -16,8 +16,8 @@ st.sidebar.title("ViDraft")
 menus = {
     "Home": [],
     "Free Stock": ["Template Video", "Search Video", "Search Image"],
-    "Image": ["Generation", "Face ID", "Inpainting", "Remove Background", "Studio"],
-    "Video": ["Generation", "Talking Face", "Remove Background", "Studio"],
+    "Image": ["Remove Background", "Compositing"],
+    "Video": ["Generation", "Talking Face", "Remove Background", "Compositing"],
     "Sound": ["Video SFX", "Video Music", "TTS(Voice)", "Image SFX", "Image Music"],
     "Scripts": []
 }
@@ -122,12 +122,20 @@ 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://arxivgpt-vidnuki.hf.space", width=None, height=768, scrolling=True)
 
+# 'Image' 메뉴에서 'Remove Background' 선택 시
+if selected_menu == "Image" and st.session_state['current_sub_menu'] == "Compositing":
+    st.header("Compositing")
+    # iframe을 사용하여 외부 URL 임베드
+    components.iframe("https://arxivgpt-vidistudio.hf.space", width=None, height=768, scrolling=True)
+     
+    
 # 'Home' 페이지 표시
 if selected_menu == "Home":
     st.image("banner2.jpg", use_column_width=True)    
@@ -192,22 +200,6 @@ elif selected_menu == "Free Stock":
                 st.video(video_file)
 
 
-# 'Image' 메뉴 선택 시 처리 로직
-elif selected_menu == "Image":
-    if st.session_state['current_sub_menu'] == "Generation":
-        st.header("Image Generation")
-        st.write("Generate images using AI.")
-    elif st.session_state['current_sub_menu'] == "Face ID":
-        st.header("Face ID")
-        st.write("Identify and analyze faces in images.")
-    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'] == "Studio":
-        st.header("Image Studio")
-        st.write("Access a suite of image editing tools.")
-
 # 'Video' 메뉴 선택 시 처리 로직
 elif selected_menu == "Video":
     if st.session_state['current_sub_menu'] == "Generation":
@@ -219,7 +211,7 @@ elif selected_menu == "Video":
     elif st.session_state['current_sub_menu'] == "Remove Background":
         st.header("Video Background Removal")
         st.write("Remove backgrounds from your videos effortlessly.")
-    elif st.session_state['current_sub_menu'] == "Studio":
+    elif st.session_state['current_sub_menu'] == "Compositing":
         st.header("Video Studio")
         st.write("Explore our video editing studio for advanced editing.")