Spaces:
Runtime error
Runtime error
File size: 2,561 Bytes
a90a406 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 |
from langchain_community.document_loaders import PyMuPDFLoader
from langchain.text_splitter import RecursiveCharacterTextSplitter
import re
class Loader:
"""Clase encargada de la carga desde PDFs,
admite PDFs con texto seleccionable unicamente. Realiza
carga y devuelve lista de chunks de texto.
"""
def __init__(self, path: str):
self.path = path
def load_docs(self, pag: slice = None):
"""Carga el PDF y devuelve lista de chunks de texto."""
loader = PyMuPDFLoader(self.path)
docs = loader.load()
if pag:
docs = docs[pag]
return [doc.page_content for doc in docs]
@staticmethod
def limpiar_texto(texto: str) -> str:
"""
Limpia el texto eliminando caracteres basura y normalizando espacios y saltos de línea.
Esta función está diseñada para preprocesar libros u otros documentos largos,
facilitando su uso en aplicaciones de Retrieval Augmented Generation (RAG).
Args:
texto (str): El texto original a limpiar.
Returns:
str: El texto limpio.
"""
# 1. Eliminar saltos de línea, tabulaciones y otros caracteres de control
texto = re.sub(r'[\r\n\t]+', ' ', texto)
# 2. Eliminar caracteres no imprimibles (códigos de control)
texto = re.sub(r'[\x00-\x1F\x7F]', '', texto)
# 3. Sustituir múltiples espacios por uno solo
texto = re.sub(r'\s+', ' ', texto)
# 4. Eliminar caracteres que no sean letras, dígitos o signos de puntuación comunes
# Se conservan letras con acentos y caracteres propios del español.
texto = re.sub(r'[^\w\s.,;:¡!¿?\-áéíóúÁÉÍÓÚñÑ]', '', texto)
# 5. Eliminar espacios al inicio y al final
texto = texto.strip()
return texto
@staticmethod
def splitter(texto, chunk_size, chunk_overlap):
"""
Divide el texto en chunks
Args:
chunk_size (int): Largo del chunk.
chunk_overlap (int): Sobreposición de chunks.
texto (list): lista de textos a procesar.
Returns:
list: Los textos en chunks.
"""
splitter = RecursiveCharacterTextSplitter(
chunk_size=chunk_size,
chunk_overlap=chunk_overlap,
length_function=len,
separators=["\n\n", "\n", " ", ""]
)
chunks = splitter.create_documents(texto)
return chunks
|