import streamlit as st

# 페이지 설정
st.set_page_config(page_title="ViDraft", layout="wide")

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

# 라디오 버튼으로 메뉴 항목 정의
menu_items = ["Home", "Free Stock", "Image", "Video", "Sound", "Scripts"]
selected_menu = st.sidebar.radio("Menu", menu_items)

# 메뉴에 따른 하위 메뉴 항목 정의
sub_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": []  # 예시: 'Scripts'에는 하위 메뉴가 없다고 가정
}

# 선택된 메뉴에 따른 하위 메뉴 표시
if selected_menu in sub_menus and sub_menus[selected_menu]:
    selected_sub_menu = st.sidebar.selectbox("Select option", [""] + sub_menus[selected_menu])
else:
    selected_sub_menu = None

# 선택된 메뉴와 하부 메뉴에 따른 내용 표시
if selected_menu == "Home":
    st.title("Welcome to ViDraft")
    st.write("This is the Home page. Select an option from the sidebar to get started.")
elif selected_sub_menu:
    st.subheader(f"{selected_menu} > {selected_sub_menu}")
    # 여기에 선택된 하위 메뉴에 따른 콘텐츠 표시 로직 추가
else:
    st.subheader(selected_menu)
    # 선택된 메뉴에 따른 콘텐츠 표시 로직 추가


    
    # 비디오 파일 경로 설정
    video_files = ["ex1.mp4", "ex2.mp4", "ex3.mp4", "ex4.mp4", "ex5.mp4", "ex6.mp4", "ex7.mp4","ex8.mp4", "ex9.mp4", "ex10.mp4", "ex11.mp4", "ex12.mp4", "ex13.mp4", "ex14.mp4","ex15.mp4", "ex16.mp4", "ex17.mp4", "ex18.mp4"]

    # 갤러리 형태로 비디오 표시
    cols = st.columns(2)  # 2개의 열로 넓게 배치
    for index, video_file in enumerate(video_files):
        with cols[index % 2]:
            st.video(video_file)