iatoolkit 0.12.0__py3-none-any.whl → 0.16.0__py3-none-any.whl

iatoolkit/base_company.py

@@ -26,10 +26,16 @@ class BaseCompany(ABC):
         self.company = self.profile_repo.get_company_by_short_name(short_name)
         return self.company
 
-    def _create_company(self, short_name: str, name: str, branding: dict | None = None) -> Company:
+    def _create_company(self,
+                        short_name: str,
+                        name: str,
+                        branding: dict | None = None,
+                        onboarding_cards: dict | None = None
+                        ) -> Company:
         company_obj = Company(short_name=short_name,
                               name=name,
-                              branding=branding)
+                              branding=branding,
+                              onboarding_cards=onboarding_cards)
         self.company = self.profile_repo.create_company(company_obj)
         return self.company
 

iatoolkit/common/routes.py

@@ -27,7 +27,7 @@ def register_views(injector, app):
     from iatoolkit.views.tasks_review_view import TaskReviewView
     from iatoolkit.views.home_view import HomeView
     from iatoolkit.views.login_view import LoginView, InitiateLoginView
-    from iatoolkit.views.external_chat_login_view import InitiateExternalChatView, ExternalChatLoginView
+    from iatoolkit.views.external_login_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
@@ -36,21 +36,19 @@ 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'))
 
-    # 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'))
+    # login for external portals
     app.add_url_rule('/<company_short_name>/initiate_external_chat',
-                         view_func=InitiateExternalChatView.as_view('initiate_chat_login'))
+                         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'))
 
-    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'))
 

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"

iatoolkit/repositories/models.py

@@ -58,6 +58,7 @@ class Company(Base):
     gemini_api_key = Column(String, nullable=True)
 
     branding = Column(JSON, nullable=True)
+    onboarding_cards = Column(JSON, nullable=True)
     parameters = Column(JSON, nullable=True, default={})
     created_at = Column(DateTime, default=datetime.now)
     allow_jwt = Column(Boolean, default=True, nullable=True)

iatoolkit/repositories/profile_repo.py

@@ -74,6 +74,7 @@ class ProfileRepo:
         if company:
             company.parameters = new_company.parameters
             company.branding = new_company.branding
+            company.onboarding_cards = new_company.onboarding_cards
         else:
             self.session.add(new_company)
             company = new_company

iatoolkit/services/branding_service.py

@@ -44,6 +44,19 @@ class BrandingService:
             "brand_info_text": "#055160",  # Texto azul oscuro
             "brand_info_border": "#b6effb",
 
+            # Estilos para el Asistente de Prompts ---
+            "prompt_assistant_bg": "#f8f9fa",
+            "prompt_assistant_border": "#dee2e6",
+            "prompt_assistant_icon_color": "#6c757d",
+            "prompt_assistant_button_bg": "#FFFFFF",
+            "prompt_assistant_button_text": "#495057",
+            "prompt_assistant_button_border": "#ced4da",
+            "prompt_assistant_dropdown_bg": "#f8f9fa",
+            "prompt_assistant_header_bg": "#e9ecef",
+            "prompt_assistant_header_text": "#495057",
+            "prompt_assistant_item_hover_bg": None,  # Usará el primario por defecto
+            "prompt_assistant_item_hover_text": None,  # Usará el texto sobre primario
+
             # Color para el botón de Enviar ---
             "send_button_color": "#212529"  # Gris oscuro/casi negro por defecto
         }
@@ -93,6 +106,17 @@ class BrandingService:
                 --brand-info-bg: {final_branding_values['brand_info_bg']};
                 --brand-info-text: {final_branding_values['brand_info_text']};
                 --brand-info-border: {final_branding_values['brand_info_border']};
+                --brand-prompt-assistant-bg: {final_branding_values['prompt_assistant_bg']};
+                --brand-prompt-assistant-border: {final_branding_values['prompt_assistant_border']};
+                --brand-prompt-assistant-icon-color: {final_branding_values['prompt_assistant_icon_color']};
+                --brand-prompt-assistant-button-bg: {final_branding_values['prompt_assistant_button_bg']};
+                --brand-prompt-assistant-button-text: {final_branding_values['prompt_assistant_button_text']};
+                --brand-prompt-assistant-button-border: {final_branding_values['prompt_assistant_button_border']};
+                --brand-prompt-assistant-dropdown-bg: {final_branding_values['prompt_assistant_dropdown_bg']};
+                --brand-prompt-assistant-header-bg: {final_branding_values['prompt_assistant_header_bg']};
+                --brand-prompt-assistant-header-text: {final_branding_values['prompt_assistant_header_text']};
+                --brand-prompt-assistant-item-hover-bg: {final_branding_values['prompt_assistant_item_hover_bg'] or final_branding_values['brand_primary_color']};
+                --brand-prompt-assistant-item-hover-text: {final_branding_values['prompt_assistant_item_hover_text'] or final_branding_values['brand_text_on_primary']};
 
             }}
         """

iatoolkit/services/onboarding_service.py

@@ -0,0 +1,43 @@
+# Copyright (c) 2024 Fernando Libedinsky
+# Product: IAToolkit
+#
+# IAToolkit is open source software.
+
+from iatoolkit.repositories.models import Company
+from typing import List, Dict, Any
+
+
+class OnboardingService:
+    """
+    Servicio para gestionar las tarjetas de contenido que se muestran
+    durante la pantalla de carga (onboarding).
+    """
+
+    def __init__(self):
+        """
+        Define el conjunto de tarjetas de onboarding por defecto.
+        """
+        self._default_cards = [
+            {'icon': 'fas fa-users', 'title': 'Clientes',
+             'text': 'Conozco en detalle a nuestros clientes: antigüedad, contactos, historial de operaciones.<br><br><strong>Ejemplo:</strong> ¿cuántos clientes nuevos se incorporaron a mi cartera este año?'},
+            {'icon': 'fas fa-cubes', 'title': 'Productos',
+             'text': 'Productos: características, condiciones, historial.'},
+
+            {'icon': 'fas fa-cogs', 'title': 'Personaliza tus Prompts',
+             'text': 'Utiliza la varita mágica y podrás 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 más grandes este año.'},
+            {'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.'}
+        ]
+
+    def get_onboarding_cards(self, company: Company | None) -> List[Dict[str, Any]]:
+        """
+        Retorna la lista de tarjetas de onboarding para una compañía.
+        Si la compañía tiene tarjetas personalizadas, las devuelve.
+        De lo contrario, devuelve las tarjetas por defecto.
+        """
+        if company and company.onboarding_cards:
+            return company.onboarding_cards
+
+        return self._default_cards

iatoolkit/static/styles/chat_iatoolkit.css

@@ -168,7 +168,7 @@
 
 /* 1. La caja principal que envuelve toda el área de entrada */
 .input-area {
-    background-color: #f8f9fa;
+    background-color: var(--brand-prompt-assistant-bg, #f8f9fa);
 }
 
 /* 2. La barra "cápsula" que envuelve el texto y los iconos */
@@ -202,7 +202,7 @@
 
 #prompt-assistant-collapse .card {
     border-radius: 1.5rem;      /* Mismo radio que el chat-input-bar */
-    border: 1px solid #dee2e6; /* Mismo borde que el chat-input-bar */
+    border: 1px solid var(--brand-prompt-assistant-border, #dee2e6); /* Mismo borde que el chat-input-bar */
     box-shadow: none;           /* Eliminamos la sombra por defecto del card */
 }
 
@@ -241,6 +241,12 @@
     color: #6c757d;
     transition: color 0.2s ease-in-out;
 }
+
+/* Anulación específica para el icono de la varita mágica */
+.chat-input-bar a[href="#prompt-assistant-collapse"] i {
+    color: var(--brand-prompt-assistant-icon-color, #6c757d);
+}
+
 .chat-input-bar .d-flex a:hover i {
     color: #343a40;
 }
@@ -302,8 +308,8 @@
 }
 
 .dropdown-menu-soft {
-    background-color: #f8f9fa;
-    border-color: #dee2e6;
+    background-color: var(--brand-prompt-assistant-dropdown-bg, #f8f9fa);
+    border-color: var(--brand-prompt-assistant-border, #dee2e6);
 }
 
 .dropdown-menu-soft .dropdown-item {
@@ -312,15 +318,15 @@
 
 .dropdown-menu-soft .dropdown-item:hover,
 .dropdown-menu-soft .dropdown-item:focus {
-    color: #ffffff;
-    background-color: #495057;
+    color: var(--brand-prompt-assistant-item-hover-text, #ffffff);
+    background-color: var(--brand-prompt-assistant-item-hover-bg, #495057);
     padding-left: 1.5rem;
     transition: all 0.15s ease-in-out;
 }
 
 .dropdown-menu-soft .dropdown-header {
-    background-color: #495057;
-    color: #ffffff;
+    background-color: var(--brand-prompt-assistant-header-bg, #495057);
+    color: var(--brand-prompt-assistant-header-text, #ffffff);
     font-weight: 600;
     margin: 4px;
     padding: 0.4rem 1rem;
@@ -331,6 +337,14 @@
     border-bottom: none;
 }
 
+/* Estilo para el botón principal del asistente de prompts */
+#prompt-select-button {
+    background-color: var(--brand-prompt-assistant-button-bg, #FFFFFF);
+    color: var(--brand-prompt-assistant-button-text, #495057);
+    border-color: var(--brand-prompt-assistant-button-border, #ced4da);
+}
+
+
 #clear-selection-button {
     position: absolute;
     top: 50%;

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

@@ -44,7 +44,7 @@
 
         <!-- Icono de cerrar sesión (al final) -->
         {% if user_email %}
-            <a href="{{ url_for('logout', company_short_name=company_short_name) }}"
+            <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('initiate_login', company_short_name=company_short_name) }}"
+              action="{{ url_for('initiate_login', company_short_name=company_short_name, external_login=True) }}"
               method="post">
           <div class="mb-3">
             <label for="company_short_name" class="form-label d-block text-muted">Empresa</label>
@@ -156,11 +156,6 @@
               return;
           }
 
-          const $button = $(this);
-          const $spinner = $button.find('.spinner-border');
-          $button.prop('disabled', true);
-          $spinner.removeClass('d-none');
-
           fetch(`/${selectedCompany}/initiate_external_chat`, {
               method: 'POST',
               headers: {
@@ -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/onboarding_shell.html

@@ -0,0 +1,167 @@
+
+<!DOCTYPE html>
+<html lang="es">
+<head>
+    <meta charset="UTF-8">
+    <title>Iniciando {{ branding.name | default('IAToolkit') }} 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">{{ branding.name | default('IAToolkit') }}</span>  <span>IA</span>
+        </h1>
+
+        <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 = {{ onboarding_cards | tojson }};
+
+            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/external_login_view.py

@@ -3,38 +3,152 @@
 #
 # IAToolkit is open source software.
 
+import os
+import logging
+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
+from iatoolkit.services.profile_service import ProfileService
 from iatoolkit.services.query_service import QueryService
-from flask import jsonify
-import logging
+from iatoolkit.services.prompt_manager_service import PromptService
+from iatoolkit.services.jwt_service import JWTService
+from iatoolkit.services.branding_service import BrandingService
+from iatoolkit.services.onboarding_service import OnboardingService
+from iatoolkit.services.jwt_service import JWTService
 
-class ExternalLoginView(MethodView):
 
+class InitiateExternalChatView(MethodView):
     @inject
     def __init__(self,
                  iauthentication: IAuthentication,
-                 query_service: QueryService
+                 branding_service: BrandingService,
+                 profile_service: ProfileService,
+                 onboarding_service: OnboardingService,
+                 jwt_service: JWTService
                  ):
         self.iauthentication = iauthentication
-        self.query_service = query_service
+        self.branding_service = branding_service
+        self.profile_service = profile_service
+        self.onboarding_service = onboarding_service
+        self.jwt_service = jwt_service
+
+    def post(self, company_short_name: str):
+        data = request.get_json()
+        if not data or 'external_user_id' not in data:
+            return jsonify({"error": "Falta external_user_id"}), 400
 
-    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)
+        external_user_id = data['external_user_id']
+
+        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
+        )
         if not iaut.get("success"):
             return jsonify(iaut), 401
 
+        # 2. Generate a short-lived initiation token.
+        initiation_token = self.jwt_service.generate_chat_jwt(
+            company_id=company.id,
+            company_short_name=company.short_name,
+            external_user_id=external_user_id,
+            expires_delta_seconds=180
+        )
+
+        # 2. Get branding and onboarding data for the shell page
+        branding_data = self.branding_service.get_company_branding(company)
+        onboarding_cards = self.onboarding_service.get_onboarding_cards(company)
+
+        # 4. Generate the URL for the iframe's SRC, now with the secure token.
+        target_url = url_for('external_login',
+                             company_short_name=company_short_name,
+                             init_token=initiation_token,
+                             _external=True)
+
+        # 5. Render the shell.
+        return render_template("onboarding_shell.html",
+                               iframe_src_url=target_url,
+                               branding=branding_data,
+                               onboarding_cards=onboarding_cards
+                               )
+
+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):
+        # 1. Validate the initiation token from the URL
+        init_token = request.args.get('init_token')
+        if not init_token:
+            return "Falta el token de iniciación.", 401
+
+        # Reutilizamos el validador de JWT, ya que el token tiene la misma estructura
+        payload = self.jwt_service.validate_chat_jwt(init_token, company_short_name)
+        if not payload:
+            return "Token de iniciación inválido o expirado.", 401
+
+        # 2. Extract user ID securely from the validated token
+        external_user_id = payload.get('external_user_id')
+        if not external_user_id:
+            return "Token con formato incorrecto.", 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:
-            # initialize the context
+            # 3. Generate a new long-lived session JWT.
+            token = self.jwt_service.generate_chat_jwt(
+                company_id=company.id,
+                company_short_name=company.short_name,
+                external_user_id=external_user_id,
+                expires_delta_seconds=3600 * 8  # 8 horas
+            )
+            if not token:
+                raise Exception("No se pudo generar el token de sesión (JWT).")
+
+            # 4. Init the company/user LLM context.
             self.query_service.llm_init_context(
                 company_short_name=company_short_name,
                 external_user_id=external_user_id
             )
 
-            return {'status': 'OK'}, 200
+            # 5. get the prompt list from backend
+            prompts = self.prompt_service.get_user_prompts(company_short_name)
+
+            # 6. get the branding data
+            branding_data = self.branding_service.get_company_branding(company)
+
+            # 7. render the chat page with the company/user information.
+            return render_template("chat.html",
+                                        company_short_name=company_short_name,
+                                        auth_method='jwt',
+                                        session_jwt=token,
+                                        external_user_id=external_user_id,
+                                        branding=branding_data,
+                                        prompts=prompts,
+                                        iatoolkit_base_url=os.getenv('IATOOLKIT_BASE_URL'),
+                                        ), 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
+            logging.exception(f"Error al inicializar el chat para {company_short_name}/{external_user_id}: {e}")
+            return jsonify({"error": "Error interno al iniciar el chat"}), 500

iatoolkit/views/login_view.py

@@ -6,14 +6,13 @@
 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
+from iatoolkit.services.onboarding_service import OnboardingService
 
 class InitiateLoginView(MethodView):
     """
