L154 give user ability to run ai judge

app.py

@@ -46,6 +46,10 @@ st.sidebar.success(
 )
 st.sidebar.warning("Select a website first!")
 option = st.sidebar.selectbox("Which website do you want to ask?", ("YSA", "Larkin"))
+st.sidebar.warning("Runnning AI Judge takes longer.")
+run_ai_judge = st.sidebar.selectbox(
+    "Shall we run AI Judge to provide additional scores?", ("Yes", "No")
+)
 special_threshold = st.sidebar.number_input(
     "Insert a number", value=0.2, placeholder="Type a number..."
 )  # 0.3
@@ -147,16 +151,17 @@ if prompt := st.chat_input(initial_input):
         final_ref = final_ref.reset_index()
 
         # add ai judge as additional rating
-        independent_ai_judge_score = []
-        for i in range(final_ref.shape[0]):
-            this_content = final_ref["answers"][i]
-            if len(this_content) > 3:
-                this_score = calculate_sts_openai_score(question, this_content)
-            else:
-                this_score = 0
-            independent_ai_judge_score.append(this_score)
-
-        final_ref["ai_judge"] = independent_ai_judge_score
+        if run_ai_judge == "Yes":
+            independent_ai_judge_score = []
+            for i in range(final_ref.shape[0]):
+                this_content = final_ref["answers"][i]
+                if len(this_content) > 3:
+                    this_score = calculate_sts_openai_score(question, this_content)
+                else:
+                    this_score = 0
+                independent_ai_judge_score.append(this_score)
+
+            final_ref["ai_judge"] = independent_ai_judge_score
 
         engineered_prompt = f"""
             Based on the context: {ref_from_db_search}