Update modules/ui.py

modules/ui.py

@@ -172,27 +172,6 @@ def register_form():
     ##
     ##
     pass
-##################################################################################################
-def display_chat_interface():
-    st.markdown("### Chat con AIdeaText")
-
-    if 'chat_history' not in st.session_state:
-        st.session_state.chat_history = []
-
-    for i, (role, text) in enumerate(st.session_state.chat_history):
-        if role == "user":
-            st.text_area(f"Tú:", value=text, height=50, key=f"user_message_{i}", disabled=True)
-        else:
-            st.text_area(f"AIdeaText:", value=text, height=50, key=f"bot_message_{i}", disabled=True)
-
-    user_input = st.text_input("Escribe tu mensaje aquí:")
-
-    if st.button("Enviar"):
-        if user_input:
-            st.session_state.chat_history.append(("user", user_input))
-            response = get_chatbot_response(user_input)
-            st.session_state.chat_history.append(("bot", response))
-            st.experimental_rerun()
 
 ################################################################################
 # Funciones para Cosmos DB MongoDB API (análisis de texto)
@@ -697,7 +676,7 @@ def display_chatbot_interface(lang_code):
     if user_input:
         # Agregar mensaje del usuario
         st.session_state.messages.append({"role": "user", "content": user_input})
-        
+
         # Mostrar mensaje del usuario
         with chat_container:
             with st.chat_message("user"):
@@ -708,7 +687,6 @@ def display_chatbot_interface(lang_code):
             with st.chat_message("assistant"):
                 message_placeholder = st.empty()
                 full_response = ""
-                
                 for chunk in get_chatbot_response(st.session_state.chatbot, user_input, lang_code):
                     full_response += chunk
                     message_placeholder.markdown(full_response + "▌")
@@ -716,9 +694,14 @@ def display_chatbot_interface(lang_code):
 
         # Agregar respuesta del asistente a los mensajes
         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)
+        try:
+            store_chat_history(st.session_state.username, st.session_state.messages)
+            st.success("Conversación guardada exitosamente")
+        except Exception as e:
+            st.error(f"Error al guardar la conversación: {str(e)}")
+            logger.error(f"Error al guardar el historial de chat para {st.session_state.username}: {str(e)}")
 
     # Scroll al final del chat
     st.markdown('<script>window.scrollTo(0,document.body.scrollHeight);</script>', unsafe_allow_html=True)