Spaces:
Running
Running
| # modules/chatbot/sidebar_chat.py | |
| import streamlit as st | |
| from .chat_process import ChatProcessor | |
| from ..database.chat_mongo_db import store_chat_history, get_chat_history | |
| import logging | |
| logger = logging.getLogger(__name__) | |
| def display_sidebar_chat(lang_code: str, chatbot_t: dict): | |
| """ | |
| Muestra el chatbot en el sidebar con soporte para contexto semántico | |
| """ | |
| with st.sidebar: | |
| # Configurar logging | |
| logging.basicConfig( | |
| level=logging.INFO, | |
| format='%(asctime)s - %(name)s - %(levelname)s - %(message)s' | |
| ) | |
| # Estilos CSS para el chat | |
| st.markdown(""" | |
| <style> | |
| .chat-container { | |
| max-height: 60vh; | |
| overflow-y: auto; | |
| padding-right: 10px; | |
| } | |
| .chat-input { | |
| position: sticky; | |
| bottom: 0; | |
| background: white; | |
| padding-top: 10px; | |
| z-index: 100; | |
| } | |
| </style> | |
| """, unsafe_allow_html=True) | |
| try: | |
| # Inicializar el procesador de chat si no existe | |
| if 'chat_processor' not in st.session_state: | |
| st.session_state.chat_processor = ChatProcessor() | |
| logger.info("ChatProcessor initialized") | |
| # Configurar contexto semántico si está activo | |
| if st.session_state.get('semantic_agent_active', False): | |
| semantic_data = st.session_state.get('semantic_agent_data') | |
| if semantic_data: | |
| st.session_state.chat_processor.set_semantic_context( | |
| text=semantic_data['text'], | |
| metrics=semantic_data['metrics'], | |
| graph_data=semantic_data['graph_data'], | |
| lang_code=lang_code # Pasar el idioma actual | |
| ) | |
| logger.info(f"Semantic context set for language: {lang_code}") | |
| with st.expander("💬 Asistente Virtual", expanded=True): | |
| # Inicializar mensajes del chat | |
| if 'sidebar_messages' not in st.session_state: | |
| initial_messages = { | |
| 'en': "Hello! How can I help you today?", | |
| 'es': "¡Hola! ¿Cómo puedo ayudarte hoy?", | |
| 'pt': "Olá! Como posso ajudar você hoje?", | |
| 'fr': "Bonjour ! Comment puis-je vous aider aujourd'hui ?" | |
| } | |
| initial_message = initial_messages.get(lang_code, initial_messages['en']) | |
| st.session_state.sidebar_messages = [ | |
| {"role": "assistant", "content": initial_message} | |
| ] | |
| logger.info("Chat messages initialized") | |
| # Contenedor del chat | |
| chat_container = st.container() | |
| with chat_container: | |
| for message in st.session_state.sidebar_messages: | |
| with st.chat_message(message["role"]): | |
| st.markdown(message["content"]) | |
| # Input del usuario | |
| user_input = st.chat_input( | |
| { | |
| 'en': "Type your message...", | |
| 'es': "Escribe tu mensaje...", | |
| 'pt': "Digite sua mensagem...", | |
| 'fr': "Tapez votre message..." | |
| }.get(lang_code, "Type your message...") | |
| ) | |
| if user_input: | |
| try: | |
| # Agregar mensaje del usuario | |
| st.session_state.sidebar_messages.append( | |
| {"role": "user", "content": user_input} | |
| ) | |
| logger.info(f"User message received: {user_input[:50]}...") | |
| # Mostrar mensaje del usuario | |
| with chat_container: | |
| with st.chat_message("user"): | |
| st.markdown(user_input) | |
| # Mostrar respuesta del asistente | |
| with st.chat_message("assistant"): | |
| response_placeholder = st.empty() | |
| full_response = "" | |
| # Obtener respuesta del procesador | |
| for chunk in st.session_state.chat_processor.process_chat_input( | |
| user_input, | |
| lang_code | |
| ): | |
| full_response += chunk | |
| response_placeholder.markdown(full_response + "▌") | |
| response_placeholder.markdown(full_response) | |
| logger.info(f"Assistant response generated: {full_response[:50]}...") | |
| # Guardar respuesta | |
| st.session_state.sidebar_messages.append( | |
| {"role": "assistant", "content": full_response} | |
| ) | |
| # Guardar en base de datos | |
| store_chat_history( | |
| username=st.session_state.username, | |
| messages=st.session_state.sidebar_messages, | |
| chat_type='semantic' if st.session_state.get('semantic_agent_active') else 'general' | |
| ) | |
| logger.info("Conversation saved to database") | |
| except Exception as e: | |
| logger.error(f"Error processing user input: {str(e)}", exc_info=True) | |
| st.error({ | |
| 'en': "Error processing message. Please try again.", | |
| 'es': "Error al procesar el mensaje. Por favor, inténtalo de nuevo.", | |
| 'pt': "Erro ao processar a mensagem. Por favor, tente novamente.", | |
| 'fr': "Erreur lors du traitement du message. Veuillez réessayer." | |
| }.get(lang_code, "Error processing message.")) | |
| # Botón para limpiar el chat | |
| if st.button({ | |
| 'en': "🔄 Clear conversation", | |
| 'es': "🔄 Limpiar conversación", | |
| 'pt': "🔄 Limpar conversa", | |
| 'fr': "🔄 Effacer la conversation" | |
| }.get(lang_code, "🔄 Clear")): | |
| st.session_state.sidebar_messages = [{ | |
| "role": "assistant", | |
| "content": { | |
| 'en': "Hello! How can I help you today?", | |
| 'es': "¡Hola! ¿Cómo puedo ayudarte hoy?", | |
| 'pt': "Olá! Como posso ajudar você hoje?", | |
| 'fr': "Bonjour ! Comment puis-je vous aider aujourd'hui ?" | |
| }.get(lang_code, "Hello! How can I help you?") | |
| }] | |
| st.rerun() | |
| logger.info("Conversation cleared") | |
| except Exception as e: | |
| logger.error(f"Error in sidebar chat: {str(e)}", exc_info=True) | |
| st.error({ | |
| 'en': "An error occurred in the chat. Please try again.", | |
| 'es': "Ocurrió un error en el chat. Por favor, inténtalo de nuevo.", | |
| 'pt': "Ocorreu um erro no chat. Por favor, tente novamente.", | |
| 'fr': "Une erreur s'est produite dans le chat. Veuillez réessayer." | |
| }.get(lang_code, "Chat error occurred.")) |