import streamlit as st

from apis import get_models


def main():
    st.title("🔍 Wren AI Cloud API Demo - Metadata")

    if "api_key" not in st.session_state or "project_id" not in st.session_state:
        st.error("Please enter your API Key and Project ID in the sidebar of Home page to get started.")
        return
    if not st.session_state.api_key or not st.session_state.project_id:
        st.error("Please enter your API Key and Project ID in the sidebar of Home page to get started.")
        return
    
    api_key = st.session_state.api_key
    project_id = st.session_state.project_id
    
    st.markdown('Using APIs: [Get models](https://wrenai.readme.io/reference/get_projects-projectid-models)')

    # Get and display models
    with st.spinner("🔄 Loading models..."):
        models_response, error = get_models(api_key, project_id)
    
    if models_response:
        models = models_response.get("models", [])
        
        if models:
            st.markdown(f"Here is your latest deployed models in MDL (Model Definition Language) format in project {project_id}.")
            st.json(models, expanded=True)
        else:
            st.info("No models found in this project.")
    else:
        st.error(f"Failed to load models: {error}")


if __name__ == "__main__":
    main()