Create sidebar_chat.py

modules/chatbot/sidebar_chat.py

@@ -0,0 +1,105 @@
+# modules/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
+import logging
+
+logger = logging.getLogger(__name__)
+
+def display_sidebar_chat(lang_code):
+    """
+    Muestra el chatbot en el sidebar
+    """
+    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()
+                
+            # 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']}
+                    ]
+
+            # 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'
+            )
+
+            # Procesar input
+            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 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
+                )
+
+            # Botón para limpiar chat
+            if st.button(t['clear_chat']):
+                st.session_state.sidebar_messages = [
+                    {"role": "assistant", "content": t['initial_message']}
+                ]
+                st.rerun()