Update app.py

app.py

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