testing a more compact demo

app.py

@@ -23,7 +23,7 @@ random.seed(datetime.now())
 np.random.seed(int(time.time()))
 
 NUMBER_OF_TRIALS = 20
-CLASSIFIER_TAG = "CHM"
+CLASSIFIER_TAG = ""
 explaination_functions = [load_chm_nns, load_knn_nns]
 selected_xai_tool = None
 
@@ -122,7 +122,7 @@ def render_experiment(query):
     class_def = class_descriptions[predicted_wnid]
 
     session_state.is_classifier_correct[query_id] = classifier_predictions[query_id][
-        f"{CLASSIFIER_TAG}-Output"
+        f"{CLASSIFIER_TAG.upper()}-Output"
     ]
 
     ################################### SHOW QUERY and PREDICTION
@@ -142,7 +142,6 @@ def render_experiment(query):
                 use_column_width=True,
             )
 
-        
         default_value = 0
         if query_id in session_state.user_feedback.keys():
             if session_state.user_feedback[query_id] == "Correct":
@@ -176,7 +175,8 @@ def render_experiment(query):
 def render_results():
     user_correct_guess = 0
     for q in session_state.user_feedback.keys():
-        if session_state.is_classifier_correct[q] == session_state.user_feedback[q]:
+        uf = True if session_state.user_feedback[q] == 'Correct' else False
+        if session_state.is_classifier_correct[q] == uf:
             user_correct_guess += 1
 
     st.write(
@@ -336,12 +336,16 @@ def main():
     global app_mode
     global session_state
     global selected_xai_tool
+    global CLASSIFIER_TAG
 
     # Set the session state
     # State Management and General Setup
     st.set_page_config(layout="wide")
     st.title("TASK - 1 - ImageNetREAL")
 
+    # st.write(classifier_predictions.keys())
+    # st.write(classifier_predictions["ILSVRC2012_val_00024646.JPEG"])
+
     options = [
         "Unselected",
         "NOXAI",
@@ -376,11 +380,11 @@ def main():
         st.markdown(f"## SELECTED METHOD ``{session_state.XAI_tool}``")
 
         if session_state.XAI_tool == "NOXAI":
+            CLASSIFIER_TAG = "knn"
             selected_xai_tool = None
-            CLASSIFIER_TAG = "KNN"
         elif session_state.XAI_tool == "KNN":
             selected_xai_tool = load_knn_nns
-            CLASSIFIER_TAG = "KNN"
+            CLASSIFIER_TAG = "knn"
         elif session_state.XAI_tool == "CHM Nearest Neighbors":
             selected_xai_tool = load_chm_nns
             CLASSIFIER_TAG = "CHM"