Update app.py

app.py

@@ -1,32 +1,49 @@
 import streamlit as st
+import os
 
 # 사이드바 타이틀 설정
 st.sidebar.title("ViDraft")
 
 # 메뉴 항목과 하위 항목 정의
 menus = {
-    "Free Stock": ["Template Video", "Template Image",  "Search Video", "Search Image"],
+    "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)", "Voice Clone", "Image SFX",  "Image Music"],
+    "Sound": ["Video SFX", "Video Music", "TTS(Voice)", "Voice Clone", "Image SFX", "Image Music"],
     "Scripts": ["Script"]
 }
 
-# 선택된 메뉴 항목을 저장할 변수
-selected_menu = None
-selected_sub_menu = None
+# 선택된 비디오를 저장할 변수
+selected_video = None
 
-# 메뉴 생성
-selected_menu = st.sidebar.selectbox("Menu", list(menus.keys()))
-selected_sub_menu = st.sidebar.selectbox("Sub Menu", menus[selected_menu])
+# 메뉴 생성 및 접이식 메뉴 구현
+for menu in menus.keys():
+    with st.sidebar.expander(menu):
+        for sub_menu in menus[menu]:
+            if st.button(sub_menu, key=sub_menu):
+                selected_menu = menu
+                selected_sub_menu = sub_menu
+                break
 
-# 'Template Video'가 선택되었을 때 'example.mp4' 비디오를 표시
+# 'Template Video'가 선택되었을 때 비디오 갤러리를 표시
 if selected_menu == "Free Stock" and selected_sub_menu == "Template Video":
-    # 비디오 파일의 경로 설정
-    video_path = "example.mp4"  # 스페이스의 루트 디렉토리에 위치한 경우
+    st.subheader("Template Videos")
     
-    # 비디오 제목을 출력
-    st.subheader("Example Video")
+    # 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)]
     
-    # 비디오 파일을 스트림릿을 통해 표시
-    st.video(video_path)
+    # 썸네일을 표시하고 선택 처리
+    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]
+    
+    # 선택된 비디오 재생
+    if selected_video:
+        st.video(selected_video)