mirror of
https://github.com/linyqh/NarratoAI.git
synced 2025-12-12 19:52:48 +00:00
35 lines
1.2 KiB
Python
35 lines
1.2 KiB
Python
import streamlit as st
|
||
import os
|
||
import glob
|
||
from app.utils import utils
|
||
|
||
def get_fonts_cache(font_dir):
|
||
if 'fonts_cache' not in st.session_state:
|
||
fonts = []
|
||
for root, dirs, files in os.walk(font_dir):
|
||
for file in files:
|
||
# 支持常见字体格式，少字体时也能被UI识别
|
||
if file.lower().endswith((".ttf", ".ttc", ".otf")):
|
||
fonts.append(file)
|
||
fonts.sort()
|
||
st.session_state['fonts_cache'] = fonts
|
||
return st.session_state['fonts_cache']
|
||
|
||
def get_video_files_cache():
|
||
if 'video_files_cache' not in st.session_state:
|
||
video_files = []
|
||
for suffix in ["*.mp4", "*.mov", "*.avi", "*.mkv"]:
|
||
video_files.extend(glob.glob(os.path.join(utils.video_dir(), suffix)))
|
||
st.session_state['video_files_cache'] = video_files[::-1]
|
||
return st.session_state['video_files_cache']
|
||
|
||
def get_songs_cache(song_dir):
|
||
if 'songs_cache' not in st.session_state:
|
||
songs = []
|
||
for root, dirs, files in os.walk(song_dir):
|
||
for file in files:
|
||
if file.endswith(".mp3"):
|
||
songs.append(file)
|
||
st.session_state['songs_cache'] = songs
|
||
return st.session_state['songs_cache']
|