@@ -22,8 +21,13 @@ class InitiateLoginView(MethodView):
     and immediately returns the loading shell page.
     """
     @inject
-    def __init__(self, profile_service: ProfileService):
+    def __init__(self,
+                 profile_service: ProfileService,
+                 branding_service: BrandingService,
+                 onboarding_service: OnboardingService):
         self.profile_service = profile_service
+        self.branding_service = branding_service
+        self.onboarding_service = onboarding_service
 
     def post(self, company_short_name: str):
         # get company info
@@ -35,7 +39,7 @@ class InitiateLoginView(MethodView):
         email = request.form.get('email')
         password = request.form.get('password')
 
-        # authenticate the user
+        # 1. authenticate the user
         response = self.profile_service.login(
             company_short_name=company_short_name,
             email=email,
@@ -53,13 +57,22 @@ class InitiateLoginView(MethodView):
                 },
                 alert_message=response["error"]), 400
 
+        # 2. Get branding and onboarding data for the shell page
+        branding_data = self.branding_service.get_company_branding(company)
+        onboarding_cards = self.onboarding_service.get_onboarding_cards(company)
+
+        target_url = url_for('login',
+                        company_short_name=company_short_name,
+                        _external=True)
 
         # 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_user_id='' # Pass an empty string, the shell will handle it
+            "onboarding_shell.html",
+            iframe_src_url=target_url,
+            external_user_id='',
+            branding=branding_data,
+            onboarding_cards=onboarding_cards
         )
 
 
@@ -76,49 +89,48 @@ class LoginView(MethodView):
         self.branding_service = branding_service
 
     def get(self, company_short_name: str):
-        # get company info
+        """
+        Handles the heavy-lifting part of the login, triggered by the iframe.
+        The user is already authenticated via the session cookie.
+        """
+        # 1. Retrieve user and company info from the session.
+        user_id = SessionManager.get('user_id')
+        if not user_id:
+            # This can happen if the session expires or is invalid.
+            # Redirecting to home is a safe fallback.
+            return redirect(url_for('home', company_short_name=company_short_name))
+
+        user_email = SessionManager.get('user')['email']
         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
 
-        return render_template('login.html',
-                               company=company,
-                               company_short_name=company_short_name)
-
-    def post(self, company_short_name: str):
-        company = self.profile_service.get_company_by_short_name(company_short_name)
-
-        # 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')
-        user_email = SessionManager.get('user')['email']
-
         try:
-            # 2. init the company/user LLM context.
+            # 2. Init the company/user LLM context (the long-running task).
             self.query_service.llm_init_context(
                 company_short_name=company_short_name,
                 local_user_id=user_id
             )
 
-            # 3. get the prompt list from backend
+            # 3. Get the prompt list from backend.
             prompts = self.prompt_service.get_user_prompts(company_short_name)
 
-            # 4. get the branding data
+            # 4. Get the branding data.
             branding_data = self.branding_service.get_company_branding(company)
 
+            # 5. Render the final chat page.
             return render_template("chat.html",
-                                    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
+                                   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",
                                    company=company,
                                    company_short_name=company_short_name,
-                                   message="Ha ocurrido un error inesperado."), 500
-
+                                   message="Ha ocurrido un error inesperado."), 500

{iatoolkit-0.12.0.dist-info → iatoolkit-0.16.0.dist-info}/METADATA

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

{iatoolkit-0.12.0.dist-info → iatoolkit-0.16.0.dist-info}/RECORD

@@ -1,12 +1,12 @@
 iatoolkit/__init__.py,sha256=4PWjMJjktixtrxF6BY405qyA50Sv967kEP2x-oil6qk,1120
-iatoolkit/base_company.py,sha256=4usUdOC57Nk1XsawAEyC7mlzb0jROadu64YcYf6poiM,4371
+iatoolkit/base_company.py,sha256=GacYVVujoxAwUhofRn5eZcR-s1bHtSQXLD-0SRnWSC0,4595
 iatoolkit/cli_commands.py,sha256=G5L9xQXZ0lVFXQWBaE_KEZHyfuiT6PL1nTQRoSdnBzc,2302
 iatoolkit/company_registry.py,sha256=tduqt3oV8iDX_IB1eA7KIgvIxE4edTcy-3qZIXh3Lzw,2549
-iatoolkit/iatoolkit.py,sha256=CCYOpsP7fcpHInrvpSUh6tlZMv1kH8lCmiwi0_aEokU,16723
+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=x877TTYj_zfcoVkOXKDRUpTvAehM6JqBguRzgu0LTkM,5093
+iatoolkit/common/routes.py,sha256=c3T4YlkYHSLD2OifSI7pfIKy4rgz9xgo5ECcGHbAHNA,4901
 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
@@ -30,13 +30,13 @@ iatoolkit/repositories/__init__.py,sha256=5JqK9sZ6jBuK83zDQokUhxQ0wuJJJ9DXB8pYCL
 iatoolkit/repositories/database_manager.py,sha256=UaU7k3s7IRuXhCHTy9GoCeP9K1ad0LBdj_n1a_QjGS0,3108
 iatoolkit/repositories/document_repo.py,sha256=Y7bF1kZB1HWJsAGjWdF7P2aVYeTYNufq9ngQXp7mDkY,1124
 iatoolkit/repositories/llm_query_repo.py,sha256=YT_t7cYGQk8rwzH_17-28aTzO-e2jUfa2rvXy8tugvA,3612
-iatoolkit/repositories/models.py,sha256=vlSC-Z1cbho5dLrvoMnEroafO7yC3JXsZKn7SqoIXKU,13158
-iatoolkit/repositories/profile_repo.py,sha256=X2tnqAdaae-SA9sQq-gBqopVdf77I_Q9J_Aa38P0tjY,4060
+iatoolkit/repositories/models.py,sha256=DVmFYncVF-JKiE7SZjN8SmsDOsEx1ADlF1PpVnnfASM,13209
+iatoolkit/repositories/profile_repo.py,sha256=vwDuVec9IDUCSyM7ecf790oqdpV36cpM2bgqP8POkHQ,4128
 iatoolkit/repositories/tasks_repo.py,sha256=icVO_r2oPagGnnBhwVFzznnvEEU2EAx-2dlWuWvoDC4,1745
 iatoolkit/repositories/vs_repo.py,sha256=UkpmQQiocgM5IwRBmmWhw3HHzHP6zK1nN3J3TcQgjhc,5300
 iatoolkit/services/__init__.py,sha256=5JqK9sZ6jBuK83zDQokUhxQ0wuJJJ9DXB8pYCLkX7X4,102
 iatoolkit/services/benchmark_service.py,sha256=CdbFYyS3FHFhNzWQEa9ZNjUlmON10DT1nKNbZQ1EUi8,5880
-iatoolkit/services/branding_service.py,sha256=iYjlUy-5X-x2F6Yg9C70FTMjuItBaUYVHcrXgiKuDbA,4976
+iatoolkit/services/branding_service.py,sha256=gjGKMkCfW9kQJ3Zwzf4XNhrOinfjSo43uD9iG07QlFg,6981
 iatoolkit/services/dispatcher_service.py,sha256=ykR1ye6McyCCuaBgwH6r3-PqcLAr4v4ApkPazMSBzbs,14040
 iatoolkit/services/document_service.py,sha256=nMXrNtbHQuc9pSaten0LvKY0kT8_WngBDmZJUP3jNPw,5936
 iatoolkit/services/excel_service.py,sha256=CJGhu7cQl9J6y_ZWSJ-M63Xm-RXR9Zs66oOR2NJErZQ,3868
@@ -45,6 +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/onboarding_service.py,sha256=cMO2Ho1-G3wAeVNl-j25LwCMJjRwj3yKHpYKnZUFLDE,2001
 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
@@ -66,7 +67,7 @@ iatoolkit/static/js/chat_feedback.js,sha256=_izl49hFEUZYREcJoaPukpTs0YjDgJYUu-Qf
 iatoolkit/static/js/chat_filepond.js,sha256=mzXafm7a506EpM37KATTK3zvAswO1E0KSUY1vKbwuRc,3163
 iatoolkit/static/js/chat_history.js,sha256=G01rKSXOpLpIavycGPbfpfYg5vmPrLhkHYbCLhY3_zs,3964
 iatoolkit/static/js/chat_main.js,sha256=BLKN5a3xP2uu0z6iF6qmgIpT1vClaZf3apcAVQBhU0g,16229
-iatoolkit/static/styles/chat_iatoolkit.css,sha256=z1sYkvredKjU4Ag75ZB8CzsRGHi3t69vgxVW6rjSBMU,10073
+iatoolkit/static/styles/chat_iatoolkit.css,sha256=TbdviimueZjOKVH5FFSjfIqJvLJJt1XGFRwUJgsIP5M,10887
 iatoolkit/static/styles/chat_info.css,sha256=17DbgoNYE21VYWfb5L9-QLCpD2R1idK4imKRLwXtJLY,1058
 iatoolkit/static/styles/chat_modal.css,sha256=pE7JY5D63Ds_d2FKdmxym4sevvg-2Mf7yo-gB7KA9vE,3730
 iatoolkit/static/styles/llm_output.css,sha256=AlxgRSOleeCk2dLAqFWVaQ-jwZiJjcpC5rHuUv3T6VU,2312
@@ -74,36 +75,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=YZmgq8G03wG1Kjq-v3FBZoIeJAI8XFE_2OI6gg-M5RU,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=ev_IvNlLStwF8mrpP2MKUmaQuASK4Z_gB4OorFrCQTw,8232
+iatoolkit/templates/home.html,sha256=Bq4wSWrd9E2VRFQ4Br_nqbhipYxZTJfRMiD0hjaSNsM,7942
 iatoolkit/templates/login.html,sha256=r4hy7MsQkfDqi6pBRNkkRiFr3GPSoHCT89R5lQLUWZc,1991
+iatoolkit/templates/onboarding_shell.html,sha256=iIYEe0O3RQ5FgjPigGh8SfC3bFYcJjtd2HeoibyDFLM,7463
 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/download_file_view.py,sha256=1gZ0ipqeCn39sTrJFo1-tlewlcSF7s_YNTvE4qd0HOw,2010
-iatoolkit/views/external_chat_login_view.py,sha256=5MwWu_acP5Lra6AHG_xUxoyo7Gb17irDaTrK1i9U6-M,4660
-iatoolkit/views/external_login_view.py,sha256=EHxN2omqTDkWyMp0FVCNDFt-JJ2hpg8LvprKanGR1gY,1355
+iatoolkit/views/external_login_view.py,sha256=IMfMpL_5xw3m6q9eK7lZ4PhwcL2zW7s7lNIK_HEl78c,6551
 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=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=Ip4d_3ie9T8soWhznahGbCcARAJfsoFcyvfR0N4XegQ,4921
+iatoolkit/views/login_view.py,sha256=tSJiSTT7gKHediAAD2IGdVTUfkTNI1aWMt-d9J-S4lk,5562
 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.12.0.dist-info/METADATA,sha256=d0z93e4UDV3PgKL1_MYfy4PC__GZo6G_Fy8oSmYctj8,9301
-iatoolkit-0.12.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-iatoolkit-0.12.0.dist-info/top_level.txt,sha256=V_w4QvDx0b1RXiy8zTCrD1Bp7AZkFe3_O0-9fMiwogg,10
-iatoolkit-0.12.0.dist-info/RECORD,,
+iatoolkit-0.16.0.dist-info/METADATA,sha256=G6vbKnCI6sqku8SXqzZhzHqSZyBWQnaCM-72AQILHvw,9301
+iatoolkit-0.16.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+iatoolkit-0.16.0.dist-info/top_level.txt,sha256=V_w4QvDx0b1RXiy8zTCrD1Bp7AZkFe3_O0-9fMiwogg,10
+iatoolkit-0.16.0.dist-info/RECORD,,

iatoolkit/views/external_chat_login_view.py

@@ -1,112 +0,0 @@
-# Copyright (c) 2024 Fernando Libedinsky
-# Product: IAToolkit
-#
-# IAToolkit is open source software.
-
-import os
-import logging
-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
-from iatoolkit.services.profile_service import ProfileService
-from iatoolkit.services.query_service import QueryService
-from iatoolkit.services.prompt_manager_service import PromptService
-from iatoolkit.services.jwt_service import JWTService
-from iatoolkit.services.branding_service import BrandingService
-
-class InitiateExternalChatView(MethodView):
-    @inject
-    def __init__(self,
-                 iauthentication: IAuthentication,
-                 ):
-        self.iauthentication = iauthentication
-
-    def post(self, company_short_name: str):
-        data = request.get_json()
-        if not data or 'external_user_id' not in data:
-            return jsonify({"error": "Falta external_user_id"}), 400
-
-        external_user_id = data['external_user_id']
-
-        # 1. verify access credentials quickly
-        iaut = self.iauthentication.verify(
-            company_short_name,
-            body_external_user_id=external_user_id
-        )
-        if not iaut.get("success"):
-            return jsonify(iaut), 401
-
-        # 2. Render the shell page, passing the final data URL and user id
-        return render_template("login_shell.html",
-                               data_source_url=url_for('external_chat_login', company_short_name=company_short_name),
-                               external_user_id=external_user_id
-                               )
-
-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 post(self, company_short_name: str):
-        data = request.get_json()
-        if not data or 'external_user_id' not in data:
-            return jsonify({"error": "Falta external_user_id"}), 400
-
-        external_user_id = data['external_user_id']
-
-        company = self.profile_service.get_company_by_short_name(company_short_name)
-        if not company:
-            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,
-                company_short_name=company.short_name,
-                external_user_id=external_user_id,
-                expires_delta_seconds=3600 * 8  # 8 horas
-            )
-            if not token:
-                raise Exception("No se pudo generar el token de sesión (JWT).")
-
-            # 2. init the company/user LLM context.
-            self.query_service.llm_init_context(
-                company_short_name=company_short_name,
-                external_user_id=external_user_id
-            )
-
-            # 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)
-
-            # 5. render the chat page with the company/user information.
-            chat_html = render_template("chat.html",
-                                        company_short_name=company_short_name,
-                                        auth_method='jwt',  # login method is JWT
-                                        session_jwt=token,  # pass the token to the front-end
-                                        external_user_id=external_user_id,
-                                        branding=branding_data,
-                                        prompts=prompts,
-                                        iatoolkit_base_url=os.getenv('IATOOLKIT_BASE_URL'),
-                                        )
-            return chat_html, 200
-
-        except Exception as e:
-            logging.exception(f"Error al inicializar el chat para {company_short_name}/{external_user_id}: {e}")
-            return jsonify({"error": "Error interno al iniciar el chat"}), 500