import streamlit as st
from utils import load_datasets, save_session_cache
from streamlit_extras.grid import grid
import pandas as pd
from streamlit_extras.dataframe_explorer import dataframe_explorer

st.title("💁 데이터 직접 선택하기")

# Section 1: Data Selection
st.header("1️⃣ 데이터 목록 필터링 하기")
dataframe = load_datasets()

dataframe['keywords'] = dataframe['keywords'].apply(lambda x: ', '.join(x) if isinstance(x, list) else x)
list_columns = [col for col in dataframe.columns if dataframe[col].apply(lambda x: isinstance(x, list)).all()]
dict_columns = [col for col in dataframe.columns if dataframe[col].apply(lambda x: isinstance(x, dict)).all()]
    # list 타입의 컬럼을 드랍
dataframe = dataframe.drop(columns=list_columns)
dataframe = dataframe.drop(columns=dict_columns)
filtered_df = dataframe_explorer(dataframe, case=False)
st.dataframe(filtered_df, use_container_width=True)
st.divider()

st.header("2️⃣ 분석 데이터 선택하기")

selectable_dataset_list = [None] + filtered_df['title'].values.tolist()

# Select data for analysis
selected_dataset = st.selectbox(
    "🔍 분석할 데이터를 선택하세요", 
    selectable_dataset_list,
    index=selectable_dataset_list.index(st.session_state.selected_dataset) if st.session_state.selected_dataset in selectable_dataset_list else 0)
st.session_state['selected_dataset'] = selected_dataset
st.divider()

# session_id=save_session_cache(st.session_state.to_dict())
def switch_to_summarization():
    st.switch_page(f"_pages/summarization.py")
st.header("3️⃣ 데이터 요약으로 넘어가기")
next_page = st.button(label=f"️✅ {st.session_state.selected_dataset}으로 요약하기" if st.session_state.selected_dataset else "⛔ 데이터를 선택해야 요약을 볼 수 있습니다.",
          use_container_width=True,
          type="secondary",
          disabled=st.session_state.selected_dataset is None,
          )
if next_page:
    switch_to_summarization()

st.divider()