|
|
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)') |
|
|
|
|
|
|
|
|
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() |
