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_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])

# 메뉴 선택에 따른 동작 구현
if selected_menu == "Free Stock" and selected_sub_menu == "Template Video":
    # 비디오 파일이 있는 디렉토리 경로 설정
    video_directory = "/path/to/your/videos"  # 경로를 실제 비디오 파일이 있는 경로로 변경하세요.
    
    # 디렉토리 내의 모든 파일 목록을 가져옴
    files = os.listdir(video_directory)
    
    # MP4 파일만 필터링
    video_files = [file for file in files if file.endswith(".mp4")]
    
    # 각 비디오 파일에 대해 스트림릿 비디오 위젯 생성
    for video_file in video_files:
        # 파일의 전체 경로 생성
        video_path = os.path.join(video_directory, video_file)
        
        # 비디오 제목을 출력
        st.subheader(video_file)
        
        # 비디오 파일을 스트림릿을 통해 표시
        st.video(video_path)