feedback

app.py

@@ -15,9 +15,11 @@ st.set_page_config(
     menu_items=None,
 )
 
+st.session_state['session_id'] = '123abc'
+
 WANDB_PROJECT = "paper_reader"
 
-weave.init(f"{WANDB_PROJECT}")
+weave_client = weave.init(f"{WANDB_PROJECT}")
 
 st.title("Chat with the Llama 3 paper 💬🦙")
 
@@ -33,16 +35,29 @@ with st.spinner('Loading the RAG pipeline...'):
     if "rag_pipeline" not in st.session_state.keys():
         st.session_state.rag_pipeline = load_rag_pipeline()
 
-    rag_pipeline = st.session_state["rag_pipeline"]
-
-
-def generate_response(query):
-    response = rag_pipeline.predict(query)
-    st.write_stream(response.response_gen)
-
+    rag_pipeline = st.session_state["rag_pipeline"]    
 
 with st.form("my_form"):
     query = st.text_area("Ask your question about the Llama 3 paper here:")
     submitted = st.form_submit_button("Submit")
-    if submitted:
-        generate_response(query)
+
+if submitted:
+    with st.spinner('Generating answer...'):
+        output = rag_pipeline.predict(query)
+        st.session_state["last_output"] = output
+        text = ""
+        for t in output["response"].response_gen:
+            text += t
+        st.session_state["last_text"] = text
+
+    st.write_stream(output["response"].response_gen)
+
+
+if "last_output" in st.session_state:
+    output = st.session_state["last_output"]
+    text = st.session_state["last_text"]
+    st.write(text)
+
+    # use the weave client to retrieve the call and attach feedback
+    st.button(":thumbsup:",   on_click=lambda: weave_client.call(output['call_id']).feedback.add_reaction("👍"), key='up')
+    st.button(":thumbsdown:", on_click=lambda: weave_client.call(output['call_id']).feedback.add_reaction("👎"), key='down')

rag/rag.py

@@ -150,7 +150,7 @@ class SimpleRAGPipeline(weave.Model):
     @weave.op()
     def predict(self, question: str):
         response = self.query_engine.query(question)
-        return response
+        return {"response": response, 'call_id': weave.get_current_call().id}
 
 
 if __name__ == "__main__":