fix llm_app and add user_id

Author: jeffkayne

examples/streamlit/llm_app.py

@@ -6,6 +6,8 @@
 
 if "response" not in st.session_state:
     st.session_state.response = ""
+if "feedback_key" not in st.session_state:
+    st.session_state.feedback_key = 0
 if "logged_prompt" not in st.session_state:
     st.session_state.logged_prompt = ""
 
@@ -15,13 +17,17 @@
     email, password = trubrics_config()
 
 if email and password:
-    collector = FeedbackCollector(
-        project="default",
-        email=email,
-        password=password,
-    )
+    try:
+        collector = FeedbackCollector(email=email, password=password, project="default")
+    except Exception:
+        st.error(f"Error authenticating '{email}' with [Trubrics](https://trubrics.streamlit.app/). Please try again.")
+        st.stop()
 else:
-    st.warning("To save some feedback to Trubrics, add your account details in the sidebar.")
+    st.info(
+        "To ask a question to an LLM and save your feedback to Trubrics, add your email and password in the sidebar."
+        " Don't have an account yet? Create one for free [here](https://trubrics.streamlit.app/)!"
+    )
+    st.stop()
 
 models = ("gpt-3.5-turbo",)
 model = st.selectbox(
@@ -32,7 +38,8 @@
 
 openai.api_key = st.secrets.get("OPENAI_API_KEY")
 if openai.api_key is None:
-    raise ValueError("OpenAI key is missing. Set OPENAI_API_KEY in st.secrets")
+    st.info("Please add your OpenAI API key to continue.")
+    st.stop()
 
 prompt = st.text_area(label="Prompt", label_visibility="collapsed", placeholder="What would you like to know?")
 button = st.button(f"Ask {model}")
@@ -41,9 +48,10 @@
     response = openai.ChatCompletion.create(model=model, messages=[{"role": "user", "content": prompt}])
     response_text = response.choices[0].message["content"]
     st.session_state.logged_prompt = collector.log_prompt(
-        config_model={"model": model}, prompt=prompt, generation=response_text, tags=["llm_app.py"]
+        config_model={"model": model}, prompt=prompt, generation=response_text, tags=["llm_app.py"], user_id=email
     )
     st.session_state.response = response_text
+    st.session_state.feedback_key += 1
 
 if st.session_state.response:
     st.markdown(f"#### :violet[{st.session_state.response}]")
@@ -55,8 +63,9 @@
         prompt_id=st.session_state.logged_prompt.id,
         model=model,
         align="flex-start",
-        single_submit=False,
         tags=["llm_app.py"],
+        key=f"feedback_{st.session_state.feedback_key}",  # overwrite with new key
+        user_id=email,
     )
 
     if feedback:

examples/streamlit/llm_chatbot.py

@@ -25,7 +25,7 @@ def init_trubrics(email, password):
         collector = FeedbackCollector(email=email, password=password, project="default")
         return collector
     except Exception:
-        st.error(f"Error authenticating '{email}'. Please try again.")
+        st.error(f"Error authenticating '{email}' with [Trubrics](https://trubrics.streamlit.app/). Please try again.")
         st.stop()
 
 
@@ -67,6 +67,7 @@ def init_trubrics(email, password):
                 key=feedback_key,
                 disable_with_score=disable_with_score,
                 prompt_id=st.session_state.prompt_ids[int(n / 2) - 1],
+                user_id=email,
             )
             if feedback:
                 trubrics_successful_feedback(feedback)
@@ -90,6 +91,7 @@ def init_trubrics(email, password):
             generation=generation,
             session_id=st.session_state.session_id,
             tags=["llm_chatbot.py"],
+            user_id=email,
         )
         st.session_state.prompt_ids.append(logged_prompt.id)
         messages.append({"role": "assistant", "content": generation})
@@ -99,6 +101,7 @@ def init_trubrics(email, password):
         **feedback_kwargs,
         key=f"feedback_{int(len(messages)/2)}",
         prompt_id=st.session_state.prompt_ids[int(len(messages) / 2) - 1],
+        user_id=email,
     )
     if feedback:
         trubrics_successful_feedback(feedback)