Update modules/ui.py

modules/ui.py

@@ -393,6 +393,7 @@ def display_discourse_analysis_interface(nlp_models, lang_code):
                 st.pyplot(graph2)
 
 ##################################################################################################
+
 def display_chatbot_interface(lang_code):
     translations = {
         'es': {
@@ -425,12 +426,11 @@ def display_chatbot_interface(lang_code):
 
     # Mostrar mensajes existentes
     with chat_container:
-        for message in st.session_state.messages:
+        for idx, message in enumerate(st.session_state.messages):
             if message["role"] == "user":
-                st.text_area("You:", value=message["content"], height=50, key=f"user_msg_{st.session_state.message_id}", disabled=True)
+                st.text_area("You:", value=message["content"], height=50, key=f"user_msg_{idx}", disabled=True)
             else:
-                st.text_area("Assistant:", value=message["content"], height=50, key=f"assistant_msg_{st.session_state.message_id}", disabled=True)
-            st.session_state.message_id += 1
+                st.text_area("Assistant:", value=message["content"], height=50, key=f"assistant_msg_{idx}", disabled=True)
 
     # Área de entrada del usuario
     user_input = st.text_input(t['input_placeholder'], key="user_input")
@@ -463,8 +463,8 @@ def display_chatbot_interface(lang_code):
         with chat_container:
             assistant_message = st.empty()
             full_response = ""
-            for char in last_message:
+            for i, char in enumerate(last_message):
                 full_response += char
-                assistant_message.text_area("Assistant:", value=full_response + "▌", height=50, key=f"assistant_msg_typing", disabled=True)
+                assistant_message.text_area("Assistant:", value=full_response + "▌", height=50, key=f"assistant_msg_typing_{i}", disabled=True)
                 time.sleep(0.01)
-            assistant_message.text_area("Assistant:", value=full_response, height=50, key=f"assistant_msg_final", disabled=True)
+            assistant_message.text_area("Assistant:", value=full_response, height=50, key=f"assistant_msg_final_{len(st.session_state.messages)-1}", disabled=True)