PDFSizeAnalyzer-MCP

import streamlit as st import pandas as pd from PIL import Image import os import tempfile import platform import webbrowser from contextlib import contextmanager from main import ( analyze_pdf_pages, split_pdf_by_chapters, extract_pdf_chapters, split_pdf_by_user_input, convert_pdf_to_images, compress_pdf, merge_pdfs ) # 初始化配置 current_dir = os.path.dirname(os.path.abspath(__file__)) output_dir = os.path.join(current_dir, 'split_pdf_results') os.makedirs(output_dir, exist_ok=True) # 自定义上下文管理器用于临时文件 def create_temp_file(suffix='', prefix='tmp'): fd, path = tempfile.mkstemp(suffix=suffix, prefix=prefix, dir=output_dir) os.close(fd) # 立即关闭文件描述符 return path # 统一文件上传处理 def handle_file_upload(): uploaded_file = st.file_uploader("上传PDF文件", type=["pdf"], key="global_upload") if uploaded_file: original_filename = uploaded_file.name tmp_path = os.path.join(output_dir, original_filename) with open(tmp_path, 'wb') as f: f.write(uploaded_file.read()) st.session_state.tmp_path = tmp_path # 存储到session保持引用 return True return False # 统一资源打开方法 def open_explorer(path): path = os.path.normpath(path) if platform.system() == 'Windows': webbrowser.open(f'file:///{path}') elif platform.system() == 'Darwin': # macOS webbrowser.open(f'file://{path}') else: # Linux webbrowser.open(f'file://{path}') # 页面配置 st.set_page_config(layout="wide", initial_sidebar_state="expanded") st.title("📑 PDF分析工具") st.caption("由于前端js可以在浏览器环境实现数据加工，实现私密文件只在本地浏览器运行，不会上传到服务器，对数据保密性更强，故此，决定将前端能实现的功能全部迁移：") st.caption("前端地址：https://jiheutools-knvf60mx.maozi.io/") st.caption("已迁移功能：分析页数，转换图片，按页码拆分PDF, PDF合并") st.caption("未迁移功能：章节信息提取，按章节拆分PDF，PDF压缩") # 侧边栏导航 with st.sidebar: st.header("功能导航") # 通过功能导航界面隐藏了部分功能，如果本地打包需要，可以展开注释 # 程序中有很多对本地文件的操作，如果程序只在本地运行，可以不受影像，但如果运行在云端，则需要考虑下载问题 # 我的处理思路是io.BytesIO() 内存缓冲区 + zipfile.ZipFile 生成压缩包 + 把内存缓冲区交给 st.download_button，全程不落盘，直接提供下载结果。 # page = st.radio("选择功能", ["分析页数", "章节信息提取", "转换图片", "按页码拆分PDF", "按章节拆分", "PDF压缩", "PDF合并"]) page = st.radio("选择功能", ["章节信息提取"]) # 主内容区 if page == "分析页数": tmp_path = None if handle_file_upload(): tmp_path = st.session_state.tmp_path try: with st.spinner("正在分析PDF结构..."): total_pages, page_info = analyze_pdf_pages(tmp_path) st.subheader(f"📊 基础信息") col1, col2 = st.columns(2) col1.metric("总页数", total_pages) finally: if tmp_path and os.path.exists(tmp_path): try: os.remove(tmp_path) except Exception as e: st.warning(f"删除临时文件时出错: {e}") # 优化图表显示 df_pages = pd.DataFrame(page_info) if not df_pages.empty: format_counts = df_pages['paper_type'].value_counts().reset_index() format_counts.columns = ['纸张类型', '数量'] with st.expander("📐 纸张规格分布"): st.bar_chart(format_counts.set_index('纸张类型')) # 分页显示详细信息 with st.expander("🔍 详细页码分布"): page_chunk = st.empty() for i, (_, row) in enumerate(df_pages.iterrows()): if i > 0 and i % 10 == 0: page_chunk.write("...") if st.button("显示更多", key=f"show_more_{i}"): continue with st.container(): st.write(f"**{row['paper_type']}** ({row['size']})") st.code(row['page_numbers'], language='text') elif page == "转换图片": if handle_file_upload(): with st.spinner("正在转换页面为图片..."): image_paths = convert_pdf_to_images(st.session_state.tmp_path) if image_paths: result_folder = os.path.dirname(image_paths[0]) st.success(f"✅ 转换完成，共生成 {len(image_paths)} 张图片,结果文件夹路径: {result_folder}") open_explorer(result_folder) # 优化图片显示 cols = st.columns(3) for idx, img_path in enumerate(image_paths[:9]): with cols[idx//3]: with Image.open(img_path) as img: st.image(img, use_container_width=True) elif page == "章节信息提取": if handle_file_upload(): with st.spinner("正在解析文档结构..."): chapters = extract_pdf_chapters(st.session_state.tmp_path) if chapters: df_chapters = pd.DataFrame(chapters) # 改进筛选器 level_options = ['全部'] + sorted(df_chapters['level'].unique().astype(str)) selected_level = st.selectbox('选择标题级别', level_options) if selected_level != '全部': df_chapters = df_chapters[df_chapters['level'] <= int(selected_level)] # 优化表格显示 st.subheader("📖 文档结构") format_df = df_chapters.copy() format_df['page_range'] = format_df.apply( lambda x: f"{x.start_page}-{x.end_page}", axis=1) st.dataframe(format_df[['level', 'title', 'page_range']], use_container_width=True, hide_index=True) # 优化JSON显示 with st.expander("📂 章节详情"): st.json(chapters) else: st.warning("⚠️ 未检测到章节信息") elif page == "按页码拆分PDF": if handle_file_upload(): user_input = st.text_input("输入页码范围（格式：1-5,6,7-9）", "1-3") if st.button("开始拆分"): try: with st.spinner("处理中..."): result = split_pdf_by_user_input(st.session_state.tmp_path, user_input) # 改进结果展示 result_dir = os.path.dirname(result[0]) st.success(f"✅ 成功拆分为 {len(result)} 个文件,结果文件夹路径: {result_dir}") open_explorer(result_dir) except ValueError as ve: st.error(f"❌ 输入格式错误：{str(ve)}") except Exception as e: st.error(f"❌ 处理失败：{str(e)}") elif page == "按章节拆分": if handle_file_upload(): with st.spinner("正在解析文档结构..."): chapters = extract_pdf_chapters(st.session_state.tmp_path) if chapters: all_options = [c['title'] for c in chapters] selected = st.multiselect("选择要拆分的章节", all_options) if selected: st.write(f"已选择：{len(selected)} 个项目") for item in selected: st.write(f"- {item}") select_all = st.checkbox("全选") if select_all: selected = [c['title'] for c in chapters] split_button = st.button("开始拆分") if split_button and selected: with st.spinner("正在拆分选定章节..."): try: result = split_pdf_by_chapters(st.session_state.tmp_path, selected) result_dir = os.path.dirname(result[0]) st.success(f"✅ 成功拆分为 {len(result)} 个文件，结果文件夹路径: {result_dir}") open_explorer(result_dir) except Exception as e: st.error(f"❌ 拆分失败：{str(e)}") elif page == "PDF压缩": st.subheader("在线PDF压缩工具（推荐）") st.markdown(""" <div style="display: flex; gap: 10px; margin-bottom: 20px;"> <a href="https://www.ilovepdf.com/zh-cn/compress_pdf" target="_blank"> <button style="padding: 8px 16px; background-color: #2196F3; color: white; border: none; border-radius: 4px; cursor: pointer;"> iLovePDF </button> </a> <a href="https://www.pdf2go.com/zh/compress-pdf" target="_blank"> <button style="padding: 8px 16px; background-color: #FF5722; color: white; border: none; border-radius: 4px; cursor: pointer;"> PDF2Go </button> </a> </div> """, unsafe_allow_html=True) st.info("""**小贴士**: 没有任何一个工具是万能的，各有擅长，请多尝试。""") st.info(""" **小贴士**: 对于ArcGIS等软件导出的矢量PDF，如果常规压缩效果不佳， 可以尝试先将PDF转为图片型PDF再进行压缩，也可考虑设置缩小页面尺寸为A4、A3。 下面本地工具将提供矢量PDF转为图片型PDF的工具，其他操作可通过在线PDF压缩工具实现。 """) if handle_file_upload(): st.subheader("附图类：矢量型PDF转为图片型PDF") quality = st.slider("选择图片质量,数字越大则图片越清晰，同时体积越大", 1, 100, 100) if st.button("转换为图片型PDF"): with st.spinner("正在转换PDF文件..."): try: compressed_path = compress_pdf(st.session_state.tmp_path, quality) st.success(f"✅ 转换完成，文件已保存到: {compressed_path}") st.info("建议将转换后的图片型PDF再用上方在线工具进一步压缩") open_explorer(os.path.dirname(compressed_path)) except Exception as e: st.error(f"❌ 转换失败: {str(e)}") elif page == "PDF合并": st.subheader("📂 PDF文件合并") # 使用列布局 col1, col2 = st.columns(2) with col1: # 多文件上传 uploaded_files = st.file_uploader("上传多个PDF文件", type=["pdf"], accept_multiple_files=True, key="merge_upload") if uploaded_files and len(uploaded_files) > 1: with col2: st.subheader("📂 文件顺序调整") st.write("拖拽文件以调整合并顺序") # 创建可排序的文件列表 files_list = [file.name for file in uploaded_files] # 使用streamlit-sortables组件实现拖拽排序 try: from streamlit_sortables import sort_items sorted_items = sort_items(files_list, direction="vertical") # 根据排序结果重新排列文件 # Create mapping from filename to file object file_map = {file.name: file for file in uploaded_files} sorted_files = [file_map[item] for item in sorted_items] uploaded_files = sorted_files except ImportError: st.warning("⚠️ 未安装streamlit-sortables组件，使用默认顺序") output_name = st.text_input("输入合并后的文件名（无需后缀）", "merged_pdf") if st.button("开始合并"): try: # 保存临时文件 file_paths = [] for uploaded_file in uploaded_files: tmp_path = os.path.join(output_dir, uploaded_file.name) with open(tmp_path, 'wb') as f: f.write(uploaded_file.read()) file_paths.append(tmp_path) output_path = os.path.join(output_dir, f"{output_name}.pdf") with st.spinner("正在合并PDF文件..."): merged_path = merge_pdfs(file_paths, output_path) st.success(f"✅ 合并完成，文件已保存到: {merged_path}") open_explorer(os.path.dirname(merged_path)) except Exception as e: st.error(f"❌ 合并失败: {str(e)}") finally: # 清理临时文件 for tmp_path in file_paths: if os.path.exists(tmp_path): try: os.remove(tmp_path) except Exception as e: st.warning(f"删除临时文件时出错: {e}") elif uploaded_files and len(uploaded_files) == 1: st.warning("⚠️ 请上传至少2个PDF文件进行合并") # 底部状态栏 st.markdown("---") st.caption("🚀 页面缓存可通过 Ctrl+Shift+R 强制刷新") st.caption("🚀 开发者：王帅，集成打包：王帅")