iatoolkit 0.9.4__tar.gz → 0.10.0__tar.gz

{iatoolkit-0.9.4 → iatoolkit-0.10.0}/PKG-INFO

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

{iatoolkit-0.9.4 → iatoolkit-0.10.0}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "iatoolkit"
-version = "0.9.4"
+version = "0.10.0"
 requires-python = ">=3.11"
 description = "IAToolkit"
 readme = "readme.md"

{iatoolkit-0.9.4 → iatoolkit-0.10.0}/src/iatoolkit/base_company.py

@@ -26,8 +26,10 @@ class BaseCompany(ABC):
         self.company = self.profile_repo.get_company_by_short_name(short_name)
         return self.company
 
-    def _create_company(self, name: str, short_name: str) -> Company:
-        company_obj = Company(name=name, short_name=short_name, allow_jwt=True)
+    def _create_company(self, short_name: str, name: str, branding: dict | None = None) -> Company:
+        company_obj = Company(short_name=short_name,
+                              name=name,
+                              branding=branding)
         self.company = self.profile_repo.create_company(company_obj)
         return self.company
 
@@ -67,6 +69,7 @@ class BaseCompany(ABC):
             **kwargs
         )
 
+
     @abstractmethod
     # initialize all the database tables  needed
     def register_company(self):

{iatoolkit-0.9.4 → iatoolkit-0.10.0}/src/iatoolkit/iatoolkit.py

@@ -278,6 +278,7 @@ class IAToolkit:
         from iatoolkit.services.profile_service import ProfileService
         from iatoolkit.services.jwt_service import JWTService
         from iatoolkit.services.dispatcher_service import Dispatcher
+        from iatoolkit.services.branding_service import BrandingService
 
         binder.bind(QueryService, to=QueryService)
         binder.bind(TaskService, to=TaskService)
@@ -290,6 +291,8 @@ class IAToolkit:
         binder.bind(ProfileService, to=ProfileService)
         binder.bind(JWTService, to=JWTService)
         binder.bind(Dispatcher, to=Dispatcher)
+        binder.bind(BrandingService, to=BrandingService)
+
 
     def _bind_infrastructure(self, binder: Binder):
         from iatoolkit.infra.llm_client import llmClient

{iatoolkit-0.9.4 → iatoolkit-0.10.0}/src/iatoolkit/repositories/models.py

@@ -57,10 +57,10 @@ class Company(Base):
     openai_api_key = Column(String, nullable=True)
     gemini_api_key = Column(String, nullable=True)
 
-    logo_file = Column(String(128), nullable=True, default='')
+    branding = Column(JSON, nullable=True)
     parameters = Column(JSON, nullable=True, default={})
     created_at = Column(DateTime, default=datetime.now)
