import streamlit as st
from pdfsum import 摘要
# 其他功能如有各自py檔也可import

st.set_page_config(page_title="PDF 工具箱", layout="wide")

st.sidebar.title("PDF 工具箱")
api_key = st.sidebar.text_input("請輸入 OpenAI API 金鑰", type="password")
gpt_model = st.sidebar.radio("選擇 GPT 模型", ["gpt-4", "gpt-4.0", "gpt-4.1", "gpt-4.5"])
功能 = st.sidebar.radio("選擇功能", ["文字摘要", "PDF 摘要", "PDF 密碼移除", "論文搜尋", "PDF 合併", "PDF 拆頁", "PDF 轉純文字"])

st.title("PDF 摘要")

if 功能 == "PDF 摘要":
    uploaded_file = st.file_uploader("上傳你的PDF檔案", type=["pdf"])
    if uploaded_file is not None:
        # 讀取PDF檔案內容
        from PyPDF2 import PdfReader
        pdf_reader = PdfReader(uploaded_file)
        內容 = ""
        for page in pdf_reader.pages:
            內容 += page.extract_text() or ""
        if st.button("產生 PDF 摘要"):
            st.info("正在產生摘要，請稍候...")
            result = 摘要(內容)
            st.success(result)
else:
    st.info("請選擇功能")