Update app.py

app.py

@@ -12,9 +12,16 @@ menus = {
     "Scripts": ["Script"]
 }
 
-# 선택된 메뉴 항목을 저장할 변수
-selected_menu = st.sidebar.selectbox("Menu", list(menus.keys()))
-selected_sub_menu = st.sidebar.selectbox("Sub Menu", menus[selected_menu])
+# 선택된 비디오를 저장할 변수
+selected_video = None
+
+# 메뉴별로 서브메뉴가 분리되어 각각 표시
+for menu, sub_menus in menus.items():
+    with st.sidebar.expander(menu):
+        for sub_menu in sub_menus:
+            if st.button(sub_menu):
+                selected_menu = menu
+                selected_sub_menu = sub_menu
 
 # 'Template Video'가 선택되었을 때 비디오 갤러리를 표시
 if selected_menu == "Free Stock" and selected_sub_menu == "Template Video":
@@ -26,12 +33,11 @@ if selected_menu == "Free Stock" and selected_sub_menu == "Template Video":
     
     # 갤러리 형태로 썸네일 표시
     cols = st.columns(3)
-    for index in range(len(thumbnails)):
+    for index, thumbnail in enumerate(thumbnails):
         with cols[index % 3]:
-            st.image(thumbnails[index], width=100, caption=f"Video {index+1}")
-            if st.button(f"Play Video {index+1}", key=f"video{index+1}"):
-                # 선택된 비디오 재생
-                st.video(video_files[index])
+            if st.image(thumbnail, width=200, caption=f"Video {index+1}", use_column_width=True):
+                selected_video = video_files[index]
 
-# 참고: Streamlit은 현재 이미지 클릭 이벤트를 직접 처리할 수 없으므로,
-# 비디오 재생을 위한 버튼을 제공합니다.
+# 선택된 비디오가 있다면 하단에 비디오 재생
+if selected_video:
+    st.video(selected_video)