Update app.py

app.py

@@ -16,27 +16,22 @@ menus = {
     "Scripts": []
 }
 
-# 사용자가 메인 메뉴를 선택할 수 있도록 selectbox 위젯으로 메인 메뉴를 표시하고 선택된 메뉴를 저장
-selected_menu = st.sidebar.selectbox("Select a Menu", list(menus.keys()))
-
-# 모든 메뉴 및 하위 메뉴를 펼쳐서 보여주기 위한 로직
-for menu_name, sub_menus in menus.items():
-    with st.sidebar.expander(menu_name):
-        if sub_menus:  # 하위 메뉴가 있는 경우
-            for sub_menu in sub_menus:
-                # 하위 메뉴 이름으로 버튼 생성, 고유한 키 제공
-                button_key = f"{menu_name}-{sub_menu}"
-                st.button(sub_menu, key=button_key)  
-        else:
-            st.write("No sub menus")  # 하위 메뉴가 없는 경우 메시지 표시
-
-# 사용자가 선택한 메뉴에 따라 특정 화면을 표시
-if selected_menu == "Home":
-    st.header("Welcome to ViDraft")
-    st.write("Choose an option from the sidebar to get started.")
-# 여기서부터 추가적인 메뉴에 따른 화면 표시 로직을 구현할 수 있습니다.
+# 세션 상태 초기화
+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":
@@ -123,4 +118,4 @@ elif selected_menu == "Sound":
 # 'Scripts' 메뉴 선택 시 처리 로직
 elif selected_menu == "Scripts":
     st.header("Scripts")
-    st.write("Script writing tools and resources.")
+    st.write("Script writing tools and resources.")