feat: exposed forced-ocr-option (#157)

* exposed forced-ocr-option

Signed-off-by: Peter Staar <[email protected]>

* refactored the code

Signed-off-by: Peter Staar <[email protected]>

* moved warning to info

Signed-off-by: Peter Staar <[email protected]>

---------

Signed-off-by: Peter Staar <[email protected]>

Author: PeterStaar-IBM

docling_eval/cli/main.py

@@ -327,6 +327,7 @@ def get_prediction_provider(
     docling_layout_model_spec: Optional[LayoutModelConfig] = None,
     docling_layout_create_orphan_clusters: Optional[bool] = None,
     docling_layout_keep_empty_clusters: Optional[bool] = None,
+    docling_force_full_page_ocr: Optional[bool] = None,
 ):
     pipeline_options: PaginatedPipelineOptions
     """Get the appropriate prediction provider with default settings."""
@@ -339,6 +340,7 @@ def get_prediction_provider(
 
         ocr_options: OcrOptions = ocr_factory.create_options(  # type: ignore
             kind="easyocr",
+            force_full_page_ocr=docling_force_full_page_ocr,
         )
         # Use all CPU cores
         accelerator_options = AcceleratorOptions(
@@ -386,6 +388,7 @@ def get_prediction_provider(
 
         ocr_options: OcrOptions = ocr_factory.create_options(  # type: ignore
             kind="ocrmac",
+            force_full_page_ocr=docling_force_full_page_ocr,
         )
 
         pipeline_options = PdfPipelineOptions(
@@ -415,6 +418,7 @@ def get_prediction_provider(
 
         ocr_options: OcrOptions = ocr_factory.create_options(  # type: ignore
             kind="easyocr",
+            force_full_page_ocr=docling_force_full_page_ocr,
         )
 
         pdf_pipeline_options = PdfPipelineOptions(
@@ -468,7 +472,7 @@ def get_prediction_provider(
                 import mlx_vlm  # type: ignore
 
                 pipeline_options.vlm_options = smoldocling_vlm_mlx_conversion_options
-
+                _log.info("running SmolDocling on MLX!")
             except ImportError:
                 _log.warning(
                     "To run SmolDocling faster, please install mlx-vlm:\n"
@@ -503,7 +507,7 @@ def get_prediction_provider(
                 import mlx_vlm  # type: ignore
 
                 pipeline_options.vlm_options = GRANITEDOCLING_MLX
-
+                _log.info("running GraniteDocling on MLX!")
             except ImportError:
                 _log.warning(
                     "To run SmolDocling faster, please install mlx-vlm:\n"
@@ -1175,6 +1179,10 @@ def create_eval(
     do_table_structure: Annotated[
         bool, typer.Option(help="Include table structure predictions (only Docling)")
     ] = True,
+    docling_force_full_page_ocr: Annotated[
+        bool,
+        typer.Option(help="Force OCR on entire page (only Docling OCR providers)"),
+    ] = False,
 ):
     """Create evaluation dataset from existing ground truth."""
     gt_dir = gt_dir or output_dir / "gt_dataset"
@@ -1218,6 +1226,7 @@ def create_eval(
             docling_layout_model_spec=docling_layout_model_spec_obj,
             docling_layout_create_orphan_clusters=docling_layout_create_orphan_clusters,
             docling_layout_keep_empty_clusters=docling_layout_keep_empty_clusters,
+            docling_force_full_page_ocr=docling_force_full_page_ocr,
         )
 
         # Get the dataset name from the benchmark
@@ -1272,6 +1281,10 @@ def create(
     do_table_structure: Annotated[
         bool, typer.Option(help="Include table structure predictions (only Docling)")
     ] = True,
+    docling_force_full_page_ocr: Annotated[
+        bool,
+        typer.Option(help="Force OCR on entire page (only Docling OCR providers)"),
+    ] = False,
 ):
     """Create both ground truth and evaluation datasets in one step."""
     # First create ground truth
@@ -1301,6 +1314,7 @@ def create(
             do_visualization=do_visualization,
             image_scale_factor=image_scale_factor,
             do_table_structure=do_table_structure,
+            docling_force_full_page_ocr=docling_force_full_page_ocr,
         )
     else:
         _log.info(