import streamlit as st
from PIL import Image

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

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

# Контент первой колонки (основной текст)
with col1:
    st.title("AI Knowledge Assistant 🤖")
    st.markdown("""
    ## Добро пожаловать в вашу персональную AI-систему для вопросов по машинному обучению! 
    
    Этот проект состоит из двух мощных инструментов:
    - **Первая страница:** Система Retrieval-Augmented Generation (RAG), которая отвечает на вопросы, используя базу знаний, собранную из транскриптов YouTube-видео по машинному обучению.
    - **Вторая страница:** Инструмент для создания текстовых саммари по видео. Просто введите ссылку на YouTube и выберите тип саммари, чтобы получить краткое содержание.

    ### Начните с первой страницы и получите ответы на вопросы по машинному обучению, или перейдите на вторую страницу для саммари видео!
    """)

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

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

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

# Добавим небольшой футер с информацией о проекте
st.markdown("""
<div style='text-align: center; color: gray;'>
    <small>Этот проект создан для помощи в изучении машинного обучения, предоставляя ответы на основе видео и облегчая получение информации с YouTube. Разработано с любовью к AI и Data Science.</small>
</div>
""", unsafe_allow_html=True)