Update app.py

app.py

@@ -1,6 +1,7 @@
 import streamlit as st
 import base64
 import fasttext
+from annotated_text import annotated_text
 
 st.set_page_config(
     page_title="detoxi.ai",
@@ -56,15 +57,17 @@ st.write("""<p style='text-align: center; font-size: 24px;'>Это прилож
 def highlight_obscene_words(text):
     label,probs=model.predict(text.lower())
     if label[0]=='__label__positive':
-        st.markdown(
-            "<span style='background:#47916B;'>приемлемо</span>", 
-            unsafe_allow_html=True
-        )
+        #st.markdown(
+        #    "<span style='background:#47916B;'>приемлемо</span>", 
+        #    unsafe_allow_html=True
+        #)
+        return (text+' ', 'приемлемо','#47916B')
     else:
-        st.markdown(
-            "<span style='background:#ffcccc;'>токсично</span>", 
-            unsafe_allow_html=True
-        )
+        #st.markdown(
+        #    "<span style='background:#ffcccc;'>токсично</span>", 
+        #    unsafe_allow_html=True
+        #)
+        return (text+' ', 'токсично','#ffcccc')
 
 # Боковая панель
 with st.sidebar:
@@ -84,6 +87,6 @@ user_input = st.text_area('',height=200)
 if st.button("Проверить текст"):
     if user_input.strip():
         st.subheader("Результат:")
-        highlight_obscene_words(user_input)
+        annotated_text(highlight_obscene_words(user_input))
     else:
         st.warning("Пожалуйста, введите текст для проверки")