Upload 6 files

.gitattributes

@@ -33,3 +33,5 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+билли.png filter=lfs diff=lfs merge=lfs -text
+top_logo1.png filter=lfs diff=lfs merge=lfs -text

app.py

@@ -0,0 +1,85 @@
+import streamlit as st
+import base64
+import fasttext
+from annotated_text import annotated_text
+
+st.set_page_config(
+    page_title="detoxi.ai",
+    page_icon="./mini_logo1.png",
+    layout="centered"
+)
+
+# Кодируем логотип в base64 (для локальных файлов)
+@st.cache_data
+def get_image_base64(path):
+    with open(path, "rb") as img_file:
+        return base64.b64encode(img_file.read()).decode()
+
+@st.cache_resource  # Кэширование модели для ускорения работы
+def load_model():
+    model = fasttext.load_model('./model_fasttext.pkl')
+    return model
+
+model = load_model()
+
+bin_str = get_image_base64("./билли.png")
+page_bg_img = '''
+<style>
+.stApp{
+background-image: linear-gradient(rgba(255, 255, 255, 0.7), 
+                        rgba(255, 255, 255, 0.7)),
+                        url("data:image/png;base64,%s");
+background-size: cover;
+background-position: center;
+background-repeat: no-repeat;
+background-attachment: fixed;
+}
+</style>
+''' % bin_str
+st.markdown(page_bg_img, unsafe_allow_html=True)
+
+logo_base64 = get_image_base64("./top_logo1.png")
+
+# Используем HTML для вставки логотипа в заголовок
+st.markdown(
+    f"""
+    <div style="display: flex; justify-content: center;">
+        <img src="data:image/png;base64,{logo_base64}" width="400">
+    </div>
+    """,
+    unsafe_allow_html=True
+)
+
+# Описание
+st.write("""<p style='text-align: center; font-size: 24px;'>Это приложение сделает твою речь менее токсичной. 
+И даже не придётся платить 300 bucks.</p>""", unsafe_allow_html=True)
+
+def highlight_obscene_words(text):
+    label,probs=model.predict(text)
+    if label[0]=='__label__positive':
+        return (text + " ", "приемлемо", "#47916B")
+    else:
+        return (text + " ", "токсично", "#ffcccc")
+
+# Боковая панель
+with st.sidebar:
+    st.header("""О приложении""")
+    st.write("""
+    Это приложение, созданно для сдачи задания по ML.
+    Оно показывает, чему мы научились за эту домашку:
+    - Благославлять создателей hugging face
+    - Писать прототипы приложений с помощью библиотеки Streamlit
+    - Дружно работать в команде
+    """, unsafe_allow_html=True)
+
+st.write("""<p style='text-align: center;'>Введите текст ниже, и приложение определит токсичность твоего предложения.</p>""", unsafe_allow_html=True)
+
+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)
+    else:
+        st.warning("Пожалуйста, введите текст для проверки")

model_fasttext.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:1134dd11e56c7f3110dad960c0cbdeb2d463846602077fcb390e2738a4b5809c
+size 49285271

requirements.txt

@@ -0,0 +1,4 @@
+pybase64
+st_annotated_text
+fasttext
+numpy==1.26.0