import streamlit as st
from utils.logger import setup_logger

logger = setup_logger(__name__)


def display_memo():
    st.header("Memo")
    current = st.session_state.get('current_upload')
    if current is not None:
        memo = st.session_state['uploads'][current]['memo']
        logger.info(f"Memo: {memo}")

        if memo is not None:
            updated_memo = {}

            for title, content in memo.items():
                st.subheader(f"{title}")
                edited_content = st.data_editor(content)
                st.markdown("---")

                updated_memo[title] = edited_content

            st.session_state['uploads'][current]['memo'] = updated_memo