import streamlit as st
from PIL import Image

# Заголовок страницы
st.set_page_config(page_title="Golden RetrAIever", page_icon="🤖", layout="wide")

# Создание колонок для размещения текста и изображения
col1, col2 = st.columns([2, 1])

# Контент первой колонки (основной текст)
with col1:
    st.title("Golden RetrAIever - AI Knowledge Assistant 🤖")
    st.markdown("""
    #### Сервис для подготовки к интервью и проф роста в DS/ML/DL  
                
    ## Как использовать продукт, чтобы подготовиться к интервью?
    - Задать прямые вопросы про ML / NN / Math
    - Делайте самопроверку
    - Переслушивайте теорию за 15 минут, а не 2 часа!
    - Кайфуйте. Если заскучали, то найдите очередной ролик на Ютуб про ML - и найдите пункт "заскучал" на этом сайте. Мы предложим решение :)
    
    Этот проект состоит из двух мощных инструментов:
    - **Первая страница:** Система Retrieval-Augmented Generation (RAG), которая отвечает на вопросы, используя базу знаний, собранную из YouTube-видео - топ 5 каналов в мире! 
    - **Вторая страница:** Инструмент для создания текстовых саммари по видео. Просто введите ссылку на YouTube и выберите тип саммари. Основано на современной педагогике. Сделано с Anthropic API с любовью к искусству промпт инжиниринга. 
    
    ### Начните с первой страницы и получите ответы на вопросы по ML, Neural Networks, математике и всему, что нужно!    """)

    # Кнопка для перехода к первому инструменту
    #st.button("Перейти к системе RAG")

# Контент второй колонки (изображение)
with col2:
    image = Image.open('images/dog.png')  # Загрузите изображение для более привлекательного вида
    st.image(image, caption="", use_column_width=True)

# Линия для разделения контента
st.markdown("---")

# Добавим небольшой футер с информацией о проекте
st.markdown("""
<div style='text-align: center; color: gray;'>
    <small>Сервис для подготовки к интервью и проф роста в DS/ML/DL  Мы на связи: Настя @anastasiap20, Саша @seppokku, Наташа @npatashkevich</small>
</div>
""", unsafe_allow_html=True)