Update modules/ui.py

modules/ui.py

@@ -419,52 +419,41 @@ def display_chatbot_interface(lang_code):
         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
 
+    # Contenedor para el chat
     chat_container = st.container()
 
-    # Mostrar mensajes existentes
+    # Mostrar todos los mensajes en el contenedor de chat
     with chat_container:
-        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_{idx}", disabled=True)
-            else:
-                st.text_area("Assistant:", value=message["content"], height=50, key=f"assistant_msg_{idx}", disabled=True)
+        for message in st.session_state.messages:
+            with st.chat_message(message["role"]):
+                st.write(message["content"])
 
-    # Área de entrada del usuario
-    user_input = st.text_input(t['input_placeholder'], key="user_input")
+    # Área de entrada del usuario (en la parte inferior)
+    user_input = st.chat_input(t['input_placeholder'])
 
     # Procesamiento del input del usuario
     if user_input:
+        # Agregar mensaje del usuario
         st.session_state.messages.append({"role": "user", "content": user_input})
-        
-        # Generar respuesta del chatbot
-        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."
-        
+        with st.chat_message("user"):
+            st.write(user_input)
+
+        # Generar y mostrar respuesta del chatbot
+        with st.chat_message("assistant"):
+            with st.spinner("Pensando..."):
+                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.write(response)
+
+        # Agregar respuesta del asistente a los mensajes
         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()
-
-    # 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 i, char in enumerate(last_message):
-                full_response += char
-                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_{len(st.session_state.messages)-1}", disabled=True)
+
+    # Scroll al final del chat
+    st.markdown('<script>window.scrollTo(0,document.body.scrollHeight);</script>', unsafe_allow_html=True)