Update src/app.py

src/app.py

@@ -142,18 +142,18 @@ if len(chat_model.conversation.messages) != len(st.session_state.messages):
 
 chat_container = st.container()
 
-with chat_container:
-    for message in st.session_state.messages:
-        role = message["role"]
-        content = message["content"]
+#with chat_container:
+#    for message in st.session_state.messages:
+#        role = message["role"]
+#        content = message["content"]
+#        
+#        if role == "system":
+#            continue
         
-        if role == "system":
-            continue
-        
-        with st.chat_message(role):
-            st.markdown(content)
+#        with st.chat_message(role):
+#            st.markdown(content)
 
-    if user_input := st.chat_input("Digite sua mensagem..."):
+if user_input := st.chat_input("Digite sua mensagem..."):
         # limpa o historico
         st.session_state.messages = []
         st.session_state.chat_model.clear_history()
@@ -165,9 +165,9 @@ with chat_container:
         st.session_state.messages.append({"role": "user", "content": user_input})
     
         with st.chat_message("user"):
-            st.markdown(user_input)
+            chat_container.st.markdown(user_input)
     
-        with st.chat_message("assistant"):
+        with chat_container.st.chat_message("assistant"):
             response_placeholder = st.empty()
             full_response = ""