import streamlit as st
import pandas as pd

from get_reco import create_recommendation

st.set_page_config(page_title="🐱Midnight Recommender ", page_icon=":books", layout="wide", initial_sidebar_state="expanded")

st.header("Midnight - Recomendação de Livros", divider="violet")

st.caption("Um chatbot simplificado recomendando livros do conjunto de dados da plataforma Skoob no Kaggle[LINK](https://www.kaggle.com/datasets/victorstein/livros-skoob")

st.markdown("Este é um chatbot que recomenda livros de vários tipos do conjunto de dados do Skoob.")
st.markdown("O chatbot usa o [OpenAI API](https://beta.openai.com/) e biblioteca langchain para responder perguntas e recomendar livros.")

df = pd.read_csv("livros-skoob.csv")
st.dataframe(df.head(4), hide_index=True, use_container_width=True,column_order=["titulo","autor","rating","genero"])
with st.sidebar:
    openai_api_key = st.text_input("OpenAI API Key", key="chatbot_api_key", type="password")
    "[Get an OpenAI API key](https://platform.openai.com/account/api-keys)"

with st.form("my_form"):
    input_text = st.text_area("Faça sua pergunta:", "Sugira um livro sobre gatos?")
    submitted = st.form_submit_button("Submeter")
    if submitted:
        if not openai_api_key:
            st.warning("Insira sua chave de API OpenAI na barra lateral")
        else:
            create_recommendation(openai_api_key,input_text)