Update app.py

app.py

@@ -1,5 +1,4 @@
 import streamlit as st
-import os
 
 # 사이드바 타이틀 설정
 st.sidebar.title("ViDraft")
@@ -13,40 +12,23 @@ menus = {
     "Scripts": ["Script"]
 }
 
-# 선택된 비디오를 저장할 변수
-selected_video = None
-
-# 메뉴 생성 및 접이식 메뉴 구현
-for menu in menus.keys():
-    with st.sidebar.expander(menu):
-        for sub_menu in menus[menu]:
-            # menu와 sub_menu를 결합하여 고유한 key 생성
-            unique_key = f"{menu}_{sub_menu}"
-            if st.button(sub_menu, key=unique_key):
-                selected_menu = menu
-                selected_sub_menu = sub_menu
-                break
-
+# 선택된 메뉴 항목을 저장할 변수
+selected_menu = None
+selected_sub_menu = None
 
+# 메뉴 생성
+selected_menu = st.sidebar.selectbox("Menu", list(menus.keys()))
+selected_sub_menu = st.sidebar.selectbox("Sub Menu", menus[selected_menu])
 
 # 'Template Video'가 선택되었을 때 비디오 갤러리를 표시
 if selected_menu == "Free Stock" and selected_sub_menu == "Template Video":
     st.subheader("Template Videos")
     
-    # 10개의 비디오 썸네일과 파일 경로 정의
-    video_paths = [f"example{i}.mp4" for i in range(1, 11)]
-    thumbnail_paths = [f"thumbnail{i}.jpg" for i in range(1, 11)]
+    # 비디오 파일 경로 설정
+    video_files = [f"ex{i}.mp4" for i in range(1, 7)]
     
-    # 썸네일을 표시하고 선택 처리
-    cols = st.columns(5)
-    for index, col in enumerate(cols):
-        with col:
-            if index < len(thumbnail_paths):
-                # 썸네일 이미지 표시
-                st.image(thumbnail_paths[index], width=100)
-                # 썸네일 클릭 처리 (여기서는 단순히 버튼으로 구현)
-                if st.button(f"Play Video {index+1}", key=f"video{index+1}"):
-                    selected_video = video_paths[index]
+    # 선택된 비디오를 저장할 변수
+    selected_video = st.radio("Select a video:", video_files)
     
     # 선택된 비디오 재생
     if selected_video: