iatoolkit 0.11.1__py3-none-any.whl → 0.14.0__py3-none-any.whl

iatoolkit/common/routes.py

@@ -26,9 +26,8 @@ def register_views(injector, app):
     from iatoolkit.views.tasks_view import TaskView
     from iatoolkit.views.tasks_review_view import TaskReviewView
     from iatoolkit.views.home_view import HomeView
-    from iatoolkit.views.chat_view import ChatView
-    from iatoolkit.views.login_view import LoginView
-    from iatoolkit.views.external_chat_login_view import ExternalChatLoginView
+    from iatoolkit.views.login_view import LoginView, InitiateLoginView
+    from iatoolkit.views.login_external_id_view import InitiateExternalChatView, ExternalChatLoginView
     from iatoolkit.views.signup_view import SignupView
     from iatoolkit.views.verify_user_view import VerifyAccountView
     from iatoolkit.views.forgot_password_view import ForgotPasswordView
@@ -37,21 +36,22 @@ def register_views(injector, app):
     from iatoolkit.views.user_feedback_view import UserFeedbackView
     from iatoolkit.views.prompt_view import PromptView
     from iatoolkit.views.chat_token_request_view import ChatTokenRequestView
-    from iatoolkit.views.external_login_view import ExternalLoginView
     from iatoolkit.views.download_file_view import DownloadFileView
 
     app.add_url_rule('/', view_func=HomeView.as_view('home'))
 
-    # main chat for iatoolkit front
-    app.add_url_rule('/<company_short_name>/chat', view_func=ChatView.as_view('chat'))
+    # login for external portals
+    app.add_url_rule('/<company_short_name>/initiate_external_chat',
+                         view_func=InitiateExternalChatView.as_view('initiate_external_chat'))
+    app.add_url_rule('/<company_short_name>/external_login',
+                     view_func=ExternalChatLoginView.as_view('external_login'))
+    app.add_url_rule('/auth/chat_token',
+                     view_func=ChatTokenRequestView.as_view('chat-token'))
 
-    # front if the company internal portal
-    app.add_url_rule('/<company_short_name>/chat_login', view_func=ExternalChatLoginView.as_view('external_chat_login'))
-    app.add_url_rule('/<company_short_name>/external_login/<external_user_id>', view_func=ExternalLoginView.as_view('external_login'))
-    app.add_url_rule('/auth/chat_token', view_func=ChatTokenRequestView.as_view('chat-token'))
-
-    # main pages for the iatoolkit frontend
+    # login for the iatoolkit integrated frontend
     app.add_url_rule('/<company_short_name>/login', view_func=LoginView.as_view('login'))
+    app.add_url_rule('/<company_short_name>/initiate_login', view_func=InitiateLoginView.as_view('initiate_login'))
+
     app.add_url_rule('/<company_short_name>/signup',view_func=SignupView.as_view('signup'))
     app.add_url_rule('/<company_short_name>/logout', 'logout', logout)
     app.add_url_rule('/logout', 'logout', logout)

iatoolkit/iatoolkit.py

@@ -144,8 +144,13 @@ class IAToolkit:
         is_https = self._get_config_value('USE_HTTPS', 'false').lower() == 'true'
         is_dev = self._get_config_value('FLASK_ENV') == 'development'
 
+        # get the iatoolkit domain
+        parsed_url = urlparse(os.getenv('IATOOLKIT_BASE_URL'))
+        domain = parsed_url.netloc
+
         self.app.config.update({
             'VERSION': self.version,
+            'SERVER_NAME': domain,
             'SECRET_KEY': self._get_config_value('FLASK_SECRET_KEY', 'iatoolkit-default-secret'),
             'SESSION_COOKIE_SAMESITE': "None" if is_https else "Lax",
             'SESSION_COOKIE_SECURE': is_https,
@@ -156,6 +161,9 @@ class IAToolkit:
             'JWT_EXPIRATION_SECONDS_CHAT': int(self._get_config_value('JWT_EXPIRATION_SECONDS_CHAT', 3600))
         })
 
+        if parsed_url.scheme == 'https':
+            self.app.config['PREFERRED_URL_SCHEME'] = 'https'
+
         # Configuración para tokenizers en desarrollo
         if is_dev:
             os.environ["TOKENIZERS_PARALLELISM"] = "false"
@@ -315,12 +323,8 @@ class IAToolkit:
         """Vincula las vistas después de que el injector ha sido creado"""
         from iatoolkit.views.llmquery_view import LLMQueryView
         from iatoolkit.views.home_view import HomeView
-        from iatoolkit.views.chat_view import ChatView
-        from iatoolkit.views.change_password_view import ChangePasswordView
 
         binder.bind(HomeView, to=HomeView)
-        binder.bind(ChatView, to=ChatView)
-        binder.bind(ChangePasswordView, to=ChangePasswordView)
         binder.bind(LLMQueryView, to=LLMQueryView)
 
         logging.info("✅ Views configuradas correctamente")

iatoolkit/services/profile_service.py

@@ -17,7 +17,6 @@ import secrets
 import string
 from datetime import datetime, timezone
 from iatoolkit.services.user_session_context_service import UserSessionContextService
-from iatoolkit.services.query_service import QueryService
 
 
 class ProfileService:
@@ -25,50 +24,43 @@ class ProfileService:
     def __init__(self,
                  profile_repo: ProfileRepo,
                  session_context_service: UserSessionContextService,
-                 query_service: QueryService,
                  mail_app: MailApp):
         self.profile_repo = profile_repo
         self.session_context = session_context_service
-        self.query_service = query_service
         self.mail_app = mail_app
         self.bcrypt = Bcrypt()
 
 
     def login(self, company_short_name: str, email: str, password: str) -> dict:
         try:
-            # check if exits
+            # check if user exists
             user = self.profile_repo.get_user_by_email(email)
             if not user:
-                return {"error": "Usuario no encontrado"}
+                return {'success': False, "message": "Usuario no encontrado"}
 
             # check the encrypted password
             if not check_password_hash(user.password, password):
-                return {"error": "Contraseña inválida"}
+                return {'success': False, "message": "Contraseña inválida"}
 
             company = self.get_company_by_short_name(company_short_name)
             if not company:
-                return {"error": "Empresa no encontrada"}
+                return {'success': False, "message": "Empresa no encontrada"}
 
-            # check that user belongs to  company
+            # check that user belongs to company
             if company not in user.companies:
-                return {"error": "Usuario no esta autorizado para esta empresa"}
+                return {'success': False, "message": "Usuario no esta autorizado para esta empresa"}
 
             if not user.verified:
-                return {"error": "Tu cuenta no ha sido verificada. Por favor, revisa tu correo."}
+                return {'success': False,
+                        "message": "Tu cuenta no ha sido verificada. Por favor, revisa tu correo."}
 
-            # clear history save user data into session manager
+            # save user data into session manager
             self.set_user_session(user=user, company=company)
 
-            # initialize company context
-            self.query_service.llm_init_context(
-                company_short_name=company_short_name,
-                local_user_id=user.id
-            )
-
-            return {"message": "Login exitoso"}
+            return {'success': True, "user": user, "message": "Login exitoso"}
         except Exception as e:
-            logging.exception(f"login error: {str(e)}")
-            return {"error": str(e)}
+            return {'success': False, "message": str(e)}
+
 
     def set_user_session(self, user: User, company: Company):
         SessionManager.set('user_id', user.id)

iatoolkit/templates/base.html

@@ -9,9 +9,9 @@
     <link href="https://cdn.jsdelivr.net/npm/bootstrap-icons/font/bootstrap-icons.css" rel="stylesheet">
     <link rel="stylesheet" href="https://cdn.jsdelivr.net/npm/filepond/dist/filepond.min.css">
     <link rel="stylesheet" href="https://cdn.jsdelivr.net/npm/sweetalert2@11/dist/sweetalert2.min.css">
-    <link rel="stylesheet" href="{{ url_for('static', filename='styles/chat_iatoolkit.css') }}">
-    <link rel="stylesheet" href="{{ url_for('static', filename='styles/chat_modal.css') }}">
-    <link rel="stylesheet" href="{{ url_for('static', filename='styles/llm_output.css') }}">
+    <link rel="stylesheet" href="{{ url_for('static', filename='styles/chat_iatoolkit.css', _external=True) }}">
+    <link rel="stylesheet" href="{{ url_for('static', filename='styles/chat_modal.css',  _external=True) }}">
+    <link rel="stylesheet" href="{{ url_for('static', filename='styles/llm_output.css', _external=True) }}">
 </head>
 <body class="d-flex flex-column p-3" style="min-height: 100vh;">
     <main class="d-flex flex-column flex-grow-1">

iatoolkit/templates/chat.html

@@ -26,7 +26,7 @@
     <div class="d-flex align-items-center">
         <!-- 1. ID de Usuario -->
         <span style="{{ branding.secondary_text_style }}">
-            {{ external_user_id or user.email }}
+            {{ external_user_id or user_email }}
         </span>
 
         <!-- 2. Separador Vertical -->
@@ -43,8 +43,8 @@
         </a>
 
         <!-- Icono de cerrar sesión (al final) -->
-        {% if user.email %}
-            <a href="{{ url_for('logout', company_short_name=company_short_name) }}"
+        {% if user_email %}
+            <a href="{{ url_for('logout', company_short_name=company_short_name, _external=True) }}"
                  class="ms-3 action-icon-style" title="Cerrar sesión" style="color: {{ branding.header_text_color }} !important;">
                 <i class="bi bi-box-arrow-right"></i>
             </a>
@@ -173,10 +173,10 @@
 </script>
 
 <!-- Carga de los scripts JS externos después de definir las variables globales -->
-<script src="{{ url_for('static', filename='js/chat_filepond.js') }}"></script>
-<script src="{{ url_for('static', filename='js/chat_history.js') }}"></script>
-<script src="{{ url_for('static', filename='js/chat_feedback.js') }}"></script>
-<script src="{{ url_for('static', filename='js/chat_main.js') }}"></script>
+<script src="{{ url_for('static', filename='js/chat_filepond.js', _external=True) }}"></script>
+<script src="{{ url_for('static', filename='js/chat_history.js', _external=True) }}"></script>
+<script src="{{ url_for('static', filename='js/chat_feedback.js', _external=True) }}"></script>
+<script src="{{ url_for('static', filename='js/chat_main.js', _external=True) }}"></script>
 
 <script>
     document.addEventListener('DOMContentLoaded', function() {

iatoolkit/templates/home.html

@@ -13,7 +13,7 @@
       <div class="border rounded p-4 shadow-sm bg-light">
         <h4 class="text-muted fw-semibold text-start mb-3">login integrado (IAToolkit)</h4>
         <form id="login-form"
-              action="{{ url_for('home', company_short_name=company_short_name) }}"
+              action="{{ url_for('initiate_login', company_short_name=company_short_name) }}"
               method="post">
           <div class="mb-3">
             <label for="company_short_name" class="form-label d-block text-muted">Empresa</label>
@@ -112,7 +112,7 @@
 
           // Actualizar action del formulario "Iniciar Sesión"
           if (selectedCompany && selectedCompany.trim() !== '') {
-              const loginAction = '/' + selectedCompany + '/login';
+              const loginAction = '/' + selectedCompany + '/initiate_login';
               $('#login-form').attr('action', loginAction); // Actualizamos la URL del form
           } else {
               $('#login-form').attr('action', '#'); // URL genérica si no hay selección
@@ -156,12 +156,7 @@
               return;
           }
 
-          const $button = $(this);
-          const $spinner = $button.find('.spinner-border');
-          $button.prop('disabled', true);
-          $spinner.removeClass('d-none');
-
-          fetch(`/${selectedCompany}/chat_login`, {
+          fetch(`/${selectedCompany}/initiate_external_chat`, {
               method: 'POST',
               headers: {
                   'Content-Type': 'application/json',
@@ -188,10 +183,6 @@
           .catch(error => {
               Swal.fire({ icon: 'error', title: 'Error de Inicio de Sesión', text: error.message });
           })
-          .finally(() => {
-              $button.prop('disabled', false);
-              $spinner.addClass('d-none');
-          });
 
        });
   });

iatoolkit/templates/login_shell.html

@@ -0,0 +1,185 @@
+<!-- templates/iatoolkit/shell.html -->
+<!DOCTYPE html>
+<html lang="es">
+<head>
+    <meta charset="UTF-8">
+    <title>Iniciando Maxxa IA...</title>
+    <script src="https://ajax.googleapis.com/ajax/libs/jquery/3.5.1/jquery.min.js"></script>
+    <link rel="stylesheet" href="https://cdnjs.cloudflare.com/ajax/libs/font-awesome/5.15.4/css/all.min.css">
+
+    <!-- Inyecta las variables CSS de la marca -->
+    {% if branding and branding.css_variables %}
+    <style>
+        {{ branding.css_variables|safe }}
+    </style>
+    {% endif %}
+
+    <style>
+        /* --- Estilos Generales --- */
+        body, html { margin: 0; padding: 0; height: 100%; overflow: hidden; font-family: -apple-system, BlinkMacSystemFont, "Segoe UI", Roboto, "Helvetica Neue", Arial, sans-serif; }
+
+        #loader-wrapper {
+            position: fixed; top: 0; left: 0; width: 100%; height: 100%;
+            background-color: #f4f7f6;
+            z-index: 1000;
+            display: flex;
+            justify-content: center;
+            align-items: center;
+            flex-direction: column;
+            padding: 20px;
+            box-sizing: border-box;
+            transition: opacity 0.5s ease-in-out;
+        }
+
+        /* --- Estilos de Branding --- */
+        #brand-header {
+            font-size: 2.5rem;
+            font-weight: 700;
+            margin: 0 0 30px 0; /* Aumentamos el margen inferior para dar espacio */
+            color: var(--brand-secondary-color, #06326B);
+        }
+        #brand-header .brand-name {
+            color: var(--brand-primary-color, #FF5100);
+        }
+
+
+        /* --- Estilos de la Tarjeta (Sin cambios) --- */
+        #card-container {
+            background-color: #fff; border-radius: 12px; box-shadow: 0 4px 20px rgba(0, 0, 0, 0.1);
+            padding: 30px; width: 90%; max-width: 450px; text-align: center; transition: opacity 0.3s ease-in-out;
+        }
+        #card-container .icon { font-size: 40px; color: var(--brand-primary-color, #FF5100); margin-bottom: 15px; }
+        #card-container h3 { font-size: 1.25rem; color: #333; margin-bottom: 10px; }
+        #card-container p { font-size: 0.95rem; color: #666; line-height: 1.5; min-height: 60px; }
+        .card-nav { display: flex; justify-content: space-between; align-items: center; margin-top: 20px; }
+        .card-nav button { background-color: var(--brand-secondary-color, #06326B); border: none; color: var(--brand-text-on-secondary, #FFFFFF); border-radius: 50%; width: 40px; height: 40px; cursor: pointer; transition: opacity 0.2s; }
+        .card-nav button:hover { opacity: 0.85; }
+        #progress-dots { display: flex; gap: 8px; }
+        .dot { width: 10px; height: 10px; border-radius: 50%; background-color: #ddd; transition: background-color 0.3s; }
+        .dot.active { background-color: var(--brand-primary-color, #FF5100); }
+
+        /* --- ESTILOS MEJORADOS: SPINNER DE CARGA --- */
+        #loading-status { margin-top: 30px; display: flex; align-items: center; gap: 15px; }
+        .spinner {
+            width: 30px; height: 30px; border: 4px solid rgba(0, 0, 0, 0.1);
+            border-top-color: var(--brand-primary-color, #FF5100);
+            border-radius: 50%;
+            animation: spin 1s linear infinite;
+        }
+        #loading-status p { font-size: 1rem; font-weight: 500; color: #555; margin: 0; }
+        @keyframes spin { to { transform: rotate(360deg); } }
+
+        /* --- Iframe (Sin cambios) --- */
+        #content-container { width: 100%; height: 100%; }
+        iframe { width: 100%; height: 100%; border: none; }
+    </style>
+</head>
+<body>
+    <div id="loader-wrapper">
+
+        <h1 id="brand-header">
+            <span class="brand-name">Maxxa</span><span>IA</span>
+        </h1>
+
+        <!-- ELIMINADO: El subtítulo de carga que estaba aquí -->
+
+        <div id="card-container">
+            <div id="card-icon" class="icon"><i class="fas fa-lightbulb"></i></div>
+            <h3 id="card-title">Título de la Tarjeta</h3>
+            <p id="card-text">Descripción de la tarjeta de capacitación.</p>
+            <div class="card-nav">
+                <button id="prev-card" aria-label="Anterior"><i class="fas fa-chevron-left"></i></button>
+                <div id="progress-dots"></div>
+                <button id="next-card" aria-label="Siguiente"><i class="fas fa-chevron-right"></i></button>
+            </div>
+        </div>
+
+        <!-- MEJORADO: Texto de estado ahora junto al spinner -->
+        <div id="loading-status">
+            <div class="spinner"></div>
+            <p>Inicializando el contexto de Maxxa para la IA...</p>
+        </div>
+
+    </div>
+
+    <div id="content-container"></div>
+
+    <script>
+        $(function() {
+            const cardsData = [
+                { icon: 'fas fa-users', title: 'Clientes',
+                    text: 'Conozco en detalle a los clientes de Maxxa: antiguedad, contactos, apoderados, historial de operaciones.<br><br><strong>Ejemplo:</strong> cuantos clientes nuevos de garantia se incorporaron a mi cartera este año?' },
+                { icon: 'fas fa-cubes', title: 'Productos',
+                    text: 'Productos contratados por los clientes: garantias, credito en cuotas, software.<br><br><strong>Ejemplo:</strong> Cuantos clientes de software tengo en cartera?' },
+                { icon: 'fas fa-exchange-alt', title: 'Operaciones',
+                    text: 'Operaciones de garantia y créditos: tasas, comisiones, acredores, fondos, cobranza, etc.<br><br><strong>Ejemplo:</strong> Dame una tabla con los clientes de mi cartera han emitido mas de 20 garantias este año? columnas: rut, nombre, #garantias, monto, comisión' },
+                { icon: 'fas fa-landmark', title: 'Chilecompra',
+                    text: 'Historial completo de la participacion en chilecompra de un cliente. <br><br><strong>Ejemplo:</strong> Que porcentaje de garantias FC adjudicadas en los últimos 12 meses por el cliente 1234567-8 las compro en Maxxa.' },
+                { icon: 'fas fa-comments', title: 'Equipos Comerciales',
+                    text: 'Conozco los equipos comerciales de crédito y garantia. <br><br><strong>Ejemplo:</strong> dime las 10 licitaciones mas grandes que han ganado clientes de mi cartera este año.' },
+                { icon: 'fas fa-cogs', title: 'Personaliza tus Prompts',
+                    text: 'Utiliza la varita magica y podras explorar los prompts predefinidos que he preparado para ti.' },
+                { icon: 'fas fa-table', title: 'Tablas y Excel',
+                    text: 'Puedes pedirme la respuesta en formato de tablas o excel. <br><br><strong>Ejemplo:</strong> dame una tabla con los 10 certificados mas grande este año, columnas: rut, cliente, fecha, monto, tasa, comision, acreedor...' },
+                { icon: 'fas fa-shield-alt', title: 'Seguridad y Confidencialidad',
+                    text: 'Toda tu información es procesada de forma segura y confidencial dentro de nuestro entorno protegido.' }
+            ];
+            let currentCardIndex = 0, autoRotateInterval;
+            const $cardContainer = $('#card-container'), $cardIcon = $('#card-icon'), $cardTitle = $('#card-title'), $cardText = $('#card-text'), $progressDots = $('#progress-dots');
+            function displayCard(index) {
+                $cardContainer.css('opacity', 0);
+                setTimeout(() => {
+                    const card = cardsData[index];
+                    $cardIcon.html(`<i class="${card.icon}"></i>`);
+                    $cardTitle.text(card.title);
+                    $cardText.html(card.text);
+                    $progressDots.find('.dot').removeClass('active').eq(index).addClass('active');
+                    $cardContainer.css('opacity', 1);
+                }, 300);
+            }
+            function startAutoRotate() { autoRotateInterval = setInterval(() => $('#next-card').click(), 5000); }
+            cardsData.forEach(() => $progressDots.append('<div class="dot"></div>'));
+            displayCard(currentCardIndex);
+            startAutoRotate();
+            $('#next-card').on('click', function() {
+                currentCardIndex = (currentCardIndex + 1) % cardsData.length;
+                displayCard(currentCardIndex);
+                clearInterval(autoRotateInterval); startAutoRotate();
+            });
+            $('#prev-card').on('click', function() {
+                currentCardIndex = (currentCardIndex - 1 + cardsData.length) % cardsData.length;
+                displayCard(currentCardIndex);
+                clearInterval(autoRotateInterval); startAutoRotate();
+            });
+
+                    const $loader = $('#loader-wrapper');
+        const $container = $('#content-container');
+
+        // URL para el iframe, pasada desde la vista InitiateExternalChatView
+        const iframeSrc = "{{ iframe_src_url }}";
+
+        // Creamos el elemento iframe
+        const iframe = document.createElement('iframe');
+        iframe.src = iframeSrc;
+
+        // Estilos para que ocupe toda la pantalla
+        iframe.style.width = '100%';
+        iframe.style.height = '100%';
+        iframe.style.border = 'none';
+        iframe.style.display = 'none'; // Empezamos oculto
+
+        // Evento que se dispara cuando el iframe ha terminado de cargar su contenido
+        iframe.onload = function() {
+            // Mostramos el iframe
+            iframe.style.display = 'block';
+            // Ocultamos la animación de carga con una transición suave
+            $loader.css('opacity', 0);
+            setTimeout(() => $loader.hide(), 500); // Lo eliminamos del DOM después de la transición
+        };
+
+        // Añadimos el iframe al contenedor en el DOM
+        $container.append(iframe);
+    });
+    </script>
+</body>
+</html>

iatoolkit/views/history_view.py

@@ -34,7 +34,7 @@ class HistoryView(MethodView):
             return jsonify(iaut), 401
 
         external_user_id = data.get("external_user_id")
-        local_user_id = data.get("local_user_id", 0)
+        local_user_id = iaut.get("local_user_id", 0)
 
         try:
             response = self.history_service.get_history(

iatoolkit/views/{external_chat_login_view.py → login_external_id_view.py}

@@ -5,7 +5,7 @@
 
 import os
 import logging
-from flask import request, jsonify, render_template
+from flask import request, jsonify, render_template, url_for, session
 from flask.views import MethodView
 from injector import inject
 from iatoolkit.common.auth import IAuthentication
@@ -15,22 +15,16 @@ from iatoolkit.services.prompt_manager_service import PromptService
 from iatoolkit.services.jwt_service import JWTService
 from iatoolkit.services.branding_service import BrandingService
 
-class ExternalChatLoginView(MethodView):
+class InitiateExternalChatView(MethodView):
     @inject
     def __init__(self,
-                 profile_service: ProfileService,
-                 query_service: QueryService,
-                 prompt_service: PromptService,
                  iauthentication: IAuthentication,
-                 jwt_service: JWTService,
-                 branding_service: BrandingService
+                 branding_service: BrandingService,
+                 profile_service: ProfileService
                  ):
-        self.profile_service = profile_service
-        self.query_service = query_service
-        self.prompt_service = prompt_service
         self.iauthentication = iauthentication
-        self.jwt_service = jwt_service
         self.branding_service = branding_service
+        self.profile_service = profile_service
 
     def post(self, company_short_name: str):
         data = request.get_json()
@@ -39,7 +33,11 @@ class ExternalChatLoginView(MethodView):
 
         external_user_id = data['external_user_id']
 
-        # 1. get access credentials
+        company = self.profile_service.get_company_by_short_name(company_short_name)
+        if not company:
+            return jsonify({"error": "Empresa no encontrada"}), 404
+
+        # 1. verify access credentials quickly
         iaut = self.iauthentication.verify(
             company_short_name,
             body_external_user_id=external_user_id
@@ -47,11 +45,51 @@ class ExternalChatLoginView(MethodView):
         if not iaut.get("success"):
             return jsonify(iaut), 401
 
+        # 2. Get branding data for the shell page
+        branding_data = self.branding_service.get_company_branding(company)
+
+        # Generamos la URL para el SRC del iframe, añadiendo el usuario como un query parameter.
+        target_url = url_for('external_login',  # Apunta a la vista del chat
+                             company_short_name=company_short_name,
+                             external_user_id=external_user_id,  # Se añadirá como ?external_user_id=...
+                             _external=True)
+
+        # Renderizamos el shell para un iframe.
+        return render_template("login_shell.html",
+                               iframe_src_url=target_url,  # Le cambiamos el nombre para más claridad
+                               branding=branding_data
+                               )
+
+class ExternalChatLoginView(MethodView):
+    @inject
+    def __init__(self,
+                 profile_service: ProfileService,
+                 query_service: QueryService,
+                 prompt_service: PromptService,
+                 iauthentication: IAuthentication,
+                 jwt_service: JWTService,
+                 branding_service: BrandingService
+                 ):
+        self.profile_service = profile_service
+        self.query_service = query_service
+        self.prompt_service = prompt_service
+        self.iauthentication = iauthentication
+        self.jwt_service = jwt_service
+        self.branding_service = branding_service
+
+    def get(self, company_short_name: str):
+        # Leemos el user_id desde los parámetros de la URL (?external_user_id=...)
+        external_user_id = request.args.get('external_user_id')
+        if not external_user_id:
+            return "Falta el parámetro external_user_id en la URL", 400
+
         company = self.profile_service.get_company_by_short_name(company_short_name)
         if not company:
+            logging.error(f'Company {company_short_name} not found')
             return jsonify({"error": "Empresa no encontrada"}), 404
 
         try:
+
             # 1. generate a new JWT, our secure access token.
             token = self.jwt_service.generate_chat_jwt(
                 company_id=company.id,
@@ -75,20 +113,15 @@ class ExternalChatLoginView(MethodView):
             branding_data = self.branding_service.get_company_branding(company)
 
             # 5. render the chat page with the company/user information.
-            user_agent = request.user_agent
-            is_mobile = user_agent.platform in ["android", "iphone", "ipad"] or "mobile" in user_agent.string.lower()
-
-            chat_html = render_template("chat.html",
+            return render_template("chat.html",
                                         company_short_name=company_short_name,
-                                        branding=branding_data,
+                                        auth_method='jwt',
+                                        session_jwt=token,
                                         external_user_id=external_user_id,
-                                        is_mobile=is_mobile,
-                                        auth_method='jwt',  # login method is JWT
-                                        session_jwt=token,  # pass the token to the front-end
-                                        iatoolkit_base_url=os.getenv('IATOOLKIT_BASE_URL'),
+                                        branding=branding_data,
                                         prompts=prompts,
-                                        external_login=True)
-            return chat_html, 200
+                                        iatoolkit_base_url=os.getenv('IATOOLKIT_BASE_URL'),
+                                        ), 200
 
         except Exception as e:
             logging.exception(f"Error al inicializar el chat para {company_short_name}/{external_user_id}: {e}")

iatoolkit/views/login_view.py

@@ -6,12 +6,82 @@
 from flask.views import MethodView
 from flask import request, redirect, render_template, url_for
 from injector import inject
+from iatoolkit.repositories.models import User
 from iatoolkit.services.profile_service import ProfileService
+from iatoolkit.services.prompt_manager_service import PromptService
+from iatoolkit.services.branding_service import BrandingService
+from iatoolkit.services.query_service import QueryService
+import os
+from iatoolkit.common.session_manager import SessionManager
+from iatoolkit.services.branding_service import BrandingService
+
+class InitiateLoginView(MethodView):
+    """
+    Handles the initial, fast part of the standard login process.
+    Authenticates user credentials, sets up the server-side session,
+    and immediately returns the loading shell page.
+    """
+    @inject
+    def __init__(self,
+                 profile_service: ProfileService,
+                 branding_service: BrandingService,):
+        self.profile_service = profile_service
+        self.branding_service = branding_service
+
+    def post(self, company_short_name: str):
+        # get company info
+        company = self.profile_service.get_company_by_short_name(company_short_name)
+        if not company:
+            return render_template('error.html',
+                                   message="Empresa no encontrada"), 404
+
+        email = request.form.get('email')
+        password = request.form.get('password')
+
+        # 1. authenticate the user
+        response = self.profile_service.login(
+            company_short_name=company_short_name,
+            email=email,
+            password=password
+        )
+
+        if not response['success']:
+            return render_template(
+                'login.html',
+                company_short_name=company_short_name,
+                company=company,
+                form_data={
+                    "email": email,
+                    "password": password,
+                },
+                alert_message=response["error"]), 400
+
+        # 2. Get branding data for the shell page
+        branding_data = self.branding_service.get_company_branding(company)
+
+        # 3. Render the shell page, passing the URL for the heavy lifting
+        # The shell's AJAX call will now be authenticated via the session cookie.
+        return render_template(
+            "login_shell.html",
+            data_source_url=url_for('login',
+                        company_short_name=company_short_name,
+                        _external=True),
+            external_user_id='',
+            branding=branding_data,
+        )
+
 
 class LoginView(MethodView):
     @inject
-    def __init__(self, profile_service: ProfileService):
+    def __init__(self,
+                 profile_service: ProfileService,
+                 query_service: QueryService,
+                 prompt_service: PromptService,
+                 branding_service: BrandingService):
         self.profile_service = profile_service
+        self.query_service = query_service
+        self.prompt_service = prompt_service
+        self.branding_service = branding_service
 
     def get(self, company_short_name: str):
         # get company info
@@ -24,33 +94,38 @@ class LoginView(MethodView):
                                company_short_name=company_short_name)
 
     def post(self, company_short_name: str):
-        # get company info
         company = self.profile_service.get_company_by_short_name(company_short_name)
-        if not company:
-            return render_template('error.html', message="Empresa no encontrada"), 404
 
-        email = request.form.get('email')
-        try:
-            password = request.form.get('password')
+        # 1. The user is already authenticated by the session cookie set by InitiateLoginView.
+        # We just retrieve the user and company IDs from the session.
+        user_id = SessionManager.get('user_id')
+        if not user_id:
+            return render_template('error.html', message="Usuario no encontrado"), 404
+
+        user_email = SessionManager.get('user')['email']
 
-            response = self.profile_service.login(
+        try:
+            # 2. init the company/user LLM context.
+            self.query_service.llm_init_context(
                 company_short_name=company_short_name,
-                email=email,
-                password=password
-                )
+                local_user_id=user_id
+            )
 
-            if "error" in response:
-                return render_template(
-                'login.html',
+            # 3. get the prompt list from backend
+            prompts = self.prompt_service.get_user_prompts(company_short_name)
+
+            # 4. get the branding data
+            branding_data = self.branding_service.get_company_branding(company)
+
+            return render_template("chat.html",
                                     company_short_name=company_short_name,
-                                    company=company,
-                                    form_data={
-                                           "email": email,
-                                           "password": password,
-                                       },
-                                    alert_message=response["error"]), 400
-
-            return redirect(url_for('chat', company_short_name=company_short_name))
+                                    auth_method="Session",
+                                    session_jwt=None,  # No JWT in this flow
+                                    user_email=user_email,
+                                    branding=branding_data,
+                                    prompts=prompts,
+                                    iatoolkit_base_url=os.getenv('IATOOLKIT_BASE_URL'),
+                                    ), 200
 
         except Exception as e:
             return render_template("error.html",

{iatoolkit-0.11.1.dist-info → iatoolkit-0.14.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: iatoolkit
-Version: 0.11.1
+Version: 0.14.0
 Summary: IAToolkit
 Author: Fernando Libedinsky
 License-Expression: MIT

{iatoolkit-0.11.1.dist-info → iatoolkit-0.14.0.dist-info}/RECORD

@@ -2,11 +2,11 @@ iatoolkit/__init__.py,sha256=4PWjMJjktixtrxF6BY405qyA50Sv967kEP2x-oil6qk,1120
 iatoolkit/base_company.py,sha256=4usUdOC57Nk1XsawAEyC7mlzb0jROadu64YcYf6poiM,4371
 iatoolkit/cli_commands.py,sha256=G5L9xQXZ0lVFXQWBaE_KEZHyfuiT6PL1nTQRoSdnBzc,2302
 iatoolkit/company_registry.py,sha256=tduqt3oV8iDX_IB1eA7KIgvIxE4edTcy-3qZIXh3Lzw,2549
-iatoolkit/iatoolkit.py,sha256=EXmHQEgLCqbBP7Cwq6-O1NPj7O3M_2Cg9JLp5xa41ig,16960
+iatoolkit/iatoolkit.py,sha256=9OIBCsHGAQGGIzExP95KFxND1bxaRuYQ08oHkbxv8WM,16996
 iatoolkit/common/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 iatoolkit/common/auth.py,sha256=8NH6MQXfddLQd1GxrO2op3IYrrP4SMQKoKzj1o1jZmc,8486
 iatoolkit/common/exceptions.py,sha256=EXx40n5htp7UiOM6P1xfJ9U6NMcADqm62dlFaKz7ICU,1154
-iatoolkit/common/routes.py,sha256=8_wvrFiB5w5Sq-5gZfIDz9JVPROd9JED5oxkX1N87gY,4923
+iatoolkit/common/routes.py,sha256=mzbljACSZeJmJ6S9BZvTTRujTRnrEU-VKTqNdmfW3hs,4904
 iatoolkit/common/session_manager.py,sha256=7D_RuJs60w-1zDr3fOGEz9JW7IZlSXuUHgUT87CzaUo,472
 iatoolkit/common/util.py,sha256=08js3KLJTXICOd5sgwDp2u_kDaZO_0xG4BIuzWZnLo8,15535
 iatoolkit/infra/__init__.py,sha256=5JqK9sZ6jBuK83zDQokUhxQ0wuJJJ9DXB8pYCLkX7X4,102
@@ -45,7 +45,7 @@ iatoolkit/services/history_service.py,sha256=ZlYfaSHOcCxxc6ICTnqflaGBlzctpJdNUwX
 iatoolkit/services/jwt_service.py,sha256=YoZ9h7_o9xBko-arNQv4MbcwnxoSWVNj4VbZmMo_QGY,3908
 iatoolkit/services/load_documents_service.py,sha256=ZpB0BZ3qX1fGJGBtZtMLbFdWWx0hkPoeCS3OqJKwCTs,7291
 iatoolkit/services/mail_service.py,sha256=2h-fcF3swZDya_o7IpgXkmuj3iEVHVCiHi7oVxU99sQ,2182
-iatoolkit/services/profile_service.py,sha256=azaCQtRJwQayg-3NV6k0lX97SFiF12emhz60S4CWv-M,17621
+iatoolkit/services/profile_service.py,sha256=Mxt_Hdz-SyYDuwWVu10hZnb00dKaOeONX8zp1TzpQt4,17387
 iatoolkit/services/prompt_manager_service.py,sha256=U-XmSpkeXvv1KRN4dytdMxSYBMRSB7y-UHcb18mk0nA,8342
 iatoolkit/services/query_service.py,sha256=CI_LUBdqolvsuh7TjY23oaQcc8s6rmmr4LqnjE81ZJc,15394
 iatoolkit/services/search_service.py,sha256=i1xGWu7ORKIIDH0aAQBkF86dVVbLQ0Yrooz5TiZ6aGo,1823
@@ -74,37 +74,36 @@ iatoolkit/system_prompts/format_styles.prompt,sha256=MSMe1qvR3cF_0IbFshn8R0z6Wx6
 iatoolkit/system_prompts/query_main.prompt,sha256=w_9ybgWgiQH4V_RbAXqsvz0M7oOuiyhxcwf-D0CgfA4,3017
 iatoolkit/system_prompts/sql_rules.prompt,sha256=y4nURVnb9AyFwt-lrbMNBHHtZlhk6kC9grYoOhRnrJo,59174
 iatoolkit/templates/about.html,sha256=ciC08grUVz5qLzdzDDqDX31xirg5PrJIRYabWpV9oA8,294
-iatoolkit/templates/base.html,sha256=LXXB8oPrcBFkf2pLfOSyAaSh66kHbs4SEujpFL3h9Nw,2163
+iatoolkit/templates/base.html,sha256=TojvSnVvXkTe7Kpt_BBWoXFfZN6dveKD0VqQjUOXdgU,2212
 iatoolkit/templates/change_password.html,sha256=DFfQSFcZ2YJZNFis2IXfzEKStxTf4i9f4eQ_6GiyNs8,2342
-iatoolkit/templates/chat.html,sha256=xaQk5McHRO8VF2LCBbr4vT0d2fFNy2GW61DwCYht-_M,9371
+iatoolkit/templates/chat.html,sha256=pQm1vDikaSsvleJK0LglL1BAeoLbwPLe4FYtLIsW4T8,9451
 iatoolkit/templates/chat_modals.html,sha256=3CQ430bwhebq6rAJ6Bk12PQDjt9YenqNXm5thC5WP2Y,5771
 iatoolkit/templates/error.html,sha256=BNF-7z8AYL5vF4ZMUFMrOBt8c85kCFrm9qSHn9EiHWg,540
 iatoolkit/templates/forgot_password.html,sha256=1lUbKg9CKnQdnySplceY_pibwYne1-mOlM38fqI1kW8,1563
 iatoolkit/templates/header.html,sha256=179agI7rnYwP_rvJNXIiVde5E8Ec5649_XKq6eew2Hk,1263
-iatoolkit/templates/home.html,sha256=HpJmz0rCMBt3XYd83hecBCESvMg85n56PV5XvE_8Q_g,8201
+iatoolkit/templates/home.html,sha256=UWtAn19_S-k6e8_UoWBFeeELbMhJRxXLwxNH7AC4Lf0,7921
 iatoolkit/templates/login.html,sha256=r4hy7MsQkfDqi6pBRNkkRiFr3GPSoHCT89R5lQLUWZc,1991
+iatoolkit/templates/login_shell.html,sha256=d_dAgbZ2JqCc96FDX_kA6sRofAZ_qc7D1N6Zyto9ZJ8,9727
 iatoolkit/templates/signup.html,sha256=J8wOjUhUe_KozyThDTWHjXpSJ1ubR2IDVobThtkSRuo,3819
 iatoolkit/templates/test.html,sha256=rwNtxC83tbCl5COZFXYvmRBxxmgFJtPNuVBd_nq9KWY,133
 iatoolkit/views/__init__.py,sha256=5JqK9sZ6jBuK83zDQokUhxQ0wuJJJ9DXB8pYCLkX7X4,102
 iatoolkit/views/change_password_view.py,sha256=rSebwecI1zwBgR2yvAhcfMwWpGDa4QbVAIllgtSOo9k,3940
 iatoolkit/views/chat_token_request_view.py,sha256=wf32_A2Sq8NHYWshCwL10Tovd1znLoD0jQjzutR3sVE,4408
-iatoolkit/views/chat_view.py,sha256=BD1N-Jo9IOoS6baX1bzdM7DKa2zepVHZnJIxiF0vNc8,2381
 iatoolkit/views/download_file_view.py,sha256=1gZ0ipqeCn39sTrJFo1-tlewlcSF7s_YNTvE4qd0HOw,2010
-iatoolkit/views/external_chat_login_view.py,sha256=h-Xr7UkQM7QxLZIDQ0fXb-BUfPbABQ1gjiP--FrXnpc,4075
-iatoolkit/views/external_login_view.py,sha256=EHxN2omqTDkWyMp0FVCNDFt-JJ2hpg8LvprKanGR1gY,1355
 iatoolkit/views/file_store_view.py,sha256=hUm5wX4E5oqJJEPEAObEj-nPiRp5EJIICULSfAWmHCs,1933
 iatoolkit/views/forgot_password_view.py,sha256=Rk8Qbe9Fz7Wlgje1rt29I15gFM-a089EBi2at4FT7kA,2715
-iatoolkit/views/history_view.py,sha256=8sqc9i8hN5bbRs3mSjurqKEZaXKANTQqATfwVAS4qSM,2064
+iatoolkit/views/history_view.py,sha256=fzZrnC-RySa7ngcPe2Hmf9_s3imx6VB6MKROMcNpjoU,2064
 iatoolkit/views/home_view.py,sha256=TihO2flkelJa9j6a0FKCMVhD-2X7BhemonB7LTne4x8,1248
 iatoolkit/views/llmquery_view.py,sha256=rv2i3oeXlNc3Sv7Qu3DZGf37r-bMSa4N25FzG7_kPAI,2432
-iatoolkit/views/login_view.py,sha256=MI87rGB1QpxEVxWy-tR2q7s9ms9pcFLIW8ZmNQh1EFE,2332
+iatoolkit/views/login_external_id_view.py,sha256=bkpsGFLxg9MPhRyLdvtfzlOIQFZgQVzNKPIJbnsHTPY,5524
+iatoolkit/views/login_view.py,sha256=mmJ-VMnT_VvQy5GkTGM8dDdG7yNjqpW-7YQgzkQmhQI,5396
 iatoolkit/views/prompt_view.py,sha256=l8KHlLmkSgSLK43VbhwKED7mCN9YyfeHHh4zvx0pT0E,1257
 iatoolkit/views/signup_view.py,sha256=NTx_2w8F6Np88FKEpDvBvJXU-bISKpDMdhhT4XFAVfk,3805
 iatoolkit/views/tasks_review_view.py,sha256=keLsLCyOTTlcoIapnB_lbuSvLwrPVZVpBiFC_7ChbLg,3388
 iatoolkit/views/tasks_view.py,sha256=a3anTXrJTTvbQuc6PSpOzidLKQFL4hWa7PI2Cppcz8w,4110
 iatoolkit/views/user_feedback_view.py,sha256=G37zmP8P4LvZrSymNJ5iFXhLZg1A3BEwRfTpH1Iam5w,2652
 iatoolkit/views/verify_user_view.py,sha256=a3q4wHJ8mKAEmgbNTOcnX4rMikROjOR3mHvCr30qGGA,2351
-iatoolkit-0.11.1.dist-info/METADATA,sha256=5vmAxDm05jysqCbNZe1lJqpc3N--YhpxBizGrQ1-CJ8,9301
-iatoolkit-0.11.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-iatoolkit-0.11.1.dist-info/top_level.txt,sha256=V_w4QvDx0b1RXiy8zTCrD1Bp7AZkFe3_O0-9fMiwogg,10
-iatoolkit-0.11.1.dist-info/RECORD,,
+iatoolkit-0.14.0.dist-info/METADATA,sha256=VYEM96_CE62ebzB8ScujIwbcjnC5T6qEzzr8hQep0Xw,9301
+iatoolkit-0.14.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+iatoolkit-0.14.0.dist-info/top_level.txt,sha256=V_w4QvDx0b1RXiy8zTCrD1Bp7AZkFe3_O0-9fMiwogg,10
+iatoolkit-0.14.0.dist-info/RECORD,,

iatoolkit/views/chat_view.py

@@ -1,58 +0,0 @@
-# Copyright (c) 2024 Fernando Libedinsky
-# Product: IAToolkit
-#
-# IAToolkit is open source software.
-
-from flask import render_template, request, jsonify
-from iatoolkit.services.profile_service import ProfileService
-from flask.views import MethodView
-from injector import inject
-import os
-from iatoolkit.common.auth import IAuthentication
-from iatoolkit.services.prompt_manager_service import PromptService
-from iatoolkit.services.branding_service import BrandingService
-
-
-class ChatView(MethodView):
-    @inject
-    def __init__(self,
-                 iauthentication: IAuthentication,
-                 prompt_service: PromptService,
-                 profile_service: ProfileService,
-                 branding_service: BrandingService
-                 ):
-        self.iauthentication = iauthentication
-        self.profile_service = profile_service
-        self.prompt_service = prompt_service
-        self.branding_service = branding_service
-
-    def get(self, company_short_name: str):
-        # get access credentials
-        iaut = self.iauthentication.verify(company_short_name)
-        if not iaut.get("success"):
-            return jsonify(iaut), 401
-
-        user_agent = request.user_agent
-        is_mobile = user_agent.platform in ["android", "iphone", "ipad"] or "mobile" in user_agent.string.lower()
-        alert_message = request.args.get('alert_message', None)
-
-        # 1. get company info
-        company = self.profile_service.get_company_by_short_name(company_short_name)
-        if not company:
-            return render_template('error.html', message="Empresa no encontrada"), 404
-
-        # 2. get the company prompts
-        prompts = self.prompt_service.get_user_prompts(company_short_name)
-
-        # 3. get the  branding data
-        branding_data = self.branding_service.get_company_branding(company)
-
-        return render_template("chat.html",
-                               branding=branding_data,
-                               company_short_name=company_short_name,
-                               is_mobile=is_mobile,
-                               alert_message=alert_message,
-                               alert_icon='success' if alert_message else None,
-                               iatoolkit_base_url=os.getenv('IATOOLKIT_BASE_URL', 'http://localhost:5000'),
-                               prompts=prompts
-                               )

iatoolkit/views/external_login_view.py

@@ -1,40 +0,0 @@
-# Copyright (c) 2024 Fernando Libedinsky
-# Product: IAToolkit
-#
-# IAToolkit is open source software.
-
-from flask.views import MethodView
-from injector import inject
-from iatoolkit.common.auth import IAuthentication
-from iatoolkit.services.query_service import QueryService
-from flask import jsonify
-import logging
-
-class ExternalLoginView(MethodView):
-
-    @inject
-    def __init__(self,
-                 iauthentication: IAuthentication,
-                 query_service: QueryService
-                 ):
-        self.iauthentication = iauthentication
-        self.query_service = query_service
-
-    def get(self, company_short_name: str, external_user_id: str):
-        # 1. get access credentials
-        iaut = self.iauthentication.verify(company_short_name, external_user_id)
-        if not iaut.get("success"):
-            return jsonify(iaut), 401
-
-        try:
-            # initialize the context
-            self.query_service.llm_init_context(
-                company_short_name=company_short_name,
-                external_user_id=external_user_id
-            )
-
-            return {'status': 'OK'}, 200
-        except Exception as e:
-            logging.exception(
-                f"Error inesperado al inicializar el contexto durante el login para company {company_short_name}: {e}")
-            return jsonify({"error_message": str(e)}), 500