Update app.py

app.py

@@ -1,6 +1,44 @@
 import streamlit as st
 import requests
 
+
+# 페이지 설정
+st.set_page_config(page_title="ViDraft", layout="wide")
+
+# 사이드바 타이틀 설정
+st.sidebar.title("ViDraft")
+
+# 메뉴 및 하위 메뉴 정의
+menus = {
+    "Home": [],
+    "Free Stock": ["Template Video", "Template Image", "Search Video", "Search Image"],
+    "Image": ["Generation", "Face ID", "Inpainting", "Remove Background", "Studio"],
+    "Video": ["Generation", "Talking Face", "Remove Background", "Studio"],
+    "Sound": ["Video SFX", "Video Music", "TTS(Voice)", "Image SFX", "Image Music"],
+    "Scripts": []
+}
+
+# 세션 상태 초기화
+if 'current_menu' not in st.session_state:
+    st.session_state['current_menu'] = 'Home'
+if 'current_sub_menu' not in st.session_state:
+    st.session_state['current_sub_menu'] = ''
+
+# 메인 메뉴 선택
+selected_menu = st.sidebar.selectbox("Menu", list(menus.keys()))
+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])
+    st.session_state['current_sub_menu'] = selected_sub_menu
+else:
+    st.session_state['current_sub_menu'] = ''
+
+# 여기서부터 `selected_menu` 변수를 기반으로 조건문을 사용하는 나머지 코드가 올바르게 작동할 것입니다.
+    
 # Pexels API 키 설정
 PEXELS_API_KEY = "5woz23MGx1QrSY0WHFb0BRi29JvbXPu97Hg0xnklYgHUI8G0w23FKH62"
 
@@ -47,39 +85,6 @@ if selected_menu == "Free Stock" and st.session_state['current_sub_menu'] == "Se
 
 
 
-# 페이지 설정
-st.set_page_config(page_title="ViDraft", layout="wide")
-
-# 사이드바 타이틀 설정
-st.sidebar.title("ViDraft")
-
-# 메뉴 및 하위 메뉴 정의
-menus = {
-    "Home": [],
-    "Free Stock": ["Template Video", "Template Image", "Search Video", "Search Image"],
-    "Image": ["Generation", "Face ID", "Inpainting", "Remove Background", "Studio"],
-    "Video": ["Generation", "Talking Face", "Remove Background", "Studio"],
-    "Sound": ["Video SFX", "Video Music", "TTS(Voice)", "Image SFX", "Image Music"],
-    "Scripts": []
-}
-
-# 세션 상태 초기화
-if 'current_menu' not in st.session_state:
-    st.session_state['current_menu'] = 'Home'
-if 'current_sub_menu' not in st.session_state:
-    st.session_state['current_sub_menu'] = ''
-
-# 메인 메뉴 선택
-selected_menu = st.sidebar.selectbox("Menu", list(menus.keys()))
-st.session_state['current_menu'] = selected_menu
-
-# 하위 메뉴 선택 (해당되는 경우)
-if selected_menu in menus:
-    selected_sub_menu = st.sidebar.selectbox("Sub Menu", [""] + menus[selected_menu])
-    st.session_state['current_sub_menu'] = selected_sub_menu
-else:
-    st.session_state['current_sub_menu'] = ''
-
 # 'Home' 페이지 표시
 if selected_menu == "Home":
     st.header("Welcome to ViDraft")