Spaces:
Build error
Build error
File size: 1,402 Bytes
e23631f |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 |
import streamlit as st
from dotenv import load_dotenv
import phoenix_helpers
import helpers
load_dotenv()
models = helpers.fetch_models()
if models:
st.subheader("Select a Model")
if "selected_model" not in st.session_state:
st.session_state.selected_model = models[0]
st.selectbox(
"Choose a model to use:",
models, key = 'selected_model',
index=models.index(st.session_state.selected_model) if st.session_state.selected_model in models else 0
)
if st.session_state.selected_model:
st.subheader("Enter a Prompt")
if "prompt" not in st.session_state:
st.session_state.prompt = ""
st.session_state.prompt = st.text_area("Enter your prompt:", value=st.session_state.prompt)
if st.button("Generate Content"):
if st.session_state.prompt:
st.subheader("Model Output")
st.session_state.generated_content = helpers.generate_content(st.session_state.selected_model, st.session_state.prompt)
st.write(st.session_state.generated_content)
st.session_state.spans_df = phoenix_helpers.get_spans_df()
# print(spans_df)
st.dataframe(st.session_state.spans_df)
st.session_state.prompt = ""
else:
st.write("Enter something to generate content.")
|