x

main.py

@@ -180,9 +180,7 @@ def benchmark_model_multithreaded(model_name, questions, open_router_key, openai
                 st.write(f"**Novelty Score:** {result['novelty_score']}")
                 results.extend(result["results"])  # Add results here
                 novelty_score += result["novelty_score"]  # Update novelty score
-                st.write(
-                    f"<span style='color:yellow'>Total novelty score across all questions (so far): {novelty_score}</span>",
-                    unsafe_allow_html=True)
+                st.warning(f"Total novelty score across all questions (so far): {novelty_score}")
 
             elif result["type"] == "summary":
                 st.write(f"<span style='color:blue'>Total novelty score for question '{result['question']}': {result['total_novelty']}</span>",
@@ -194,8 +192,7 @@ def benchmark_model_multithreaded(model_name, questions, open_router_key, openai
                          unsafe_allow_html=True)
 
 
-    st.write(f"<span style='color:yellow'>Final total novelty score across all questions: {novelty_score}</span>",
-             unsafe_allow_html=True)
+    st.warning(f"Final total novelty score across all questions: {novelty_score}")
     return results
 
 
@@ -213,9 +210,8 @@ def benchmark_model_sequential(model_name, questions, open_router_key, openai_ap
                 st.write(f"**Novelty Score:** {result['novelty_score']}")
                 results.extend(result["results"])
                 novelty_score += result["novelty_score"] # Add to novelty score
-                st.write(
-                    f"<span style='color:yellow'>Total novelty score across processed questions: {novelty_score}</span>",
-                    unsafe_allow_html=True)
+                st.warning(
+                    f"Total novelty score across processed questions: {novelty_score}")
 
             elif result["type"] == "summary":
                 st.write(f"<span style='color:blue'>Total novelty score for question '{result['question']}': {result['total_novelty']}</span>",
@@ -227,7 +223,6 @@ def benchmark_model_sequential(model_name, questions, open_router_key, openai_ap
                 st.write(f"<span style='color:red'>Error in thread: {result['message']}</span>",
                          unsafe_allow_html=True)
 
-    st.write(f"<span style='color:yellow'>Final total novelty score across all questions: {novelty_score}</span>",
-             unsafe_allow_html=True)
+    st.warning(f"Final total novelty score across all questions: {novelty_score}")
 
     return results