import streamlit as st
import time

st.set_page_config(page_title="Meu Chatbot", page_icon="💬", layout="centered")
st.title("💬 Meu Chatbot")

if "messages" not in st.session_state:
    st.session_state.messages = [
        {"role": "assistant", "content": "Olá! Sou o teu chatbot. Em que posso ajudar?"}
    ]

for message in st.session_state.messages:
    with st.chat_message(message["role"]):
        st.markdown(message["content"])

def gerar_resposta(prompt):
    resposta = f"Tu disseste: {prompt}. Isto é uma resposta de teste mais bonita."
    for palavra in resposta.split():
        yield palavra + " "
        time.sleep(0.03)

if prompt := st.chat_input("Escreve a tua mensagem..."):
    st.session_state.messages.append({"role": "user", "content": prompt})

    with st.chat_message("user"):
        st.markdown(prompt)

    with st.chat_message("assistant"):
        resposta_final = st.write_stream(gerar_resposta(prompt))

    st.session_state.messages.append({"role": "assistant", "content": resposta_final})