于大爷.在线工具集-9级Word标题一键生成！自己怒写“word带9级标题格式的多级编号在线自动生成工具”，彻底解决word多级编号痛点

import streamlit as st
from docx import Document
import io
import os

# 设置页面配置
st.set_page_config(
    page_title="word带9级标题格式的多级编号在线自动生成工具",
    page_icon="📝",
    layout="centered",
    initial_sidebar_state="collapsed"
)

# 定义标题模板字典
titleDict={
    "(一)1(1)圈1半1A.a.a)i.":'''\n        (一)\n            1\n                (1)\n                    ➀\n                        1)\n                            A.\n                                a.\n                                    a)\n                                        i.\n    ''',
    # 其他模板...
}

# 标题和介绍
st.title("📝 word带9级标题格式的多级编号在线自动生成工具")
st.markdown("通过选择不同的多级编号格式模板，快速生成带编号带标题格式的Word标题文档，直接下载使用，再也不用麻烦多级编号的问题。")

# 配置区域
st.subheader("🔧 配置选项")

# 第一行：模板选择和标题输入
col_template, col_title = st.columns([1, 1])

with col_template:
    # 选择模板
    selected_template_key = st.selectbox(
        "选择多级编号格式模板",
        list(titleDict.keys()),
        format_func=lambda x: x
    )
    # 显示选择的模板对应的标题层级预览
    st.text_area("标题层级预览", titleDict[selected_template_key], height=150)

# 获取对应的模板文件名
selected_template = selected_template_key + ".docx"

with col_title:
    # 输入标题
    title_text = st.text_input(
        "输入标题文本",
        value="文档标题",
        placeholder="请输入要生成的标题文本"
    )

# 第二行：生成按钮
generate_btn = st.button("生成Word文档", type="primary", use_container_width=True)

if generate_btn:
    if not title_text.strip():
        st.error("请输入有效的标题文本")
    else:
        try:
            # 显示加载状态
            with st.spinner("正在生成文档..."):
                # 加载模板
                template_path = os.path.join(TEMPLATE_DIR, selected_template)
                doc = Document(template_path)

                # 清空文档内容（保留样式）
                for para in doc.paragraphs:
                    if para.text.strip():
                        para.text = ""

                # 生成9级标题
                for i in range(9):
                    level = i + 1
                    heading_text = f"{title_text} {level}"
                    doc.add_paragraph(heading_text, style=f"Heading {level}")

                # 保存到内存缓冲区
                buffer = io.BytesIO()
                doc.save(buffer)
                buffer.seek(0)

                # 提供下载
                st.success("文档生成成功！")
                st.download_button(
                    label="下载Word文档",
                    data=buffer,
                    file_name=f"{title_text}_多级标题.docx",
                    mime="application/vnd.openxmlformats-officedocument.wordprocessingml.document",
                    type="primary"
                )
        except Exception as e:
            st.error(f"生成文档时出错: {str(e)}")