app.py

import gradio as gr
import pandas as pd
import pubmedpdf as pp

def process_csv(file):
    try:
        df = pd.read_csv(file.name)
        if 'PMCID' in df.columns:
            pmcid_list = df['PMCID'].dropna().astype(str).tolist()
            pmcid_str = ','.join(pmcid_list)
            pp.main(pmcid_str)
            return f"成功处理以下 PMCID:\n{pmcid_str}"
        else:
            return "CSV 文件中没有名为 'PMCID' 的列。"
    except Exception as e:
        return f"处理文件时出错: {e}"

gr.Interface(
    fn=process_csv,
    inputs=gr.File(label="上传 CSV 文件"),
    outputs=gr.Textbox(label="处理结果"),
    title="PubMed PDF 下载器",
    description="上传包含 PMCID 列的 CSV 文件，自动下载对应的 PubMed PDF。"
).launch()