Update streamlit_app.py

streamlit_app.py

@@ -1,17 +1,31 @@
 import streamlit as st
-import streamlit.components.v1 as components
+import time
 
-st.set_page_config(layout="wide")
-st.title("Site com Streamlit + Gradio")
+st.set_page_config(page_title="Meu Chatbot", page_icon="💬", layout="centered")
+st.title("💬 Meu Chatbot")
 
-col1, col2 = st.columns(2)
+if "messages" not in st.session_state:
+    st.session_state.messages = [
+        {"role": "assistant", "content": "Olá! Sou o teu chatbot. Em que posso ajudar?"}
+    ]
 
-with col1:
-    st.subheader("Parte Streamlit")
-    st.write("Aqui podes meter gráficos, tabelas, filtros, métricas, etc.")
-    st.metric("Utilizadores", 120)
-    st.metric("Projetos", 8)
+for message in st.session_state.messages:
+    with st.chat_message(message["role"]):
+        st.markdown(message["content"])
 
-with col2:
-    st.subheader("Parte Gradio")
-    components.iframe("http://127.0.0.1:7860", height=500, scrolling=True)
+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})