import streamlit as st

from ners.web.interfaces.predictions import Predictions

st.set_page_config(page_title="Predictions", page_icon="🔮", layout="wide")

if "config" in st.session_state:
    predictions = Predictions(
        st.session_state.config,
        st.session_state.experiment_tracker,
        st.session_state.experiment_runner,
    )
    predictions.index()
else:
    st.error("Please run the main app first to initialize the configuration.")
    st.markdown("Go back to the [main page](/) to start the application.")