在Streamlit应用中高效展示同一文件夹下的多个GIF图像

import streamlit as st
import base64
import os
import glob # 用于更方便地查找文件
import platform # 用于检测操作系统

sys_platform = platform.system()
work_dir = os.path.dirname(os.path.abspath(__file__)) # 获取当前脚本的绝对路径

# 根据操作系统确定路径分隔符
if sys_platform == "Windows":
    gif_folder_path = os.path.join(work_dir, "forecast", "fr")
else:
    gif_folder_path = os.path.join(work_dir, "forecast", "fr") # 在Unix/Linux/macOS上，路径分隔符会自动处理

# 检查目录是否存在，如果不存在则创建（可选，取决于您的需求）
if not os.path.exists(gif_folder_path):
    st.error(f"GIF文件夹不存在: {gif_folder_path}")
    st.stop() # 停止应用运行

# 切换到GIF文件目录
os.chdir(gif_folder_path)

# 使用glob查找所有GIF文件
gif_files = glob.glob("*.gif")

if not gif_files:
    st.warning(f"在目录 '{gif_folder_path}' 中未找到任何GIF文件。")
    st.stop()

st.markdown("### 本地GIF文件展示", unsafe_allow_html=True)

for gif_filename in gif_files:
    # 重新构建完整路径，因为os.chdir后，glob返回的是文件名
    full_gif_path = os.path.join(gif_folder_path, gif_filename)

    with open(full_gif_path, "rb") as file_:
        contents = file_.read()
        data_url = base64.b64encode(contents).decode("utf-8")

    st.markdown(
        f'<img src="data:image/gif;base64,{data_url}" alt="{gif_filename}" style="max-width:100%; height:auto;">',
        unsafe_allow_html=True,
    )
    st.write(f"--- {gif_filename} ---") # 可选：显示文件名或分隔符

import streamlit as st
import base64
import os
import glob
import platform

def display_multiple_gifs():
    st.title("Streamlit 应用中的多GIF展示")

    sys_platform = platform.system()
    work_dir = os.path.dirname(os.path.abspath(__file__))

    # 假设GIF文件位于脚本同级目录下的 'forecast/fr/' 文件夹
    # 请根据您的实际路径进行调整
    gif_relative_path = os.path.join("forecast", "fr")
    gif_folder_path = os.path.join(work_dir, gif_relative_path)

    st.write(f"正在扫描GIF目录: `{gif_folder_path}`")

    # 检查目录是否存在
    if not os.path.exists(gif_folder_path):
        st.error(f"指定的GIF文件夹不存在: {gif_folder_path}")
        st.stop()

    # 切换到GIF文件目录，以便glob能正确找到文件
    # 记录原始工作目录，以便之后恢复（如果需要）
    original_cwd = os.getcwd()
    try:
        os.chdir(gif_folder_path)
        gif_files = glob.glob("*.gif")
    finally:
        # 确保无论发生什么，都尝试恢复到原始工作目录
        os.chdir(original_cwd)

    if not gif_files:
        st.warning(f"在目录 '{gif_folder_path}' 中未找到任何GIF文件。")
        st.stop()

    st.markdown("### 本地GIF文件列表", unsafe_allow_html=True)

    for gif_filename in gif_files:
        # 再次构建完整路径，因为glob返回的是文件名，且我们已恢复了cwd
        full_gif_path = os.path.join(gif_folder_path, gif_filename)

        try:
            with open(full_gif_path, "rb") as file_:
                contents = file_.read()
                data_url = base64.b64encode(contents).decode("utf-8")

            st.markdown(
                f'<img src="data:image/gif;base64,{data_url}" alt="{gif_filename}" style="max-width:100%; height:auto;">',
                unsafe_allow_html=True,
            )
            st.write(f"--- 显示: `{gif_filename}` ---")
        except Exception as e:
            st.error(f"无法加载或显示GIF文件 '{gif_filename}': {e}")

if __name__ == "__main__":
    display_multiple_gifs()