app.py

import streamlit as st
import getpass
import os
from modelo import get_chain
st.set_page_config(layout="wide")

# Initialization
if 'historial' not in st.session_state:
    st.session_state['historial'] = ['Hola soy tu asistente del dia de hoy, en que te puedo ayudar']

def get_data():
    return st.session_state["historial"]

def add_data(value: str):
    st.session_state["historial"] = value


os.environ["OPENAI_API_KEY"] = st.secrets['OPENAI_API_KEY']  # agregada en la config de hugginface
st.markdown("<h1 style='text-align: center; color: yellow;'>Chatbot SII</h1>", unsafe_allow_html=True)
st.header("Un ChatBot 🤖🦾 entrenado con preguntas frecuentes del sitio del servicios de impuestos interno de Chile.")

pregunta = st.text_area('Ingresa tu pregunta:', value="Que es un APA?")
tmp_button = st.button("CLICK")

chain = get_chain(st.secrets['OPENAI_API_KEY'])
if tmp_button: #Esperar al boton
    out = chain.invoke(pregunta)
    st.write(f"<p style='text-align: right; background-color: yellow;color: black;'>{out['query']}</p>", unsafe_allow_html=True)
    #st.rerun() #Restart app
else:
    st.stop()