app.py

import streamlit as st

# 사이드바 타이틀 설정
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_menu = None
selected_sub_menu = None

for menu, sub_menus in menus.items():
    with st.sidebar.expander(menu):
        selected = st.radio("", sub_menus, key=menu)
        if selected:
            selected_menu = menu
            selected_sub_menu = selected

# 'Template Video'가 선택되었을 때 비디오 갤러리를 표시
if selected_menu == "Free Stock" and selected_sub_menu == "Template Video":
    st.subheader("Template Videos")
    
    # 비디오 파일 경로 설정
    video_files = ["ex1.mp4", "ex2.mp4", "ex3.mp4", "ex4.mp4", "ex5.mp4", "ex6.mp4"]
    
    # 갤러리 형태로 비디오 표시 (2열 구성으로 변경하여 넓게 배치)
    cols = st.columns(2)  # 2개의 열 생성으로 변경
    for index, video_file in enumerate(video_files):
        with cols[index % 2]:  # 열 인덱스 조정
            st.video(video_file, use_column_width=True)  # 열 너비 사용 설정