import streamlit as st

def main():
    st.image(['https://media0.giphy.com/media/S5iqkyBY9o4KY/giphy.gif?cid=9b38fe914fwujlyn3qfhsqlgkw7ojjb563f9pkbllvy5gx73&ep=v1_gifs_related&rid=giphy.gif&ct=g', 
              ])
    st.title("Определение тональности текста")
    
    st.write("""
        Проект позволяет пользователю ввести текст и получить его классификацию по тональности.
    """)

    st.write("""
        Состав проекта:
        1. **Классификация отзыва на фильм**:
            - Пользователь вводит отзыв на фильм.
            - Предсказывается его тональность тремя моделями.
            - Выводится время выполнения предсказания.

        2. **Оценка токсичности сообщения**:
            - Пользователь вводит текст сообщения.
            - Оценивается степень токсичности текста с помощью модели rubert-tiny-toxicity.
            - Выводится время выполнения предсказания.

        3. **Генерация текста с помощью GPT-модели**:
            - Генерируется текст по пользовательскому сообщению с возможностью настройки параметров.
    """)


if __name__ == "__main__":
    main()

st.write("""
         **Над проектом работали:**
         - **[Андрей Мирошниченко](https://github.com/Andriano2323?query=And)**
         - **[Сайыына Антонова](https://github.com/SainaAntonova)**
         """)