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