Update experiments.py

experiments.py

@@ -0,0 +1,27 @@
+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
+    )
+
+    datasetName = st.text_input("Dataset name")
+
+    if datasetName:
+        dataset = phoenix_helpers.get_dataset(name=datasetName)
+        # df = dataset = dataset.as_dataframe()
+        eval_res = phoenix_helpers.modelExperiment(st.session_state.selected_model,dataset)
+        st.dataframe(eval_res)