Introduce MarkupSafe

app.py

@@ -3,6 +3,7 @@ import torch
 from transformers import AutoTokenizer, AutoModelForSequenceClassification
 import nltk
 from nltk.tokenize import sent_tokenize
+from markupsafe import escape
 
 # Set page config at the very beginning
 st.set_page_config(page_title="LLM Detector", layout="centered")
@@ -79,9 +80,8 @@ if st.button("Analyze and Highlight"):
     if text_input:
         with st.spinner("Analyzing text..."):
             overall_probability = classify_text(text_input)
-            st.markdown(
+            st.html(
                 f"<h3>Overall probability of being AI-generated: <span style='color: {'red' if overall_probability > 0.5 else 'green'};'>{overall_probability:.2%}</span></h3>",
-                unsafe_allow_html=True,
             )
 
             st.markdown("### Sentence-level analysis:")
@@ -89,9 +89,8 @@ if st.button("Analyze and Highlight"):
 
             for sentence, score in zip(sentences, scores):
                 color = get_color(score)
-                st.markdown(
-                    f"<div style='background-color: {color}; padding: 10px; margin: 5px 0; border-radius: 5px;'><strong>{score:.2%}</strong> - {sentence}</div>",
-                    unsafe_allow_html=True,
+                st.html(
+                    f"<div style='background-color: {color}; padding: 10px; margin: 5px 0; border-radius: 5px;'><strong>{score:.2%}</strong> - {escape(sentence)}</div>",
                 )
     else:
         st.warning("Please enter some text to analyze.")