| 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) |
| |
| |
| |
|
|
|
|
| |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
