main.py

import streamlit as st

from emma.streamlit_utils import (
    ensure_infrahub_address_and_branch,
    schema_bootstrap_message,
    set_page_config,
)
from menu import menu

set_page_config(title="Emma - An assistant for Infrahub")

# Empty schema info goes first
schema_bootstrap_message()

# Set columns to receive content
left, right = st.columns([1, 1.6], gap="medium", vertical_alignment="center")

# Left is for Emma avatar
left.image(image="static/emma.png", caption="Hello, I'm Emma")

# Right is for the text
with right:
    st.write("# Welcome! :wave:")
    st.markdown(
        """
        Emma is an agent designed to help you interact with Infrahub.

        Emma will look for the **INFRAHUB_ADDRESS** environment variable to connect to your Infrahub instance.
        However, you can also set or update the address directly in the UI if needed.

        Use the sidebar to navigate between different functionalities, such as exploring data, importing data,
        generating schemas, and visualizing schemas.
        """
    )

    st.info(
        icon="⚠️",
        body="""
            Emma is an experimental and quickly evolving tool developed by the team at OpsMill.

            As such, she may not always work perfectly. Give us feedback if she misbehaves!
            """,
    )

    if "infrahub_address" not in st.session_state:
        st.session_state.infrahub_address = None

    ensure_infrahub_address_and_branch()

menu()