dsProject/dsRag/Milvus/X3_SplitTxt.py

import os

from Util.SplitDocxUtil import SplitDocxUtil


def split_into_blocks(text):
    """按行遍历文本，发现'问题X'或'话题X'时开始分割"""
    blocks = []
    current_block = []
    in_block = False
    
    for line in text.splitlines():
        if line.startswith(('问题', '话题')) and any(c.isdigit() for c in line[:5]):
            if in_block:
                blocks.append('\n'.join(current_block))
                current_block = []
            in_block = True
        
        if in_block:
            current_block.append(line)
    
    if current_block:
        blocks.append('\n'.join(current_block))
    
    return [(i+1, block) for i, block in enumerate(blocks)]

def process_document(input_path, output_dir):
    """处理文档主函数"""
    text = SplitDocxUtil.read_docx(input_path)
    if not text:
        print("无法读取输入文件内容")
        return False
    
    # 确保输出目录存在并清空目录
    if os.path.exists(output_dir):
        for file in os.listdir(output_dir):
            os.remove(os.path.join(output_dir, file))
    os.makedirs(output_dir, exist_ok=True)

    chunks = split_into_blocks(text)
    print(f"共分割出{len(chunks)}个段落块")

    saved_count = 0
    for chunk_num, chunk in chunks:
        chunk = chunk.strip()  # 确保去除空白字符
        output_file = os.path.join(output_dir, f"{chunk_num}.txt")
        if save_to_txt(chunk, output_file, mode='w'):
            saved_count += 1

    print(f"处理完成，共保存{saved_count}个文件到目录: {output_dir}")
    return saved_count > 0

# 保留原有的save_to_txt函数
def save_to_txt(content, file_path, mode='w'):
    """将内容保存到文本文件"""
    try:
        with open(file_path, mode, encoding='utf-8') as f:
            f.write(content)
        return True
    except Exception as e:
        print(f"保存文件{file_path}时出错: {str(e)}")
        return False

if __name__ == "__main__":
    #input_file = '../static/Txt/小学数学教学中的若干问题.docx'
    input_file = '../static/Txt/小学数学知识点.docx'
    output_dir = '../Txt/processed_chunks'
    process_document(input_file, output_dir)