iatoolkit 0.66.2__py3-none-any.whl → 0.71.2__py3-none-any.whl

iatoolkit/static/js/chat_onboarding_button.js

@@ -28,6 +28,7 @@
     const elIcon = qs(root, ui.icon);
     const elTitle = qs(root, ui.title);
     const elText = qs(root, ui.text);
+    const elExample = qs(root, ui.example);
     const elDots = qs(root, ui.dots);
     const elPrev = qs(root, ui.prev);
     const elNext = qs(root, ui.next);
@@ -43,6 +44,11 @@
       if (elIcon) elIcon.innerHTML = `<i class="${c.icon || 'bi bi-lightbulb'}"></i>`;
       if (elTitle) elTitle.textContent = c.title || '';
       if (elText) elText.innerHTML = c.text || '';
+      if (elExample && c.example) {
+          elExample.innerHTML = ('Ejemplo: ' + c.example) || '';
+      }
+      else
+          elExample.innerHTML = '';
       if (elDots) createDots(elDots, cards.length, idx);
     }
 

iatoolkit/static/styles/chat_iatoolkit.css

@@ -23,16 +23,33 @@ li {
   padding-left: 0.25rem; /* Pequeño espacio extra entre la viñeta y el texto */
 }
 
-/* Estilo para la viñeta (el punto) de cada elemento de la lista */
-li::marker {
-  color: var(--brand-primary-color); /* La viñeta usa el color principal de la marca */
-  font-size: 1.1em; /* Ligeramente más grande para que sea más visible */
-}
-
 .text-muted {
   color: #64748b;
 }
 
