Rearrange folders

src/app.py

@@ -6,10 +6,10 @@
 from fastapi import FastAPI, UploadFile, File
 from fastapi.responses import PlainTextResponse
 
-from analyze_pdf_fast import analyze_pdf_fast
 from catch_exceptions import catch_exceptions
 from configuration import service_logger, XMLS_PATH
-from src.analyze_pdf import analyze_pdf
+from pdf_layout_analysis.run_pdf_layout_analysis import analyze_pdf
+from pdf_layout_analysis.run_pdf_layout_analysis_fast import analyze_pdf_fast
 
 service_logger.info(f"Is PyTorch using GPU: {torch.cuda.is_available()}")
 

src/analyze_pdf.py → src/pdf_layout_analysis/run_pdf_layout_analysis.py

@@ -5,13 +5,13 @@
 from typing import AnyStr
 from data_model.SegmentBox import SegmentBox
 from ditod.VGTTrainer import VGTTrainer
-from get_json_annotations import get_annotations
-from get_model_configuration import get_model_configuration
-from get_most_probable_pdf_segments import get_most_probable_pdf_segments
-from get_reading_orders import get_reading_orders
-from src.PdfImages import PdfImages
+from vgt.get_json_annotations import get_annotations
+from vgt.get_model_configuration import get_model_configuration
+from vgt.get_most_probable_pdf_segments import get_most_probable_pdf_segments
+from vgt.get_reading_orders import get_reading_orders
+from data_model.PdfImages import PdfImages
 from src.configuration import service_logger, JSON_TEST_FILE_PATH, IMAGES_ROOT_PATH
-from src.create_word_grid import create_word_grid, remove_word_grids
+from vgt.create_word_grid import create_word_grid, remove_word_grids
 from detectron2.checkpoint import DetectionCheckpointer
 from detectron2.data.datasets import register_coco_instances
 from detectron2.data import DatasetCatalog

src/analyze_pdf_fast.py → src/pdf_layout_analysis/run_pdf_layout_analysis_fast.py

@@ -4,10 +4,10 @@
 from fast_trainer.ParagraphExtractorTrainer import ParagraphExtractorTrainer
 from fast_trainer.model_configuration import MODEL_CONFIGURATION as PARAGRAPH_EXTRACTION_CONFIGURATION
 from pdf_features.PdfFeatures import PdfFeatures
+from pdf_layout_analysis.run_pdf_layout_analysis import pdf_content_to_pdf_path
 from pdf_tokens_type_trainer.TokenTypeTrainer import TokenTypeTrainer
 from pdf_tokens_type_trainer.ModelConfiguration import ModelConfiguration
 
-from analyze_pdf import pdf_content_to_pdf_path
 from configuration import ROOT_PATH, service_logger
 from data_model.SegmentBox import SegmentBox
 

src/get_json_annotations.py → src/vgt/get_json_annotations.py

@@ -1,7 +1,7 @@
 import json
 from os import makedirs
 from pdf_features.PdfToken import PdfToken
-from PdfImages import PdfImages
+from data_model.PdfImages import PdfImages
 from configuration import DOCLAYNET_TYPE_BY_ID
 from configuration import JSONS_ROOT_PATH, JSON_TEST_FILE_PATH
 

src/get_most_probable_pdf_segments.py → src/vgt/get_most_probable_pdf_segments.py

@@ -9,7 +9,7 @@
 from pdf_features.PdfToken import PdfToken
 from pdf_features.Rectangle import Rectangle
 from pdf_token_type_labels.TokenType import TokenType
-from PdfImages import PdfImages
+from data_model.PdfImages import PdfImages
 from configuration import ROOT_PATH, DOCLAYNET_TYPE_BY_ID
 from data_model.Prediction import Prediction
 

src/get_reading_orders.py → src/vgt/get_reading_orders.py

@@ -3,7 +3,7 @@
 from pdf_features.PdfToken import PdfToken
 from pdf_token_type_labels.TokenType import TokenType
 
-from PdfImages import PdfImages
+from data_model.PdfImages import PdfImages
 
 
 def find_segment_for_token(token: PdfToken, segments: list[PdfSegment], tokens_by_segments):