Update modules/ui.py

modules/ui.py

@@ -413,17 +413,16 @@ def display_chatbot_interface(lang_code):
     }
     t = translations[lang_code]
     st.title(t['title'])
+
     if 'chatbot' not in st.session_state:
         st.session_state.chatbot = initialize_chatbot()
     if 'messages' not in st.session_state:
         st.session_state.messages = [{"role": "assistant", "content": t['initial_message']}]
     if 'message_id' not in st.session_state:
         st.session_state.message_id = 0
-    if 'message_counter' not in st.session_state:
-        st.session_state.message_counter = 0
-    
+
     chat_container = st.container()
-    
+
     # Mostrar mensajes existentes
     with chat_container:
         for message in st.session_state.messages:
@@ -435,40 +434,37 @@ def display_chatbot_interface(lang_code):
 
     # Área de entrada del usuario
     user_input = st.text_input(t['input_placeholder'], key="user_input")
-    
+
     # Procesamiento del input del usuario
     if user_input:
         st.session_state.messages.append({"role": "user", "content": user_input})
-        st.session_state.message_id += 1
-        with chat_container:
-            st.text_area("You:", value=user_input, height=50, key=f"user_msg_{st.session_state.message_id}", disabled=True)
         
         # Generar respuesta del chatbot
-        try: 
+        try:
             response = get_chatbot_response(st.session_state.chatbot, user_input, lang_code)
         except Exception as e:
             st.error(f"Error al generar la respuesta del chatbot: {str(e)}")
             response = "Lo siento, hubo un error al procesar tu solicitud."
-            
-        st.session_state.messages.append({"role": "assistant", "content": response})
-        st.session_state.message_id += 1
         
-        # Mostrar la respuesta del chatbot con efecto de escritura
-        with chat_container:
-            assistant_message = st.empty()
-            full_response = ""
-            for char in response:
-                full_response += char
-                assistant_message.text_area("Assistant:", value=full_response + "▌", height=50, key=f"assistant_msg_{st.session_state.message_id}_typing_{st.session_state.message_counter}", disabled=True)
-                st.session_state.message_counter += 1
-                time.sleep(0.01)
-            assistant_message.text_area("Assistant:", value=full_response, height=50, key=f"assistant_msg_{st.session_state.message_id}", disabled=True)
+        st.session_state.messages.append({"role": "assistant", "content": response})
         
         # Guardar la conversación en la base de datos
         store_chat_history(st.session_state.username, st.session_state.messages)
         
         # Limpiar el input del usuario
         st.session_state.user_input = ""
-    
-    # Forzar una actualización de la interfaz
-    st.experimental_rerun()
+        
+        # Forzar una actualización de la interfaz
+        st.experimental_rerun()
+
+    # Mostrar el último mensaje del asistente con efecto de escritura
+    if st.session_state.messages and st.session_state.messages[-1]["role"] == "assistant":
+        last_message = st.session_state.messages[-1]["content"]
+        with chat_container:
+            assistant_message = st.empty()
+            full_response = ""
+            for char in last_message:
+                full_response += char
+                assistant_message.text_area("Assistant:", value=full_response + "▌", height=50, key=f"assistant_msg_typing", disabled=True)
+                time.sleep(0.01)
+            assistant_message.text_area("Assistant:", value=full_response, height=50, key=f"assistant_msg_final", disabled=True)