import streamlit as st
import sys
from pathlib import Path

# Add project root to Python path for imports
sys.path.insert(0, str(Path(__file__).parent.parent))

from frontend.sidebar import render_sidebar
from frontend.editor import render_editor
from frontend.chat import render_chat
from frontend.state import init_state

init_state()


def main():
    st.set_page_config(page_title="Lightweight code editor", layout="wide")

    st.markdown(
        """
        <style>
        .block-container { padding-top: 1rem; }
        [data-testid="stSidebarContent"] { padding-top: 0rem; margin-top: -2rem; }
        </style>
    """,
        unsafe_allow_html=True,
    )

    st.title("Lightweight code editor")

    init_state()

    render_sidebar()

    if st.session_state.get("radio_interface_options") == "Code Editor":
        render_editor()
    elif st.session_state.get("radio_interface_options") == "Chat with AI Assistant":
        render_chat()

if __name__ == "__main__":
    main()