| 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",
|
| )
|
|
|
|
|
| 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 |