+/* Estilos para botones con branding */
+.btn-branded-primary {
+    background-color: var(--brand-primary-color);
+    border-color: var(--brand-primary-color);
+    color: var(--brand-text-on-primary);
+}
+.btn-branded-primary:hover {
+    background-color: var(--brand-text-on-primary);
+    color: var(--brand-primary-color);
+    border-color: var(--brand-primary-color);
+}
+
+.btn-branded-secondary {
+    background-color: var(--brand-secondary-color);
+    border-color: var(--brand-secondary-color);
+    color: var(--brand-text-on-secondary);
+}
+.btn-branded-secondary:hover {
+    background-color: var(--brand-text-on-secondary);
+    color: var(--brand-secondary-color);
+    border-color: var(--brand-secondary-color);
+}
+
 .chat-layout-container {
     display: flex;
     flex-direction: column;
@@ -42,21 +59,6 @@ li::marker {
     padding-bottom: 0.2rem;
 }
 
-/* Esta parte para el prompt-assistant sigue siendo correcta */
-main.chat-layout-container.prompt-assistant-open {
-    height: auto;
-    overflow: visible;
-}
-
-
-/* CSS para el chat principal    */
-/* esta clase defines los atributos de cada bloque */
-.chat-block {
-    padding: 1rem;
-    border: 1px solid #dee2e6;
-    border-radius: 0.375rem; /* Borde redondeado estándar de Bootstrap */
-    box-shadow: 0 2px 8px rgba(0,0,0,0.06); /* Sombra suave y unificada */
-}
 
 /* sección del encabezado en el chat principal */
 .company-section {
@@ -74,9 +76,24 @@ main.chat-layout-container.prompt-assistant-open {
     background-color: #fff; /* Fondo blanco para el contenedor del chat */
 }
 
-/* La caja principal que envuelve toda el área de entrada */
-.input-area {
-    background-color: var(--brand-prompt-assistant-bg, #f8f9fa);
+/* CSS para el chat principal    */
+/* esta clase defines los atributos de cada bloque */
+.chat-block {
+    padding: 1rem;
+    border: 1px solid #dee2e6;
+    border-radius: 0.375rem; /* Borde redondeado estándar de Bootstrap */
+    box-shadow: 0 2px 8px rgba(0,0,0,0.06); /* Sombra suave y unificada */
+}
+
+/* El textarea "invisible" en el centro */
+.chat-textarea {
+    flex-grow: 1;
+    background: transparent !important;
+    border: none !important;
+    box-shadow: none !important;
+    resize: none;
+    max-height: 150px;
+    min-height: 38px;
 }
 
 /* La barra "cápsula" que envuelve el texto y los iconos */
@@ -93,6 +110,11 @@ main.chat-layout-container.prompt-assistant-open {
     border-color: #86b7fe;
     box-shadow: 0 0 0 0.25rem rgba(13, 110, 253, 0.25);
 }
+/* La caja principal que envuelve toda el área de entrada */
+.input-area {
+    background-color: var(--brand-prompt-assistant-bg, #f8f9fa);
+}
+
 
 #question {
     width: 100%;
@@ -109,12 +131,6 @@ main.chat-layout-container.prompt-assistant-open {
     box-shadow: 0 0 0 0.2rem rgba(0,123,255,.25);
 }
 
-.answer-section, .error-section {
-    max-width: 100%;
-    word-wrap: break-word;
-    margin-bottom: 10px;
-}
-
 /* Estilo del mensaje ingresado por el usuario */
 #chat-container .message {
   max-width: 75%;
@@ -134,6 +150,13 @@ main.chat-layout-container.prompt-assistant-open {
   background-color: #f7f7f8;
 }
 
+.answer-section, .error-section {
+    max-width: 100%;
+    word-wrap: break-word;
+    margin-bottom: 10px;
+}
+
+
 .answer-section {
     align-self: flex-start;
     max-width: 75%;
@@ -184,7 +207,6 @@ main.chat-layout-container.prompt-assistant-open {
     opacity: 0.8;
 }
 
-
 /* Spinner */
 .spinning {
     animation: spin 1s linear infinite;
@@ -195,7 +217,6 @@ main.chat-layout-container.prompt-assistant-open {
     color: var(--brand-primary-color);
 }
 
-
 @keyframes spin {
     0% { transform: rotate(0deg); }
     100% { transform: rotate(360deg); }
@@ -236,108 +257,6 @@ main.chat-layout-container.prompt-assistant-open {
 }
 
 
-
-/* Estos estilos son para las paginas publicas: home, signup,  olvide la clave, etc */
-
-    /* --- Encabezado (top) de company --- */
-    .custom-company-header {
-        height: 78px;
-        margin-bottom: 20px;
-        padding: 0 1.5rem; /* Padding interno para que el texto no toque los bordes */
-        border-radius: 0.375rem; /* Bordes redondeados para que coincida con los formularios */
-
-        /* Los estilos de color y sombra se mantienen */
-        background-color: var(--brand-header-bg);
-        color: var(--brand-header-text);
-        box-shadow: 0 2px 8px rgba(0, 0, 0, 0.08); /* Sombra ligeramente más suave */
-    }
-
-    /* Estilo específico para el nombre de la marca en el encabezado (Letra más grande) */
-    .custom-company-header .brand-name {
-        font-size: 1.8rem; /* Tamaño de fuente aumentado */
-        font-weight: 600;
-        color: inherit;
-        text-decoration: none;
-        transition: opacity 0.2s ease-in-out;
-    }
-
-    /* Efecto hover para el enlace del nombre de la marca */
-    .custom-company-header .brand-name:hover {
-        opacity: 0.85;
-    }
-
-    /* Estilo para el texto "Powered by" --- */
-    .custom-company-header .powered-by {
-        font-size: 0.9rem;
-        font-weight: 400;
-        opacity: 0.75; /* Menos prominente que el nombre de la marca */
-        color: inherit;
-    }
-
-    /* --- Estilo para el enlace de IAToolkit --- */
-    .custom-company-header .iatoolkit-link {
-        color: inherit; /* Hereda el color del texto padre */
-        text-decoration: none; /* ¡Elimina el subrayado! */
-        font-weight: 600; /* Un poco más de peso para diferenciarlo */
-        transition: opacity 0.2s ease-in-out;
-    }
-
-    .custom-company-header .form-container {
-        display: flex;
-        flex-direction: column;
-        gap: 1rem;
-    }
-
-    /* titulo en home */
-    .home-title {
-        font-size: 3.2rem;
-        font-weight: 800;
-        line-height: 1.2;
-        color: var(--brand-primary-color);
-    }
-    .list-example {
-            display: block;              /* Coloca el ejemplo en su propia línea */
-            font-size: 0.9rem;          /* Lo hace ligeramente más pequeño */
-            color: #6c757d;             /* Color gris (muted) para texto secundario */
-            font-style: italic;         /* Cursiva para diferenciarlo */
-            padding-left: 1.75rem;      /* Lo indenta para alinearlo con el texto del tema */
-            margin-top: 0.25rem;        /* Pequeño espacio superior */
-        }
-
-    .text-brand-primary {
-        font-weight: 800;
-        color: var(--brand-primary-color);
-        }
-
-/* contenedor de formularios: login, signup, forgot password, etc. */
-    .branded-form-container {
-        background-color: #ffffff;
-        border: 1px solid #dee2e6;
-        border-top: 4px solid #adb5bd;
-        border-radius: 0.375rem;
-        padding: 2rem;
-        box-shadow: 0 4px 12px rgba(0,0,0,0.08);
-    }
-
-    .branded-form-label {
-        font-size: 0.85rem;
-        font-weight: 600;
-        color: #495057; /* Un gris oscuro profesional */
-        text-transform: uppercase;
-        letter-spacing: 0.05em; /* Un poco de espacio extra entre letras */
-        margin-bottom: 0.3rem;
-        display: block;
-    }
-
-    .branded-form-title {
-        color: var(--brand-primary-color);
-        font-size: 1.75rem; /* Un tamaño más prominente */
-        font-weight: 700;   /* Equivalente a fw-bold de Bootstrap */
-        text-align: center;
-        margin-bottom: 1.5rem; /* Espacio consistente debajo del título */
-    }
-
-
 /* Aplica el color secundario de la marca al icono de ver archivos */
 #view-files-button i {
     color: var(--brand-secondary-color, #6c757d);
@@ -353,16 +272,6 @@ main.chat-layout-container.prompt-assistant-open {
     margin-bottom: 12px !important; /* Anula el mb-2 (8px) para añadir un poco más de espacio */
 }
 
-/* 4. El textarea "invisible" en el centro */
-.chat-textarea {
-    flex-grow: 1;
-    background: transparent !important;
-    border: none !important;
-    box-shadow: none !important;
-    resize: none;
-    max-height: 150px;
-    min-height: 38px;
-}
 
 /* Estilo UNIFICADO para los enlaces de iconos de acción */
 .action-icon-style {
@@ -394,7 +303,7 @@ main.chat-layout-container.prompt-assistant-open {
     color: #343a40;
 }
 
-/* 6. Anulación específica para el botón de ENVIAR usando su ID (Máxima Prioridad) */
+/* Anulación específica para el botón de ENVIAR usando su ID (Máxima Prioridad) */
 #send-button i {
     font-size: 1.7rem; /* Ligeramente más grande */
 }
@@ -474,30 +383,30 @@ main.chat-layout-container.prompt-assistant-open {
     border-bottom: none;
 }
 
+/* estilos para el selector de prompts */
 .prompt-select-button {
-    /* Estado por defecto: Estilo "outline" sutil */
-    background-color: var(--brand-primary-color);
-    color: var(--brand-text-on-primary);
-    border: 1px solid var(--brand-primary-color); /* Borde fino con el color primario */
-
-    /* Estilos de botón estándar */
-    border-radius: 0.25rem;
+    color: #202123;
+    border: 2px solid var(--brand-primary-color);
+    border-radius: 0.25rem;     /* Estilos de botón estándar */
     padding: 0.5rem 1rem;
     cursor: pointer;
     text-align: center;
-    font-weight: 500; /* Un poco más de peso para el texto */
+    font-weight: 500;
 
     /* Transición suave para el cambio en hover */
     transition: background-color 0.2s ease-in-out, color 0.2s ease-in-out;
 }
 
 .prompt-select-button:hover {
-    /* Estado hover: Se rellena para dar feedback claro */
-    background-color: transparent;
-    color: var(--brand-primary-color);
-}
+    background-color: var(--brand-primary-color);
+    color: var(--brand-text-on-primary);
 
+    /* Marca el borde con un glow limpio */
+    border-color: var(--brand-primary-color);
+    box-shadow: 0 0 0 3px rgba( var(--brand-primary-rgb), 0.25 );
+}
 
+/* estilo de la x que limpia el prompt seleccionado */
 #clear-selection-button {
     position: absolute;
     top: 50%;
@@ -510,7 +419,9 @@ main.chat-layout-container.prompt-assistant-open {
 }
 
 #clear-selection-button:hover {
-    color: #212529;
+    color: #ffffff;
+    background-color: #6c757d;
+    border-radius: 0.25rem;
 }
 
 #send-button i {

iatoolkit/static/styles/chat_modal.css

@@ -39,28 +39,6 @@
     filter: invert(1) grayscale(100%) brightness(200%); /* Hace el botón de cerrar visible en fondos oscuros */
 }
 
-/* Estilos para botones con branding */
-.btn-branded-primary {
-    background-color: var(--brand-primary-color);
-    border-color: var(--brand-primary-color);
-    color: var(--brand-text-on-primary);
-}
-.btn-branded-primary:hover {
-    background-color: var(--brand-text-on-primary);
-    color: var(--brand-primary-color);
-    border-color: var(--brand-primary-color);
-}
-
-.btn-branded-secondary {
-    background-color: var(--brand-secondary-color);
-    border-color: var(--brand-secondary-color);
-    color: var(--brand-text-on-secondary);
-}
-.btn-branded-secondary:hover {
-    background-color: var(--brand-text-on-secondary);
-    color: var(--brand-secondary-color);
-    border-color: var(--brand-secondary-color);
-}
 
 /* Tabla con el resultado del historial */
 .thead-branded th {
@@ -109,7 +87,7 @@
         box-shadow: none;
     }
 
-    /* 2. Estilo para el anillo de foco, aplicado ÚNICAMENTE cuando el botón está CERRADO y tiene foco. */
+    /* Estilo para el anillo de foco, aplicado ÚNICAMENTE cuando el botón está CERRADO y tiene foco. */
     .modal-body .accordion-button.collapsed:focus {
         background-color: #fff; /* Asegura fondo blanco al hacer foco en un item cerrado */
         border-color: transparent; /* Evita bordes no deseados */
@@ -153,17 +131,3 @@
 .remove-file-btn {
     flex-shrink: 0;
 }
-
-/* Estilos para alertas generales del chat */
-.alert-branded-info {
-    background-color: var(--brand-info-bg);
-    color: var(--brand-info-text);
-    border-color: var(--brand-info-border);
-}
-.alert-branded-info strong,
-.alert-branded-info .alert-link {
-    color: inherit;
-}
-.alert-branded-info .bi { /* Asegura que los iconos también tomen el color */
-    color: inherit;
-}

iatoolkit/static/styles/onboarding.css

@@ -36,6 +36,13 @@
   min-height: 60px;
 }
 
+.ob-example {
+  font-size: 0.95rem;
+  color: #444;
+  line-height: 1.5;
+  min-height: 60px;
+}
+
 /* Navegación */
 .ob-nav {
   display: flex;

iatoolkit/system_prompts/query_main.prompt

@@ -19,19 +19,11 @@ Eres un asistente que responde preguntas o ejecuta tareas según el contexto de
 Eres un asistente que responde preguntas sobre empresas y sus clientes.
 
 **Reglas obligatorias de contexto:**
-1. Cada vez que el usuario consulte por un cliente,
-debes memorizarlo y usarlo como cliente de contexto.
-2. Si el usuario hace una pregunta **sin especificar un cliente**,
-siempre debes asumir que la pregunta se refiere al **último cliente identificado** en la conversación.
-3. Nunca cambies de cliente de contexto a menos que el usuario especifique uno nuevo.
-4. Si el usuario pregunta por un cliente que no está en tus registros, responde indicando que no tienes información, pero **no borres el contexto anterior**.
-5. No respondas con “no se encontró información del cliente” salvo que nunca se haya identificado ningún cliente antes en la conversación.
-6. No debes incluir explicaciones, comentarios o texto adicional.
+En caso que te hagan preguntas especificas sobre un cliente,  debes asumir que la pregunta se
+refiere al **último cliente identificado** en la conversación.
 
 **IMPORTANTE:**
 
-No respondas nunca sobre un cliente anterior si ya se identificó uno nuevo, y nunca pierdas el contexto salvo que el usuario lo cambie explícitamente.
-
 ### **Instrucciones**
 1. Devuelve siempre la respuesta en formato JSON.
 2. Usa la información de contexto para responder la consulta del usuario de forma clara y concisa.

iatoolkit/templates/change_password.html

@@ -1,6 +1,6 @@
 {% extends "base.html" %}
 
-{% block title %}{{ t('ui.change_password.title') }} - {{ company.name }}{% endblock %}
+{% block title %}{{ t('ui.change_password.title') }} - {{ branding.name }} {% endblock %}
 
 {% block styles %}
     <style>

iatoolkit/templates/chat.html

@@ -90,8 +90,8 @@
     </div>
 
 <div id="chat-container"
-     class="chat-block mt-2 flex-grow-1"
-     style="overflow-y: auto;">
+     class="chat-block mt-2 flex-grow-1" style="overflow-y: auto;">
+
     <!-- Mensaje de bienvenida  -->
     <div class="answer-section">
          {{ t('ui.chat.welcome_message') }}
@@ -113,7 +113,10 @@
                 <div class="{{ prompt_col_class }}">
                     <div class="position-relative h-100">
                         <div class="input-group dropup h-100">
-                            <button type="button" id="prompt-select-button" class="btn prompt-select-button  border dropdown-toggle w-100 text-start" data-bs-toggle="dropdown" aria-expanded="false">
+                            <button type="button" id="prompt-select-button"
+                                    class="btn prompt-select-button  border dropdown-toggle w-100 text-start"
+                                    data-bs-toggle="dropdown"
+                                    aria-expanded="false">
                                 {{ t('ui.chat.prompts_available') }} ....
                             </button>
                             <ul class="dropdown-menu dropdown-menu-soft w-100">
@@ -137,7 +140,11 @@
                                 {% endif %}
                             </ul>
                         </div>
-                        <button type="button" id="clear-selection-button" class="btn clear-specific-data-button" style="display: none;"><i class="bi bi-x-circle-fill"></i></button>
+                        <button type="button"
+                                id="clear-selection-button"
+                                class="btn clear-specific-data-button position-absolute end-0 me-1" style="display: none;">
+                            <i class="bi bi-x-circle-fill"></i>
+                        </button>
                     </div>
                     <input type="hidden" id="prompt-select-value" name="prompt_select_value">
                     <input type="hidden" id="prompt-select-description" name="prompt-select-description">
@@ -265,6 +272,7 @@ document.addEventListener('DOMContentLoaded', function () {
       icon: '#ob-icon',
       title: '#ob-title',
       text: '#ob-text',
+        example: '#ob-example',
       dots: '#ob-dots',
       prev: '#ob-prev',
       next: '#ob-next'

iatoolkit/templates/chat_modals.html

@@ -34,6 +34,9 @@
                             </tbody>
                         </table>
                     </div>
+                    <div id="no-history-message" style="display: none;">
+                        {{ t('js_messages.no_history_found') }}
+                    </div>
                 </div>
             </div>
             <div class="modal-footer">
@@ -111,6 +114,7 @@
           <div id="ob-icon" class="ob-icon"><i class="bi bi-lightbulb"></i></div>
           <h6 id="ob-title" class="ob-title"></h6>
           <div id="ob-text" class="ob-text"></div>
+            <div id="ob-example" class="ob-example"></div>
           <div class="ob-nav">
             <button id="ob-prev" class="ob-btn" aria-label="Anterior"><i class="bi bi-chevron-left"></i></button>
             <div id="ob-dots" class="ob-dots"></div>

iatoolkit/templates/error.html

@@ -1,6 +1,6 @@
 {% extends "base.html" %}
 
-{% block title %}Error {% endblock %}
+{% block title %}Error{% endblock %}
 
 {% block styles %}
 {# Carga las variables de CSS personalizadas de la empresa #}

iatoolkit/templates/login_simulation.html

@@ -1,15 +1,25 @@
 {% extends "base.html" %}
 
-{% block title %}Prueba de Login para {{ company_short_name }}{% endblock %}
+{% block title %}Login Test - {{ branding.name }}{% endblock %}
+{% block styles %}
+    <style>
+        {{ branding.css_variables | safe }}
+    </style>
+    <link rel="stylesheet" href="{{ url_for('static', filename='styles/chat_public.css') }}">
+{% endblock %}
 
 {% block content %}
+<div class="container mt-4">
+
+{% include '_company_header.html' %}
+
 <div class="container-fluid">
   <div class="row flex-fill mt-5 justify-content-center">
     <div class="col-12 col-lg-6">
-      <div class="border rounded p-4 p-md-5 shadow-sm bg-light">
-        <h3 class="text-muted fw-semibold text-start mb-3">
-          Login Externo para <span style="color:#0d6efd;">{{ company_short_name }}</span>
-        </h3>
+      <div class="branded-form-container">
+        <h4 class="branded-form-title">
+          Login Externo para {{ company_short_name }}
+        </h4>
         <div class="text-center mb-4">
           <p class="text-muted widget-intro-text">
             Este formulario simula el inicio de una sesión externa. Al enviar, serás redirigido a la URL de login final.
@@ -22,7 +32,7 @@
             <label for="external_user_id" class="form-label d-block">External user ID</label>
             <input type="text" id="external_user_id" name="external_user_id" class="form-control" required>
           </div>
-          <button type="submit" class="btn btn-primary">
+          <button type="submit" class="btn btn-branded-primary">
             Redirigir a External Login
           </button>
         </form>
@@ -30,5 +40,6 @@
     </div>
   </div>
 </div>
+</div>
 {% endblock %}
 

iatoolkit/templates/onboarding_shell.html

@@ -1,6 +1,6 @@
 {% extends "base.html" %}
 
-{% block title %}Iniciando {{ branding.name | default('IAToolkit') }} IA...{% endblock %}
+{% block title %}Iniciando {{ branding.name }} IA...{% endblock %}
 
 {% block styles %}
 {% endblock %}
@@ -45,6 +45,8 @@
         <div id="card-icon" class="ob-icon"><i class="fas fa-lightbulb"></i></div>
         <h3 id="card-title" class="ob-title">Título de la Tarjeta</h3>
         <p id="card-text" class="ob-text">Descripción de la tarjeta de capacitación.</p>
+        <p id="card-example" class="ob-example"></p>
+
         <div class="ob-nav">
           <button id="prev-card" class="ob-btn btn-branded-primary" aria-label="Anterior">
             <i class="fas fa-chevron-left"></i>
@@ -88,6 +90,7 @@
         icon: '#card-icon',
         title: '#card-title',
         text: '#card-text',
+          example: '#card-example',
         dots: '#progress-dots',
         prev: '#prev-card',
         next: '#next-card',

iatoolkit/views/base_login_view.py

@@ -11,7 +11,7 @@ from iatoolkit.services.profile_service import ProfileService
 from iatoolkit.services.auth_service import AuthService
 from iatoolkit.services.query_service import QueryService
 from iatoolkit.services.branding_service import BrandingService
-from iatoolkit.services.onboarding_service import OnboardingService
+from iatoolkit.services.configuration_service import ConfigurationService
 from iatoolkit.services.prompt_manager_service import PromptService
 from iatoolkit.services.i18n_service import I18nService
 from iatoolkit.common.util import Utility
@@ -31,7 +31,7 @@ class BaseLoginView(MethodView):
                  jwt_service: JWTService,
                  branding_service: BrandingService,
                  prompt_service: PromptService,
-                 onboarding_service: OnboardingService,
+                 config_service: ConfigurationService,
                  query_service: QueryService,
                  i18n_service: I18nService,
                  utility: Utility
@@ -41,14 +41,14 @@ class BaseLoginView(MethodView):
         self.jwt_service = jwt_service
         self.branding_service = branding_service
         self.prompt_service = prompt_service
-        self.onboarding_service = onboarding_service
+        self.config_service = config_service
         self.query_service = query_service
         self.i18n_service = i18n_service
         self.utility = utility
 
 
     def _handle_login_path(self,
-                           company: Company,
+                           company_short_name: str,
                            user_identifier: str,
                            target_url: str,
                            redeem_token: str = None):
@@ -56,13 +56,12 @@ class BaseLoginView(MethodView):
         Centralized logic to decide between the fast path and the slow path.
         """
         # --- Get the company branding and onboarding_cards
-        branding_data = self.branding_service.get_company_branding(company)
-        onboarding_cards = self.onboarding_service.get_onboarding_cards(company)
-        company_short_name = company.short_name
+        branding_data = self.branding_service.get_company_branding(company_short_name)
+        onboarding_cards = self.config_service.get_configuration(company_short_name, 'onboarding_cards')
 
         # this service decides is the context needs to be rebuilt or not
         prep_result = self.query_service.prepare_context(
-            company_short_name=company.short_name, user_identifier=user_identifier
+            company_short_name=company_short_name, user_identifier=user_identifier
         )
 
         if prep_result.get('rebuild_needed'):

iatoolkit/views/change_password_view.py

@@ -34,7 +34,7 @@ class ChangePasswordView(MethodView):
                 return render_template('error.html',
                                        message=self.i18n_service.t('errors.templates.company_not_found')), 404
 
-            branding_data = self.branding_service.get_company_branding(company)
+            branding_data = self.branding_service.get_company_branding(company_short_name)
 
             try:
                 # Decodificar el token
@@ -66,7 +66,7 @@ class ChangePasswordView(MethodView):
             return render_template('error.html',
                                    message=self.i18n_service.t('errors.templates.company_not_found')), 404
 
-        branding_data = self.branding_service.get_company_branding(company)
+        branding_data = self.branding_service.get_company_branding(company_short_name)
         try:
             # Decodificar el token
             email = self.serializer.loads(token, salt='password-reset', max_age=3600)
@@ -98,7 +98,6 @@ class ChangePasswordView(MethodView):
                     'change_password.html',
                     token=token,
                     company_short_name=company_short_name,
-                    company=company,
                     branding=branding_data,
                     form_data={"temp_code": temp_code,
                                "new_password": new_password,