1

2026-05-17 10:14:14 -06:00
parent d8773b2508
commit 64b3d15b90
61 changed files with 7612 additions and 0 deletions
@@ -0,0 +1,51 @@
+from __future__ import annotations
+
+from pathlib import Path
+
+from backend.app.managers.knowledge_manager import KnowledgeManager
+from backend.app.services.markdown_builder import build_markdown
+from backend.app.services.processors.audio_processor import AudioProcessor
+from backend.app.services.processors.docx_processor import DocxProcessor
+from backend.app.services.processors.pdf_processor import PdfProcessor
+from backend.app.services.processors.text_processor import TextProcessor
+from backend.app.services.processors.video_processor import VideoProcessor
+
+
+class IngestionService:
+    def __init__(self, manager: KnowledgeManager):
+        self.manager = manager
+
+    def process_source(self, source_id: int, use_ocr: bool = False, page_ranges: str | None = None) -> dict:
+        source = self.manager.get_source(source_id)
+        context = self.manager.get_week_context(source["week_id"])
+        path = Path(source["stored_path"])
+        processor = self._processor_for(source["source_type"], use_ocr, page_ranges)
+        processed = processor.process(path)
+        markdown = build_markdown(
+            title=processed.title,
+            body=processed.body,
+            metadata={
+                "subject": context["subject_name"],
+                "subject_slug": context["subject_slug"],
+                "week": context["week_number"],
+                "source_file": source["original_name"],
+                "source_type": source["source_type"],
+                "processor": processed.processor,
+                "page_ranges": page_ranges if source["source_type"] == "pdf" and page_ranges else None,
+                "language": "es",
+            },
+        )
+        return self.manager.create_document(source_id, processed.title, markdown, processor=processed.processor, page_ranges=page_ranges)
+
+    def _processor_for(self, source_type: str, use_ocr: bool, page_ranges: str | None):
+        if source_type == "text":
+            return TextProcessor()
+        if source_type == "docx":
+            return DocxProcessor()
+        if source_type == "pdf":
+            return PdfProcessor(use_ocr=use_ocr, page_ranges=page_ranges)
+        if source_type == "audio":
+            return AudioProcessor()
+        if source_type == "video":
+            return VideoProcessor(self.manager.tmp_dir)
+        raise RuntimeError(f"Tipo de fuente no soportado: {source_type}")