-    allow_jwt = Column(Boolean, default=False, nullable=True)
+    allow_jwt = Column(Boolean, default=True, nullable=True)
 
     documents = relationship("Document",
                              back_populates="company",

{iatoolkit-0.9.4 → iatoolkit-0.10.0}/src/iatoolkit/repositories/profile_repo.py

@@ -73,7 +73,7 @@ class ProfileRepo:
         company = self.session.query(Company).filter_by(name=new_company.name).first()
         if company:
             company.parameters = new_company.parameters
-            company.logo_file = new_company.logo_file
+            company.branding = new_company.branding
         else:
             self.session.add(new_company)
             company = new_company

iatoolkit-0.10.0/src/iatoolkit/services/branding_service.py

@@ -0,0 +1,60 @@
+# Copyright (c) 2024 Fernando Libedinsky
+# Product: IAToolkit
+#
+# IAToolkit is open source software.
+
+from iatoolkit.repositories.models import Company
+
+
+class BrandingService:
+    """
+    Servicio centralizado que gestiona la configuración de branding.
+    """
+
+    def __init__(self):
+        """
+        Define los estilos de branding por defecto para la aplicación.
+        """
+        self._default_branding = {
+            # Colores del contenedor del encabezado
+            "header_background_color": "#FFFFFF",  # Fondo blanco por defecto
+            "header_text_color": "#6C757D",  # Color de texto 'muted' de Bootstrap
+
+            # Estilos para el nombre de la compañía
+            "company_name_font_weight": "bold",  # Nombre en negrita por defecto
+            "company_name_font_size": "1rem",  # Tamaño de fuente estándar
+        }
+
+    def get_company_branding(self, company: Company | None) -> dict:
+        """
+        Retorna los estilos de branding finales para una compañía,
+        fusionando los valores por defecto con los personalizados.
+
+        Args:
+            company: El objeto Company, que puede contener un dict de branding.
+
+        Returns:
+            Un diccionario con todos los estilos de branding listos para usar en la plantilla.
+        """
+        # Empezamos con una copia de los valores por defecto
+        final_branding_values = self._default_branding.copy()
+
+        # Si la compañía existe y tiene branding personalizado, lo fusionamos.
+        if company and company.branding:
+            final_branding_values.update(company.branding)
+
+        # Construimos las cadenas de estilo completas
+        header_style = (
+            f"background-color: {final_branding_values['header_background_color']}; "
+            f"color: {final_branding_values['header_text_color']};"
+        )
+        company_name_style = (
+            f"font-weight: {final_branding_values['company_name_font_weight']}; "
+            f"font-size: {final_branding_values['company_name_font_size']};"
+        )
+
+        return {
+            "name": company.name if company else "IAToolkit",
+            "header_style": header_style,
+            "company_name_style": company_name_style
+        }

{iatoolkit-0.9.4 → iatoolkit-0.10.0}/src/iatoolkit/templates/chat.html

@@ -4,39 +4,49 @@
 
 {% block content %}
 <!-- Sección de encabezado con el usuario conectado -->
-<div class="company-section d-flex justify-content-between align-items-center">
-    <!-- Izquierda: Datos del usuario -->
-    <span class="text-muted small fw-bold">
-        {{ company_short_name }} - {{ external_user_id or user.email }}
+<div class="company-section d-flex justify-content-between align-items-center px-3 py-2"
+     style="{{ branding.header_style }}">
+
+    <!-- Izquierda: Nombre de la Empresa -->
+    <span style="{{ branding.company_name_style }}">
+        {{ branding.name }}
     </span>
 
     <!-- Derecha: Grupo de iconos de acción -->
+    <!-- Derecha: Grupo de información y acciones del usuario -->
     <div class="d-flex align-items-center">
-        <!-- "Powered by"  -->
-        <span class="text-muted" style="font-size: 0.75rem;">
-            Powered by <strong>IAToolkit</strong>
+        <!-- 1. ID de Usuario -->
+        <span class="small fw-semibold">
+            {{ external_user_id or user.email }}
         </span>
 
-        <!-- Icono de Historial -->
+        <!-- 2. Separador Vertical -->
+        <div class="vr mx-3"></div>
+
+        <!-- 3. Iconos de Acción -->
         <a href="javascript:void(0);" id="history-button"
-           class="ms-3 action-icon-style" title="Historial con mis consultas">
+           class="action-icon-style" title="Historial con mis consultas" style="color: inherit;">
             <i class="bi bi-clock-history"></i>
         </a>
-
-        <!-- Icono de Feedback -->
         <a href="javascript:void(0);" id="send-feedback-button"
-           class="ms-3 action-icon-style" title="Tu feedback es muy importante">
+           class="ms-3 action-icon-style" title="Tu feedback es muy importante" style="color: inherit;">
             <i class="bi bi-emoji-smile"></i>
         </a>
 
-        <!-- Icono de cerrar sesión -->
+        <!-- 4. Powered by (con mayor separación) -->
+        <span class="text-muted ms-4" style="font-size: 0.75rem;">
+            Powered by <strong>IAToolkit</strong>
+        </span>
+
+        <!-- 5. Icono de cerrar sesión (al final) -->
         {% if user.email %}
             <a href="{{ url_for('logout', company_short_name=company_short_name) }}"
-                 class="ms-3 action-icon-style" title="Cerrar sesión">
+                 class="ms-3 action-icon-style" title="Cerrar sesión" style="color: inherit;">
                 <i class="bi bi-box-arrow-right"></i>
             </a>
         {% endif %}
     </div>
+
 </div>
 
 <div id="chat-container"

{iatoolkit-0.9.4 → iatoolkit-0.10.0}/src/iatoolkit/templates/home.html

@@ -95,8 +95,6 @@
 
 <script>
   // Variables pasadas desde Flask (HomeView)
-  const CHAT_TOKEN_REQUEST_URL = "{{ chat_token_request_url|safe }}";
-  const PUBLIC_CHAT_URL_TEMPLATE = "{{ public_chat_url_template|safe }}";
   const API_KEY = "{{ api_key|safe }}";
 
   $(document).ready(function () {

{iatoolkit-0.9.4 → iatoolkit-0.10.0}/src/iatoolkit/views/chat_view.py

@@ -10,6 +10,7 @@ from injector import inject
 import os
 from iatoolkit.common.auth import IAuthentication
 from iatoolkit.services.prompt_manager_service import PromptService
+from iatoolkit.services.branding_service import BrandingService
 
 
 class ChatView(MethodView):
@@ -17,10 +18,13 @@ class ChatView(MethodView):
     def __init__(self,
                  iauthentication: IAuthentication,
                  prompt_service: PromptService,
-                 profile_service: ProfileService):
+                 profile_service: ProfileService,
+                 branding_service: BrandingService
+                 ):
         self.iauthentication = iauthentication
         self.profile_service = profile_service
         self.prompt_service = prompt_service
+        self.branding_service = branding_service
 
     def get(self, company_short_name: str):
         # get access credentials
@@ -40,8 +44,11 @@ class ChatView(MethodView):
         # 2. get the company prompts
         prompts = self.prompt_service.get_user_prompts(company_short_name)
 
+        # 3. get the  branding data
+        branding_data = self.branding_service.get_company_branding(company)
+
         return render_template("chat.html",
-                               company=company,
+                               branding=branding_data,
                                company_short_name=company_short_name,
                                is_mobile=is_mobile,
                                alert_message=alert_message,

{iatoolkit-0.9.4 → iatoolkit-0.10.0}/src/iatoolkit/views/external_chat_login_view.py

@@ -13,6 +13,7 @@ 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 ExternalChatLoginView(MethodView):
     @inject
@@ -21,13 +22,15 @@ class ExternalChatLoginView(MethodView):
                  query_service: QueryService,
                  prompt_service: PromptService,
                  iauthentication: IAuthentication,
-                 jwt_service: JWTService
+                 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()
@@ -35,6 +38,7 @@ class ExternalChatLoginView(MethodView):
             return jsonify({"error": "Falta external_user_id"}), 400
 
         external_user_id = data['external_user_id']
+
         # 1. get access credentials
         iaut = self.iauthentication.verify(
             company_short_name,
@@ -67,13 +71,16 @@ class ExternalChatLoginView(MethodView):
             # 3. get the prompt list from backend
             prompts = self.prompt_service.get_user_prompts(company_short_name)
 
-            # 4. render the chat page with the company/user information.
+            # 4. get the branding data
+            branding_data = self.branding_service.get_company_branding(company)
+
+            # 5. render the chat page with the company/user information.
             user_agent = request.user_agent
             is_mobile = user_agent.platform in ["android", "iphone", "ipad"] or "mobile" in user_agent.string.lower()
 
             chat_html = render_template("chat.html",
-                                        company=company,
                                         company_short_name=company_short_name,
+                                        branding=branding_data,
                                         external_user_id=external_user_id,
                                         is_mobile=is_mobile,
                                         auth_method='jwt',  # login method is JWT

{iatoolkit-0.9.4 → iatoolkit-0.10.0}/src/iatoolkit.egg-info/PKG-INFO

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

{iatoolkit-0.9.4 → iatoolkit-0.10.0}/src/iatoolkit.egg-info/SOURCES.txt

@@ -44,6 +44,7 @@ src/iatoolkit/repositories/tasks_repo.py
 src/iatoolkit/repositories/vs_repo.py
 src/iatoolkit/services/__init__.py
 src/iatoolkit/services/benchmark_service.py
+src/iatoolkit/services/branding_service.py
 src/iatoolkit/services/dispatcher_service.py
 src/iatoolkit/services/document_service.py
 src/iatoolkit/services/excel_service.py