import streamlit as st
from typing import Union, List
from src.utils import logger

logger = logger.get_logger()

PAGE_TITLE = "PDF Extractor"
PAGE_LAYOUT = "wide"


def config_homepage(page_title=PAGE_TITLE):
    """
    Configures the Streamlit homepage with essential settings.

    This function sets up the page title, icon, layout, and sidebar state.
    It also defines custom menu items for better navigation.

    Args:
        page_title (str): The title displayed on the browser tab (default is PAGE_TITLE).

    Key Features:
    - Ensures `st.set_page_config()` is called only once to avoid errors.
    - Uses constants for improved maintainability and consistency.
    - Provides links for help, bug reporting, and an 'About' section.

    Example:
        >>> config_homepage("My Custom App")
    """
    if "page_config_set" not in st.session_state:
        st.set_page_config(
            page_title=page_title,
            layout=PAGE_LAYOUT,
            initial_sidebar_state="collapsed",
        )
        # st.session_state.page_config_set = True

def upload_file(
    file_types: Union[str, List[str]] = "pdf",
    label: str = "📤 Upload a file",
    help_text: str = "Upload your file for processing.",
    allow_multiple: bool = True,
):
    """
    Streamlit file uploader widget with options.

    Args:
        file_types (str or list): Allowed file type(s), e.g., "pdf" or ["pdf", "docx"].
        label (str): Label displayed above the uploader.
        help_text (str): Tooltip help text.
        allow_multiple (bool): Allow multiple file uploads.

    Returns:
        Uploaded file(s): A single file object or a list of file objects.
    """
    if isinstance(file_types, str):
        file_types = [file_types]

    uploaded_files = st.file_uploader(
        label=label,
        type=file_types,
        help=help_text,
        accept_multiple_files=allow_multiple
    )

    if st.button("Submit"):
        st.session_state.pdf_file = uploaded_files
        return uploaded_files