Update modules/ui.py

modules/ui.py

@@ -399,16 +399,19 @@ def display_chatbot_interface(lang_code):
             'title': "AIdeaText - Chatbot Multilingüe",
             'input_placeholder': "Escribe tu mensaje aquí...",
             'send_button': "Enviar",
+            'initial_message': "¡Hola! ¿En qué puedo ayudarte hoy?"
         },
         'en': {
             'title': "AIdeaText - Multilingual Chatbot",
             'input_placeholder': "Type your message here...",
             'send_button': "Send",
+            'initial_message': "Hello! How can I assist you today?"
         },
         'fr': {
             'title': "AIdeaText - Chatbot Multilingue",
             'input_placeholder': "Écrivez votre message ici...",
             'send_button': "Envoyer",
+            'initial_message': "Bonjour! Comment puis-je vous aider aujourd'hui?"
         }
     }
 
@@ -420,21 +423,27 @@ def display_chatbot_interface(lang_code):
         st.session_state.chatbot = initialize_chatbot()
 
     if 'messages' not in st.session_state:
-        st.session_state.messages = []
+        st.session_state.messages = [{"role": "assistant", "content": t['initial_message']}]
 
+    # Mostrar mensajes previos
     for message in st.session_state.messages:
         with st.chat_message(message["role"]):
             st.markdown(message["content"])
 
+    # Input del usuario
     if prompt := st.chat_input(t['input_placeholder']):
         st.session_state.messages.append({"role": "user", "content": prompt})
         with st.chat_message("user"):
             st.markdown(prompt)
 
         with st.chat_message("assistant"):
-            response = get_chatbot_response(st.session_state.chatbot, prompt, lang_code)
-            st.markdown(response)
-        st.session_state.messages.append({"role": "assistant", "content": response})
+            message_placeholder = st.empty()
+            full_response = ""
+            for response in get_chatbot_response(st.session_state.chatbot, prompt, lang_code):
+                full_response += response + " "
+                message_placeholder.markdown(full_response + "▌")
+            message_placeholder.markdown(full_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)