Update evaluation call

app.py

@@ -3,7 +3,7 @@ import streamlit.components.v1 as components
 import streamlit as st
 import pandas as pd
 import logging
-from deeploy import Client
+from deeploy import Client, CreateEvaluation
 from constants import (
     relationship_dict,
     occupation_dict,
@@ -146,17 +146,22 @@ def hide_expander():
 def show_expander():
     st.session_state.expander_toggle = True
 
-def submit_and_clear(evaluation: str):
-    if evaluation == "yes":
-        st.session_state.evaluation_input["result"] = 0 # Agree with the prediction
+def submit_and_clear(agree: str, comment: str = None):
+    if agree == "yes":
+        evaluation_input: CreateEvaluation = {
+            "agree": True,
+            "comment": comment,
+        }
     else:
         desired_output = not predictions[0]
-        st.session_state.evaluation_input["result"] = 1
-        st.session_state.evaluation_input["value"] = {"predictions": [desired_output]}
+        evaluation_input: CreateEvaluation = {
+            "agree": False,
+            "desired_output": { "predictions": [desired_output] },
+            "comment": comment,
+        }
+
     try:
-        client.evaluate(
-            deployment_id, request_log_id, prediction_log_id, st.session_state.evaluation_input
-        )
+        client.evaluate(st.session_state.deployment_id, prediction_log_id, evaluation_input)
         st.session_state.evaluation_submitted = True
         st.session_state.predict_button_clicked = False
         st.session_state.exp = None
@@ -384,8 +389,6 @@ elif st.session_state.predict_button_clicked and st.session_state.exp is not Non
             )
             st.session_state.evaluation_input = {}
             comment = st.text_input("Your assessment:", placeholder="For example: 'Income is too low, given applicant's background'")
-            if comment:
-                st.session_state.evaluation_input["explanation"] = comment
 
             cols = st.columns(4)
             col_yes, col_no = cols[:2]
@@ -396,7 +399,7 @@ elif st.session_state.predict_button_clicked and st.session_state.exp is not Non
                     use_container_width=True,
                     help="Click if you agree with the prediction",
                     on_click=submit_and_clear,
-                    args=["yes"]
+                    args=["yes", comment]
                 )
                 ChangeButtonColour("Yes, I agree", "white", "green")