import streamlit as st from typing import Union, List from application.utils import logger logger = logger.get_logger() PAGE_TITLE = "PDF Extractor" PAGE_LAYOUT = "wide" # PAGE_ICON = "src/frontend/images/page_icon.jpg" # GITHUB_LINK = "https://github.com/Vela-Test1993/yuvabe-care-companion-ai" # ABOUT_US = "An AI-powered assistant for personalized healthcare guidance." 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, # page_icon=PAGE_ICON, layout=PAGE_LAYOUT, initial_sidebar_state="collapsed", # menu_items={ # "Get help": GITHUB_LINK, # "Report a bug": GITHUB_LINK, # "About": ABOUT_US # } ) # 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 = False, ): """ 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 # def extract_text_from_pdf(file) -> str: # """ # Extracts and returns the full text content from a PDF file. # :param file: PDF file object (BytesIO or UploadedFile from Streamlit) # :return: Extracted text as a string # """ # text = "" # with fitz.open(stream=file.read(), filetype="pdf") as doc: # for page in doc: # text += page.get_text() # return text.strip()