iatoolkit 0.3.9__py3-none-any.whl → 0.107.4__py3-none-any.whl

iatoolkit/templates/onboarding_shell.html

@@ -0,0 +1,106 @@
+{% extends "base.html" %}
+
+{% block title %}Iniciando {{ branding.name }} IA...{% endblock %}
+
+{% block styles %}
+{% endblock %}
+
+{% block content %}
+
+<link rel="stylesheet" href="{{ url_for('static', filename='styles/llm_output.css', _external=True) }}?v=6">
+<link rel="stylesheet" href="{{ url_for('static', filename='styles/onboarding.css', _external=True) }}?v=6">
+
+<style>
+  {% if branding and branding.css_variables %}
+    {{ branding.css_variables|safe }}
+  {% endif %}
+
+  {# 2. Ahora definimos las reglas que usan esas variables #}
+  .onboarding-shell-root { position: relative; height: 100vh; }
+  .onboarding-shell-root #loader-wrapper { position: absolute; inset: 0; background: #f4f7f6; z-index: 1000; display:flex; align-items:center; justify-content:center; padding:20px; box-sizing:border-box; transition: opacity .5s; }
+  .onboarding-shell-root .ob-stack { width:100%; max-width:520px; display:flex; flex-direction:column; gap:16px; }
+  .onboarding-shell-root .ob-loading-band { display:flex; align-items:center; justify-content:center; gap:12px; padding:12px 14px; background:#fff; border-radius:12px; box-shadow:0 4px 20px rgba(0,0,0,.08); }
+  .onboarding-shell-root .spinner { width:28px; height:28px; border:4px solid rgba(0,0,0,.1); border-top-color: var(--brand-primary-color, #FF5100); border-radius:50%; animation: spin 1s linear infinite; }
+  @keyframes spin { to { transform: rotate(360deg); } }
+  .onboarding-shell-root #content-container { width:100%; height:100%; }
+  .onboarding-shell-root #content-container iframe { width:100%; height:100%; border:none; }
+
+  /* Forzar colores de marca (sube especificidad y evita overrides) */
+  .onboarding-shell-root .ob-brand-header { color: var(--brand-secondary-color, #06326B) !important; }
+  .onboarding-shell-root .ob-brand-header .brand-name { color: var(--brand-primary-color, #FF5100) !important; }
+  .onboarding-shell-root .ob-icon { color: var(--brand-primary-color, #FF5100) !important; }
+  .onboarding-shell-root .ob-btn { background-color: var(--brand-secondary-color, #06326B) !important; color: var(--brand-text-on-secondary, #FFFFFF) !important; border:none !important; }
+  .onboarding-shell-root .ob-dots > div.active { background-color: var(--brand-primary-color, #FF5100) !important; }
+</style>
+
+<div class="onboarding-shell-root ob-root">
+  <div id="loader-wrapper">
+    <div class="ob-stack">
+      <h1 id="ob-brand-header" class="ob-brand-header">
+        <span class="brand-name text-brand-primary">{{ branding.name | default('IAToolkit') }}</span>
+      </h1>
+
+      <div id="card-container" class="ob-card">
+        <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>
+          </button>
+          <div id="progress-dots" class="ob-dots"></div>
+          <button id="next-card" class="ob-btn btn-branded-primary" aria-label="Siguiente">
+            <i class="fas fa-chevron-right"></i>
+          </button>
+        </div>
+      </div>
+
+      <div id="loading-status" class="ob-loading-band">
+        <div class="spinner"></div>
+        <p>{{ t('ui.chat.init_context') }} </p>
+      </div>
+    </div>
+  </div>
+
+  <div id="content-container"></div>
+</div>
+{% endblock %}
+
+{% block scripts %}
+<link rel="stylesheet" href="https://cdnjs.cloudflare.com/ajax/libs/font-awesome/5.15.4/css/all.min.css">
+
+<script src="{{ url_for('static', filename='js/chat_onboarding_button.js', _external=True) }}"></script>
+<script>
+  (function() {
+    const cardsData = {{ onboarding_cards | tojson }};
+    const iframeSrc = "{{ iframe_src_url }}";
+
+    if (!window.initOnboarding) {
+      console.error('initOnboarding no está disponible. Verifica la carga de chat_onboarding.js');
+      return;
+    }
+
+    const onboarding = initOnboarding({
+      mode: 'shell',
+      cards: cardsData,
+      ui: {
+        icon: '#card-icon',
+        title: '#card-title',
+        text: '#card-text',
+          example: '#card-example',
+        dots: '#progress-dots',
+        prev: '#prev-card',
+        next: '#next-card',
+        loader: '#loader-wrapper',
+        container: '#content-container'
+      },
+      autoRotateMs: 5000,
+      shell: { iframeSrc: iframeSrc }
+    });
+
+    onboarding.start();
+  })();
+</script>
+{% endblock %}

iatoolkit/templates/signup.html

@@ -0,0 +1,79 @@
+{% extends "base.html" %}
+
+{% block title %}{{ t('ui.signup.title') }} - {{ branding.name }}{% endblock %}
+
+{% block styles %}
+    {# ¡Importante! Añadimos los estilos para el branding #}
+    <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' %}
+
+    <section class="hero-section">
+        <div class="container">
+            <div class="row justify-content-center">
+                <!-- Se ha reducido el ancho de la columna a lg-6 y md-8 -->
+                <div class="col-lg-6 col-md-8">
+                    <div class="branded-form-container">
+                        <h4 class="branded-form-title">{{ t('ui.signup.title') }}</h4>
+                        <form action="{{ url_for('signup', company_short_name=company_short_name) }}" method="post">
+                            <div class="mb-3">
+                                <label for="email" class="form-label text-secondary">{{ t('ui.signup.email_label') }}</label>
+                                <input type="email" autocomplete="off" id="email" name="email"
+                                       class="form-control" required
+                                       value="{{ form_data.email if form_data else '' }}">
+                            </div>
+
+                            <div class="row">
+                                <div class="col-md-6 mb-3">
+                                    <label for="first_name" class="form-label text-secondary">{{ t('ui.signup.first_name_label') }}</label>
+                                    <input type="text" id="first_name" name="first_name"
+                                           class="form-control" required
+                                           value="{{ form_data.first_name if form_data else '' }}">
+                                </div>
+                                <div class="col-md-6 mb-3">
+                                    <label for="last_name" class="form-label text-secondary">{{ t('ui.signup.last_name_label') }}</label>
+                                    <input type="text" id="last_name" name="last_name"
+                                           class="form-control" required
+                                           value="{{ form_data.last_name if form_data else '' }}">
+                                </div>
+                            </div>
+
+                            <div class="mb-3">
+                                <label for="password" class="form-label text-secondary">{{ t('ui.signup.password_label') }}</label>
+                                <input type="password" id="password" name="password" class="form-control" required>
+                                <div class="d-flex align-items-start text-muted mt-2" style="font-size: 0.8rem;">
+                                    <i class="bi bi-info-circle me-2" style="font-size: 0.9rem; line-height: 1.4;"></i>
+                                    <span>{{ t('ui.change_password.password_instructions') }}</span>
+                                </div>
+                            </div>
+
+                            <div class="mb-3">
+                                <label for="confirm_password" class="form-label text-secondary">{{ t('ui.signup.confirm_password_label') }}</label>
+                                <input type="password" id="confirm_password" name="confirm_password" class="form-control" required>
+                            </div>
+
+                            <!-- language: html -->
+                            <input type="hidden" name="lang" value="{{ lang or 'en' }}">
+
+                            <!-- Botón actualizado con la clase de branding -->
+                            <button type="submit" class="btn btn-branded-primary w-100 fw-bold py-2 mt-3">{{ t('ui.signup.signup_button') }}</button>
+                        </form>
+                        <!-- Nota de privacidad -->
+                        <p class="text-muted small mb-0 text-center mt-4">
+                            {{ t('ui.signup.disclaimer') }}
+                        </p>
+                    </div>
+                </div>
+            </div>
+        </div>
+    </section>
+</div>
+
+{% endblock %}

iatoolkit/views/__init__.py

@@ -0,0 +1,5 @@
+# Copyright (c) 2024 Fernando Libedinsky
+# Product: IAToolkit
+#
+# IAToolkit is open source software.
+

iatoolkit/views/base_login_view.py

@@ -0,0 +1,96 @@
+# iatoolkit/views/base_login_view.py
+# Copyright (c) 2024 Fernando Libedinsky
+# Product: IAToolkit
+#
+# IAToolkit is open source software.
+
+from flask.views import MethodView
+from flask import render_template, url_for
+from injector import inject
+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.configuration_service import ConfigurationService
+from iatoolkit.services.prompt_service import PromptService
+from iatoolkit.services.i18n_service import I18nService
+from iatoolkit.common.util import Utility
+from iatoolkit.services.jwt_service import JWTService
+from iatoolkit.repositories.models import Company
+
+
+class BaseLoginView(MethodView):
+    """
+    Base class for views that initiate a session and decide the context
+    loading path (fast or slow).
+    """
+    @inject
+    def __init__(self,
+                 profile_service: ProfileService,
+                 auth_service: AuthService,
+                 jwt_service: JWTService,
+                 branding_service: BrandingService,
+                 prompt_service: PromptService,
+                 config_service: ConfigurationService,
+                 query_service: QueryService,
+                 i18n_service: I18nService,
+                 utility: Utility
+                 ):
+        self.profile_service = profile_service
+        self.auth_service = auth_service
+        self.jwt_service = jwt_service
+        self.branding_service = branding_service
+        self.prompt_service = prompt_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_short_name: str,
+                           user_identifier: str,
+                           target_url: str,
+                           redeem_token: str = None):
+        """
+        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_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
+        )
+
+        if prep_result.get('rebuild_needed'):
+            # --- SLOW PATH: Render the loading shell ---
+            return render_template(
+                "onboarding_shell.html",
+                iframe_src_url=target_url,
+                branding=branding_data,
+                onboarding_cards=onboarding_cards
+            )
+        else:
+            # --- FAST PATH: Render the chat page directly ---
+            # LLM configuration: default model and availables
+            default_llm_model, available_llm_models = self.config_service.get_llm_configuration(company_short_name)
+
+            prompts = self.prompt_service.get_user_prompts(company_short_name)
+
+            # Get the entire 'js_messages' block in the correct language.
+            js_translations = self.i18n_service.get_translation_block('js_messages')
+
+            return render_template(
+                "chat.html",
+                company_short_name=company_short_name,
+                user_identifier=user_identifier,
+                prompts=prompts,
+                branding=branding_data,
+                onboarding_cards=onboarding_cards,
+                js_translations=js_translations,
+                redeem_token=redeem_token,
+                llm_default_model=default_llm_model,
+                llm_available_models = available_llm_models,
+                )

iatoolkit/views/change_password_view.py

@@ -0,0 +1,116 @@
+# Copyright (c) 2024 Fernando Libedinsky
+# Product: IAToolkit
+#
+# IAToolkit is open source software.
+
+from flask.views import MethodView
+from flask import render_template, request, url_for, session, redirect, flash
+from iatoolkit.services.profile_service import ProfileService
+from iatoolkit.services.branding_service import BrandingService
+from iatoolkit.services.i18n_service import I18nService
+from itsdangerous import URLSafeTimedSerializer, SignatureExpired
+from flask_bcrypt import Bcrypt
+from injector import inject
+import os
+
+
+class ChangePasswordView(MethodView):
+    @inject
+    def __init__(self,
+                 profile_service: ProfileService,
+                 branding_service: BrandingService,
+                 i18n_service: I18nService):
+        self.profile_service = profile_service
+        self.branding_service = branding_service
+        self.i18n_service = i18n_service
+
+        self.serializer = URLSafeTimedSerializer(os.getenv("IATOOLKIT_SECRET_KEY"))
+        self.bcrypt = Bcrypt()
+
+    def get(self, company_short_name: str, token: str):
+        try:
+            company = self.profile_service.get_company_by_short_name(company_short_name)
+            if not company:
+                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_short_name)
+
+            try:
+                # Decodificar el token
+                email = self.serializer.loads(token, salt='password-reset', max_age=3600)
+            except SignatureExpired as e:
+                flash(self.i18n_service.t('errors.change_password.token_expired'), 'error')
+                return render_template('forgot_password.html',
+                                    branding=branding_data)
+
+            return render_template('change_password.html',
+                                   company_short_name=company_short_name,
+                                   company=company,
+                                   branding=branding_data,
+                                   token=token,
+                                   email=email)
+        except Exception as e:
+            message = self.i18n_service.t('errors.templates.processing_error', error=str(e))
+            return render_template(
+                "error.html",
+                company_short_name=company_short_name,
+                branding=branding_data,
+                message=message
+            ), 500
+
+    def post(self, company_short_name: str, token: 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=self.i18n_service.t('errors.templates.company_not_found')), 404
+
+        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)
+        except SignatureExpired:
+            flash(self.i18n_service.t('errors.change_password.token_expired'), 'error')
+
+            return render_template('forgot_password.html',
+                                   company_short_name=company_short_name,
+                                   company=company,
+                                   branding=branding_data)
+
+        try:
+            # Obtener datos del formulario
+            temp_code = request.form.get('temp_code')
+            new_password = request.form.get('new_password')
+            confirm_password = request.form.get('confirm_password')
+
+            response = self.profile_service.change_password(
+                email=email,
+                temp_code=temp_code,
+                new_password=new_password,
+                confirm_password=confirm_password
+            )
+
+            if "error" in response:
+                flash(response["error"], 'error')
+
+                return render_template(
+                    'change_password.html',
+                    token=token,
+                    company_short_name=company_short_name,
+                    branding=branding_data,
+                    form_data={"temp_code": temp_code,
+                               "new_password": new_password,
+                               "confirm_password": confirm_password}), 400
+
+            flash(self.i18n_service.t('flash_messages.password_changed_success'), 'success')
+            return redirect(url_for('home', company_short_name=company_short_name))
+
+        except Exception as e:
+            message = self.i18n_service.t('errors.templates.processing_error', error=str(e))
+            return render_template(
+                "error.html",
+                company_short_name=company_short_name,
+                branding=branding_data,
+                message=message
+            ), 500

iatoolkit/views/chat_view.py

@@ -0,0 +1,76 @@
+from flask import render_template, redirect, url_for, request
+from flask.views import MethodView
+from injector import inject
+from iatoolkit.services.profile_service import ProfileService
+from iatoolkit.services.branding_service import BrandingService
+from iatoolkit.services.configuration_service import ConfigurationService
+from iatoolkit.services.prompt_service import PromptService
+from iatoolkit.services.i18n_service import I18nService
+
+class ChatView(MethodView):
+    """
+    Handles direct access to the chat interface.
+    Validates if the user has an active session for the company.
+    """
+
+    @inject
+    def __init__(self,
+                 profile_service: ProfileService,
+                 branding_service: BrandingService,
+                 config_service: ConfigurationService,
+                 prompt_service: PromptService,
+                 i18n_service: I18nService):
+        self.profile_service = profile_service
+        self.branding_service = branding_service
+        self.config_service = config_service
+        self.prompt_service = prompt_service
+        self.i18n_service = i18n_service
+
+    def get(self, company_short_name: str):
+        # 1. Validate Company
+        company = self.profile_service.get_company_by_short_name(company_short_name)
+        if not company:
+            return render_template('error.html',
+                                   message=self.i18n_service.t('errors.templates.company_not_found')), 404
+
+        # 2. Check Session
+        session_info = self.profile_service.get_current_session_info()
+        user_identifier = session_info.get('user_identifier')
+        session_company = session_info.get('company_short_name')
+
+        # If no user or session belongs to another company -> Redirect to Home
+        if not user_identifier or session_company != company_short_name:
+            return redirect(url_for('home', company_short_name=company_short_name))
+
+        # 3. Prepare Context for Chat
+        # (This logic mirrors the FAST PATH in BaseLoginView)
+        try:
+            branding_data = self.branding_service.get_company_branding(company_short_name)
+            onboarding_cards = self.config_service.get_configuration(company_short_name, 'onboarding_cards')
+            default_llm_model, available_llm_models = self.config_service.get_llm_configuration(company_short_name)
+            prompts = self.prompt_service.get_user_prompts(company_short_name)
+            js_translations = self.i18n_service.get_translation_block('js_messages')
+
+            return render_template(
+                "chat.html",
+                company_short_name=company_short_name,
+                user_identifier=user_identifier,
+                prompts=prompts,
+                branding=branding_data,
+                onboarding_cards=onboarding_cards,
+                js_translations=js_translations,
+                llm_default_model=default_llm_model,
+                llm_available_models=available_llm_models,
+                # redeem_token is None for direct access
+                redeem_token=None
+            )
+        except Exception as e:
+            # Fallback error handling
+            message = self.i18n_service.t('errors.templates.processing_error', error=str(e))
+            branding_data = self.branding_service.get_company_branding(company_short_name)
+            return render_template(
+                "error.html",
+                company_short_name=company_short_name,
+                branding=branding_data,
+                message=message
+            ), 500

iatoolkit/views/embedding_api_view.py

@@ -0,0 +1,65 @@
+# iatoolkit/views/embedding_api_view.py
+# Copyright (c) 2024 Fernando Libedinsky
+# Product: IAToolkit
+#
+# IAToolkit is open source software.
+
+from flask import request, jsonify
+from flask.views import MethodView
+from iatoolkit.services.embedding_service import EmbeddingService
+from iatoolkit.services.auth_service import AuthService
+from injector import inject
+import logging
+
+class EmbeddingApiView(MethodView):
+    """
+    Handles API requests to generate an embedding for a given text.
+    Authentication is based on the active Flask session.
+    """
+
+    @inject
+    def __init__(self,
+                 auth_service: AuthService,
+                 embedding_service: EmbeddingService):
+        self.auth_service = auth_service
+        self.embedding_service = embedding_service
+
+    def post(self, company_short_name: str):
+        """
+        Generates an embedding for the text provided in the request body.
+        Expects a JSON payload with a "text" key.
+        """
+        try:
+            # 1. Authenticate the user from the current session
+            auth_result = self.auth_service.verify(anonymous=True)
+            if not auth_result.get("success"):
+                return jsonify(auth_result), auth_result.get("status_code", 401)
+
+            # 2. Validate incoming request data
+            if not request.is_json:
+                return jsonify({"error": "Request must be JSON"}), 400
+
+            data = request.get_json()
+            text = data.get('text')
+
+            if not text:
+                return jsonify({"error": "The 'text' key is required."}), 400
+
+            # 3. Call the embedding service, now passing the company_short_name
+            embedding_b64 = self.embedding_service.embed_text(
+                company_short_name=company_short_name,
+                text=text,
+                to_base64=True
+            )
+
+            model_name = self.embedding_service.get_model_name(company_short_name)
+            response = {
+                "embedding": embedding_b64,
+                "model": model_name
+            }
+            return jsonify(response), 200
+
+        except Exception as e:
+            logging.exception(f"Unexpected error in EmbeddingApiView: {e}")
+            # Return a generic error message to the client
+            return jsonify({"error": "An internal error occurred while generating the embedding."}), 500

iatoolkit/views/forgot_password_view.py

@@ -0,0 +1,75 @@
+# Copyright (c) 2024 Fernando Libedinsky
+# Product: IAToolkit
+#
+# IAToolkit is open source software.
+
+from flask.views import MethodView
+from flask import render_template, request, url_for, redirect, session, flash
+from injector import inject
+from iatoolkit.services.profile_service import ProfileService
+from iatoolkit.services.branding_service import BrandingService
+from iatoolkit.services.i18n_service import I18nService
+from itsdangerous import URLSafeTimedSerializer
+import os
+
+class ForgotPasswordView(MethodView):
+    @inject
+    def __init__(self, profile_service: ProfileService,
+                 branding_service: BrandingService,
+                 i18n_service: I18nService):
+        self.profile_service = profile_service
+        self.branding_service = branding_service
+        self.i18n_service = i18n_service
+
+        self.serializer = URLSafeTimedSerializer(os.getenv("IATOOLKIT_SECRET_KEY"))
+
+    def get(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=self.i18n_service.t('errors.templates.company_not_found')), 404
+
+        branding_data = self.branding_service.get_company_branding(company_short_name)
+        current_lang = request.args.get("lang", "en")
+        return render_template('forgot_password.html',
+                               company_short_name=company_short_name,
+                               branding=branding_data,
+                               lang=current_lang
+                               )
+
+    def post(self, company_short_name: str):
+
+        try:
+            company = self.profile_service.get_company_by_short_name(company_short_name)
+            if not company:
+                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_short_name)
+            email = request.form.get('email')
+
+            # create a safe token and url for it
+            token = self.serializer.dumps(email, salt='password-reset')
+            reset_url = url_for('change_password',
+                                company_short_name=company_short_name,
+                                token=token, _external=True)
+
+            response = self.profile_service.forgot_password(
+                        company_short_name=company_short_name,
+                        email=email, reset_url=reset_url)
+            if "error" in response:
+                flash(response["error"], 'error')
+                return render_template(
+                    'forgot_password.html',
+                    company_short_name=company_short_name,
+                    branding=branding_data,
+                    form_data={"email": email}), 400
+
+            flash(self.i18n_service.t('flash_messages.forgot_password_success'), 'success')
+            lang = request.args.get("lang", "en")
+            return redirect(url_for('home', company_short_name=company_short_name, lang=lang))
+
+        except Exception as e:
+            flash(self.i18n_service.t('errors.general.unexpected_error'), 'error')
+            return redirect(url_for('home', company_short_name=company_short_name))