from pathlib import Path

import pandas as pd
import streamlit as st

from draw_utils import draw_plots


st.set_page_config(
    layout='wide',
    initial_sidebar_state='auto',
    page_title='Face Emotion Recognition',
    page_icon='👻',
    )


st.write("#### Отображение результатов детекции видео")

csv_path = 'video_annotations.csv'

@st.cache_data
def load_df():
    df = pd.read_csv(csv_path)
    return df

if Path('video_annotations.csv').exists():
    df = load_df()
    st.download_button(
        label='Скачать csv аннотации', 
        data=df.to_csv().encode('utf-8'), 
        file_name=csv_path,
        mime='text/csv',
        )
    draw_plots(df)
else:
    st.write(
        '#### Для отображения результатов детекции выберите видео' \
        'на вкладке Video Detection и нажмите "Детекция видео"'
    )