Update app.py

app.py

@@ -1,59 +1,44 @@
 import streamlit as st
+import os
 
-# 사이드바 타이틀
-st.sidebar.title("접이식 메뉴")
+# 사이드바 타이틀 설정
+st.sidebar.title("ViDraft")
 
 # 메뉴 항목과 하위 항목 정의
 menus = {
-    "Display": ["Display text", "Display interactive widgets", "Display data", "Display media", "Display code", "Display progress and status"],
-    "Data": ["Connect to data sources", "Mutate data", "Placeholders, help, and options"],
-    "Control": ["Optimize performance", "Cache global resources", "Deprecated caching"],
-    "Layout": ["Columns", "Tabs", "Control flow"],
-    "Interactivity": ["Build chat-based apps", "Personalize apps for users"],
+    "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 in menus:
-    with st.sidebar.expander(menu):
-        for sub_menu in menus[menu]:
-            if st.button(sub_menu, key=sub_menu):  # 고유한 key를 제공하여 각 버튼을 구별
-                selected_menu = sub_menu
-                break
+# 메뉴 생성
+selected_menu = st.sidebar.selectbox("Menu", list(menus.keys()))
+selected_sub_menu = st.sidebar.selectbox("Sub Menu", menus[selected_menu])
 
-# 선택된 메뉴에 따른 동작 구현
-if 'selected_menu' in locals():
-    if selected_menu == "Display interactive widgets":
-        # 슬라이더 위젯을 사용하고 반환 값을 slider_val 변수에 저장
-        slider_val = st.slider('Slide me', min_value=0, max_value=10)
-        # 저장된 slider_val 변수 값을 사용하여 출력
-        st.write(f"슬라이더의 값: {slider_val}")
-
-        st.text('Fixed width text')
-        st.markdown('_Markdown_')  # see #*
-        st.caption('Balloons. Hundreds of them...')
-        st.latex(r''' e^{i\pi} + 1 = 0 ''')
-        st.write('Most objects')  # df, err, func, keras!
-        st.write(['st', 'is <', 3])  # see *
-        st.title('My title')
-        st.header('My header')
-        st.subheader('My sub')
-        st.code('for i in range(8): foo()')
-
-    elif selected_menu == "Display interactive widgets":
-        # Interactive widgets
-        button_clicked = st.button('Hit me')
-        checkbox_checked = st.checkbox('Check me out')
-        radio_option = st.radio('Pick one:', ['nose', 'ear'])
-        selectbox_option = st.selectbox('Select', [1, 2, 3])
-        multiselect_options = st.multiselect('Multiselect', [1, 2, 3])
-        slider_value = st.slider('Slide me', min_value=0, max_value=10)
-        select_slider_option = st.select_slider('Slide to select', options=[1, '2'])
-        text_input = st.text_input('Enter some text')
-        number_input = st.number_input('Enter a number')
-        text_area = st.text_area('Area for textual entry')
-        date_input = st.date_input('Date input')
-        time_input = st.time_input('Time entry')
-        file_uploader = st.file_uploader('File uploader')
-        color_picker = st.color_picker('Pick a color')
+# 메뉴 선택에 따른 동작 구현
+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)