Streamlit应用中多标签PDF文件的高效管理与预览

pip install streamlit-pdf-viewer

import streamlit as st
from streamlit_pdf_viewer import pdf_viewer
import os

st.title("PDF查看器示例")

# 示例1: 通过文件路径显示PDF
# 假设你有一个名为 'example.pdf' 的文件在当前目录下
# if os.path.exists("example.pdf"):
#     st.subheader("通过文件路径显示")
#     pdf_viewer("example.pdf")
# else:
#     st.warning("请确保当前目录下存在 'example.pdf' 文件以查看此示例。")

# 示例2: 通过上传文件并获取其字节内容显示PDF
uploaded_file = st.file_uploader("上传一个PDF文件", type=["pdf"])

if uploaded_file is not None:
    st.subheader("通过字节内容显示")
    pdf_bytes = uploaded_file.read()
    pdf_viewer(pdf_bytes)
else:
    st.info("请上传一个PDF文件以在下方显示。")

import streamlit as st
from streamlit_pdf_viewer import pdf_viewer
from io import BytesIO

st.set_page_config(layout="wide") # 设置页面布局为宽屏以获得更好的显示效果
st.title("多标签PDF文件预览器")

st.markdown("""
    本应用允许您上传多个PDF文件，并以动态标签页的形式展示每个文件，
    方便您在不同文档之间进行切换查看。
""")

# 1. 获取上传的PDF文件
uploaded_files = st.file_uploader(
    "请上传一个或多个PDF文件",
    type=["pdf"],
    accept_multiple_files=True,
    help="您可以拖拽多个PDF文件到此处"
)

# 2. 存储文件名和文件内容（字节）的字典
# 键为文件名，值为文件字节内容
pdf_data = {}

if uploaded_files:
    for uploaded_file in uploaded_files:
        # 读取文件内容并存储
        pdf_data[uploaded_file.name] = uploaded_file.read()

    # 3. 动态创建标签页
    # st.tabs 需要一个标签名称列表
    # 如果没有文件，则不创建标签页
    if pdf_data:
        # 使用with语句创建标签页上下文
        # selected_tab 变量将存储当前被选中标签的名称
        tabs = st.tabs(list(pdf_data.keys()))

        # 4. 在每个标签页中渲染对应的PDF文件
        for i, tab_name in enumerate(pdf_data.keys()):
            with tabs[i]: # 进入当前标签页的上下文
                st.subheader(f"正在显示：{tab_name}")
                # 调用 pdf_viewer 函数，传入对应的PDF字节内容
                pdf_viewer(pdf_data[tab_name], width=800, height=700) # 可以调整宽度和高度
    else:
        st.info("请上传PDF文件以在上方标签页中显示。")
else:
    st.info("等待您上传PDF文件...")