Update modules/ui.py

modules/ui.py

@@ -420,38 +420,40 @@ def display_chatbot_interface(lang_code):
     if 'messages' not in st.session_state:
         st.session_state.messages = [{"role": "assistant", "content": t['initial_message']}]
 
-    # Contenedor para el chat
+    # Contenedor principal para el chat
     chat_container = st.container()
 
-    # Mostrar todos los mensajes en el contenedor de chat
+    # Mostrar mensajes existentes
     with chat_container:
         for message in st.session_state.messages:
             with st.chat_message(message["role"]):
-                st.write(message["content"])
+                st.markdown(message["content"])
 
-    # Área de entrada del usuario (en la parte inferior)
+    # Área de entrada del usuario
     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})
-        with st.chat_message("user"):
-            st.write(user_input)
+        
+        # Mostrar mensaje del usuario
+        with chat_container:
+            with st.chat_message("user"):
+                st.markdown(user_input)
 
-        # Generar y mostrar respuesta del chatbot
-        with chat_container: 
+        # Generar respuesta del chatbot
+        with chat_container:
             with st.chat_message("assistant"):
-                message_placeholder = st.empty()  
+                message_placeholder = st.empty()
                 full_response = ""
-
-                for chunk in get_chatbot_response(st.session_state.chatbot, prompt, lang_code):
+                
+                for chunk in get_chatbot_response(st.session_state.chatbot, user_input, lang_code):
                     full_response += chunk
                     message_placeholder.markdown(full_response + "▌")
                 message_placeholder.markdown(full_response)
-                
+
         # Agregar respuesta del asistente a los mensajes
-        st.session_state.messages.append({"role": "assistant", "content": response})
+        st.session_state.messages.append({"role": "assistant", "content": full_response})
         
         # Guardar la conversación en la base de datos
         store_chat_history(st.session_state.username, st.session_state.messages)