Function upload_files

app.py:93–138

()

        pdf_path = os.path.join(base_dir, 'input_pdf', staged_name)
        pdf_file.save(pdf_path)
        write_initial_json(base_dir, staged_name, original_filename)

        return jsonify({
            'status': 'success',
            'message': '文件上传成功',
            'session_id': session_id
        })

    except Exception as e:
        return jsonify({'status': 'error', 'message': str(e)})

@app.route('/download_result/<session_id>')
def download_result(session_id):
    output_folder = os.path.join('data', session_id, 'output_pdf')
    input_folder = os.path.join('data', session_id, 'input_pdf')

    pdf_files = [f for f in os.listdir(output_folder) if f.endswith('.pdf')]
    if not pdf_files:
        return jsonify({'status': 'error', 'message': '未找到输出 PDF 文件'})
    file_path = os.path.join(output_folder, pdf_files[0])

    book_name = "处理结果"
    try:
        json_files = [f for f in os.listdir(input_folder) if f.endswith('.json')]
        if json_files:
            json_path = os.path.join(input_folder, json_files[0])
            with open(json_path, 'r', encoding='utf-8') as f:
                json_data = json.load(f)
            extracted_name = json_data.get('book_name', '')
            if extracted_name:
                book_name = extracted_name
    except Exception as e:
        logger.error(f"读取 JSON 时出错：{e}")

    # 清理文件名中的非法字符，仅保留字母、数字、中文、下划线、连字符和空格
    # Windows/Linux/macOS 通用安全字符集
    safe_book_name = re.sub(r'[<>:"/\\|?*]', '', book_name)
    # 去除首尾空格
    safe_book_name = safe_book_name.strip()
    
    if not safe_book_name:
        safe_book_name = "处理结果"

    # 直接使用书名作为文件名，去掉时间和 TOC 标注
    download_filename = f"{safe_book_name}.pdf"

    response = send_file(file_path, as_attachment=True, download_name=download_filename)