from chunkr_ai.models import (
Configuration,
CroppingStrategy,
GenerationConfig,
GenerationStrategy,
SegmentProcessing,
SegmentFormat
)
config = Configuration(
segment_processing=SegmentProcessing(
Caption=GenerationConfig(
crop_image=CroppingStrategy.AUTO,
format=SegmentFormat.MARKDOWN,
strategy=GenerationStrategy.AUTO,
description=False
),
Formula=GenerationConfig(
crop_image=CroppingStrategy.AUTO,
format=SegmentFormat.MARKDOWN,
strategy=GenerationStrategy.LLM,
description=False
),
Footnote=GenerationConfig(
crop_image=CroppingStrategy.AUTO,
format=SegmentFormat.MARKDOWN,
strategy=GenerationStrategy.AUTO,
description=False
),
ListItem=GenerationConfig(
crop_image=CroppingStrategy.AUTO,
format=SegmentFormat.MARKDOWN,
strategy=GenerationStrategy.AUTO,
description=False
),
Page=GenerationConfig(
crop_image=CroppingStrategy.AUTO,
format=SegmentFormat.MARKDOWN,
strategy=GenerationStrategy.AUTO,
description=False
),
PageFooter=GenerationConfig(
crop_image=CroppingStrategy.AUTO,
format=SegmentFormat.MARKDOWN,
strategy=GenerationStrategy.IGNORE,
description=False
),
PageHeader=GenerationConfig(
crop_image=CroppingStrategy.AUTO,
format=SegmentFormat.MARKDOWN,
strategy=GenerationStrategy.IGNORE,
description=False
),
Picture=GenerationConfig(
crop_image=CroppingStrategy.ALL,
format=SegmentFormat.MARKDOWN,
strategy=GenerationStrategy.AUTO,
description=False
),
SectionHeader=GenerationConfig(
crop_image=CroppingStrategy.AUTO,
format=SegmentFormat.MARKDOWN,
strategy=GenerationStrategy.AUTO,
description=False
),
Table=GenerationConfig(
crop_image=CroppingStrategy.AUTO,
format=SegmentFormat.HTML,
strategy=GenerationStrategy.LLM,
description=True
),
Text=GenerationConfig(
crop_image=CroppingStrategy.AUTO,
format=SegmentFormat.MARKDOWN,
strategy=GenerationStrategy.AUTO,
description=False
),
Title=GenerationConfig(
crop_image=CroppingStrategy.AUTO,
format=SegmentFormat.MARKDOWN,
strategy=GenerationStrategy.AUTO,
description=False
)
)
)