Update modules/chatbot/sidebar_chat.py

modules/chatbot/sidebar_chat.py

@@ -1,105 +1,93 @@
 # modules/chatbot/sidebar_chat.py
+# modules/chatbot/chatbot/sidebar_chat.py
 import streamlit as st
-from .chatbot import ClaudeAPIChat, initialize_chatbot, get_chatbot_response
-from ..database.chat_mongo_db import store_chat_history, get_chat_history
+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):
+def display_sidebar_chat(lang_code: str, chatbot_t: dict):
     """
     Muestra el chatbot en el sidebar
+    Args:
+        lang_code: Código del idioma
+        chatbot_t: Traducciones del chatbot del archivo central
     """
-    translations = {
-        'es': {
-            'chat_title': "Asistente AIdeaText",
-            'input_placeholder': "¿Tienes alguna pregunta?",
-            'initial_message': "¡Hola! Soy tu asistente. ¿En qué puedo ayudarte?",
-            'expand_chat': "Abrir asistente",
-            'clear_chat': "Limpiar chat"
-        },
-        'en': {
-            'chat_title': "AIdeaText Assistant",
-            'input_placeholder': "Any questions?",
-            'initial_message': "Hi! I'm your assistant. How can I help?",
-            'expand_chat': "Open assistant",
-            'clear_chat': "Clear chat"
-        }
-    }
-    
-    t = translations.get(lang_code, translations['en'])
-
     with st.sidebar:
         # Chatbot expandible
-        with st.expander(t['expand_chat'], expanded=False):
-            # Inicializar chatbot si no existe
-            if 'sidebar_chatbot' not in st.session_state:
-                st.session_state.sidebar_chatbot = initialize_chatbot()
+        with st.expander(chatbot_t.get('expand_chat', 'Open assistant'), expanded=False):
+            try:
+                # Inicializar procesador si no existe
+                if 'chat_processor' not in st.session_state:
+                    st.session_state.chat_processor = ChatProcessor()
                 
-            # Inicializar mensajes si no existen
-            if 'sidebar_messages' not in st.session_state:
-                # Intentar recuperar historial previo
-                history = get_chat_history(st.session_state.username, 'sidebar', 10)
-                if history:
-                    st.session_state.sidebar_messages = history[0]['messages']
-                else:
-                    st.session_state.sidebar_messages = [
-                        {"role": "assistant", "content": t['initial_message']}
-                    ]
+                # Inicializar mensajes si no existen
+                if 'sidebar_messages' not in st.session_state:
+                    # Intentar recuperar historial previo
+                    history = get_chat_history(st.session_state.username, 'sidebar', 10)
+                    if history:
+                        st.session_state.sidebar_messages = history[0]['messages']
+                    else:
+                        st.session_state.sidebar_messages = [
+                            {"role": "assistant", "content": chatbot_t['initial_message']}
+                        ]
 
-            # Contenedor del chat
-            chat_container = st.container()
+                # Contenedor del chat
+                chat_container = st.container()
 
-            # Mostrar mensajes
-            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.text_input(
-                t['input_placeholder'],
-                key='sidebar_chat_input'
-            )
+                # Mostrar mensajes existentes
+                with chat_container:
+                    for message in st.session_state.sidebar_messages:
+                        with st.chat_message(message["role"]):
+                            st.markdown(message["content"])
 
-            # Procesar input
-            if user_input:
-                # Agregar mensaje del usuario
-                st.session_state.sidebar_messages.append(
-                    {"role": "user", "content": user_input}
+                # Input del usuario
+                user_input = st.text_input(
+                    chatbot_t['input_placeholder'],
+                    key='sidebar_chat_input'
                 )
 
-                # Generar y mostrar respuesta
-                with chat_container:
-                    with st.chat_message("assistant"):
-                        message_placeholder = st.empty()
-                        full_response = ""
-                        
-                        for chunk in get_chatbot_response(
-                            st.session_state.sidebar_chatbot,
-                            user_input,
-                            lang_code
-                        ):
-                            full_response += chunk
-                            message_placeholder.markdown(full_response + "▌")
-                        
-                        message_placeholder.markdown(full_response)
-                        
-                        # Guardar respuesta
-                        st.session_state.sidebar_messages.append(
-                            {"role": "assistant", "content": full_response}
-                        )
-                
-                # Guardar conversación
-                store_chat_history(
-                    st.session_state.username,
-                    'sidebar',
-                    st.session_state.sidebar_messages
-                )
+                if user_input:
+                    # Agregar mensaje del usuario
+                    st.session_state.sidebar_messages.append(
+                        {"role": "user", "content": user_input}
+                    )
+
+                    # Generar y mostrar respuesta
+                    with chat_container:
+                        with st.chat_message("assistant"):
+                            message_placeholder = st.empty()
+                            full_response = ""
+                            
+                            for chunk in st.session_state.chat_processor.process_chat_input(
+                                user_input,
+                                lang_code
+                            ):
+                                full_response += chunk
+                                message_placeholder.markdown(full_response + "▌")
+                            
+                            message_placeholder.markdown(full_response)
+                            
+                            # Guardar respuesta
+                            st.session_state.sidebar_messages.append(
+                                {"role": "assistant", "content": full_response}
+                            )
+                    
+                    # Guardar conversación
+                    store_chat_history(
+                        st.session_state.username,
+                        'sidebar',
+                        st.session_state.sidebar_messages
+                    )
+
+                # Botón para limpiar chat
+                if st.button(chatbot_t['clear_chat']):
+                    st.session_state.sidebar_messages = [
+                        {"role": "assistant", "content": chatbot_t['initial_message']}
+                    ]
+                    st.rerun()
 
-            # Botón para limpiar chat
-            if st.button(t['clear_chat']):
-                st.session_state.sidebar_messages = [
-                    {"role": "assistant", "content": t['initial_message']}
-                ]
-                st.rerun()
+            except Exception as e:
+                logger.error(f"Error en sidebar chat: {str(e)}")
+                st.error(chatbot_t['error_message'])