app.py

import streamlit as st
import os

# 사이드바 타이틀 설정
st.sidebar.title("ViDraft")

# 메뉴 항목과 하위 항목 정의
menus = {
    "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"],
    "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



# '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)]
    
    # 썸네일을 표시하고 선택 처리
    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)