Update modules/ui.py

modules/ui.py

@@ -8,10 +8,11 @@ import pandas as pd
 from streamlit_player import st_player  # Necesitarás instalar esta librería: pip install streamlit-player
 from spacy import displacy
 
+
 #Importaciones locales
 #Importaciones locales de autenticación y base de datos 
 from .auth import authenticate_user, register_user, get_user_role
-from .database import get_student_data, store_morphosyntax_result, store_semantic_result
+from .database import get_student_data, store_morphosyntax_result, store_semantic_result, store_chat_history
 
 #Importaciones locales funciones de análisis
 from .morpho_analysis import generate_arc_diagram, get_repeated_words_colors, highlight_repeated_words, POS_COLORS, POS_TRANSLATIONS
@@ -395,17 +396,17 @@ def display_discourse_analysis_interface(nlp_models, lang_code):
 def display_chatbot_interface(lang_code):
     translations = {
         'es': {
-            'title': "AIdeaText - Chatbot Llama 2",
+            'title': "AIdeaText - Chatbot Multilingüe",
             'input_placeholder': "Escribe tu mensaje aquí...",
             'send_button': "Enviar",
         },
         'en': {
-            'title': "AIdeaText - Llama 2 Chatbot",
+            'title': "AIdeaText - Multilingual Chatbot",
             'input_placeholder': "Type your message here...",
             'send_button': "Send",
         },
         'fr': {
-            'title': "AIdeaText - Chatbot Llama 2",
+            'title': "AIdeaText - Chatbot Multilingue",
             'input_placeholder': "Écrivez votre message ici...",
             'send_button': "Envoyer",
         }
@@ -431,6 +432,9 @@ def display_chatbot_interface(lang_code):
             st.markdown(prompt)
 
         with st.chat_message("assistant"):
-            response = get_chatbot_response(st.session_state.chatbot, prompt)
+            response = get_chatbot_response(st.session_state.chatbot, prompt, lang_code)
             st.markdown(response)
-        st.session_state.messages.append({"role": "assistant", "content": response})
+        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)