src/ui_components/__init__.py

"""
UI Components Module

This module contains UI-related components including styles, visualizations,
and utility functions for the Streamlit application.
"""

from .styles import get_custom_css
from .components import (
    display_chunk_statistics_charts,
    display_chunk_statistics_table
)
from .utils import extract_chunk_statistics
from .visual_documents import (
    display_visual_search_results,
    display_visual_document_statistics,
    display_visual_document_details
)
from .saliency import (
    generate_tile_aware_saliency,
    can_generate_saliency,
    get_saliency_metadata_summary,
    DEFAULT_ALPHA,
    DEFAULT_COLORMAP,
    DEFAULT_THRESHOLD_PERCENTILE
)

__all__ = [
    "get_custom_css",
    "display_chunk_statistics_charts",
    "display_chunk_statistics_table",
    "extract_chunk_statistics",
    "display_visual_search_results",
    "display_visual_document_statistics",
    "display_visual_document_details",
    # Saliency functions
    "generate_tile_aware_saliency",
    "can_generate_saliency",
    "get_saliency_metadata_summary",
    "DEFAULT_ALPHA",
    "DEFAULT_COLORMAP",
    "DEFAULT_THRESHOLD_PERCENTILE"
]