iatoolkit 0.9.4__py3-none-any.whl → 0.10.1__py3-none-any.whl

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/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/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/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/services/branding_service.py

@@ -0,0 +1,70 @@
+# 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 texto primario (ej. nombre de la compañía)
+            "primary_font_weight": "bold",
+            "primary_font_size": "1rem",
+
+            # Estilos para el texto secundario (ej. ID de usuario)
+            "secondary_font_weight": "600",  # Semibold
+            "secondary_font_size": "0.875rem"  # Equivale a la clase 'small' de Bootstrap
+
+        }
+
+    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']};"
+        )
+        primary_text_style = (
+            f"font-weight: {final_branding_values['primary_font_weight']}; "
+            f"font-size: {final_branding_values['primary_font_size']};"
+        )
+        secondary_text_style = (
+            f"font-weight: {final_branding_values['secondary_font_weight']}; "
+            f"font-size: {final_branding_values['secondary_font_size']};"
+        )
+
+        return {
+            "name": company.name if company else "IAToolkit",
+            "header_style": header_style,
+            "primary_text_style": primary_text_style,
+            "secondary_text_style": secondary_text_style
+        }

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.primary_text_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 style="{{ branding.secondary_text_style }}">
+            {{ 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="{{ branding.secondary_text_style }}">
+            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/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/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/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.dist-info → iatoolkit-0.10.1.dist-info}/METADATA

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

{iatoolkit-0.9.4.dist-info → iatoolkit-0.10.1.dist-info}/RECORD

@@ -1,8 +1,8 @@
 iatoolkit/__init__.py,sha256=4PWjMJjktixtrxF6BY405qyA50Sv967kEP2x-oil6qk,1120
-iatoolkit/base_company.py,sha256=gM3X7xedNaVLodOObvjUN9eJKPO5-HmSLaBFa9jowCk,4277
+iatoolkit/base_company.py,sha256=4usUdOC57Nk1XsawAEyC7mlzb0jROadu64YcYf6poiM,4371
 iatoolkit/cli_commands.py,sha256=G5L9xQXZ0lVFXQWBaE_KEZHyfuiT6PL1nTQRoSdnBzc,2302
 iatoolkit/company_registry.py,sha256=tduqt3oV8iDX_IB1eA7KIgvIxE4edTcy-3qZIXh3Lzw,2549
-iatoolkit/iatoolkit.py,sha256=KFx4bQJwSAQdIIAMzyfjQNA5q0k4HS4P738JnYY533k,16830
+iatoolkit/iatoolkit.py,sha256=EXmHQEgLCqbBP7Cwq6-O1NPj7O3M_2Cg9JLp5xa41ig,16960
 iatoolkit/common/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 iatoolkit/common/auth.py,sha256=8NH6MQXfddLQd1GxrO2op3IYrrP4SMQKoKzj1o1jZmc,8486
 iatoolkit/common/exceptions.py,sha256=EXx40n5htp7UiOM6P1xfJ9U6NMcADqm62dlFaKz7ICU,1154
@@ -30,12 +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=RJH0dtq2VIDJFaFDufV05NUnNW4WtNpdTjhONtfLyCw,13179
-iatoolkit/repositories/profile_repo.py,sha256=ZULbh7ZOI9BFVi7HEfXtx3SaStgFnsCBVM_zuli_F8I,4062
+iatoolkit/repositories/models.py,sha256=vlSC-Z1cbho5dLrvoMnEroafO7yC3JXsZKn7SqoIXKU,13158
+iatoolkit/repositories/profile_repo.py,sha256=X2tnqAdaae-SA9sQq-gBqopVdf77I_Q9J_Aa38P0tjY,4060
 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=UG-cuVrMmxXRqSxOniDXew52D4WZYH8PZ84DW8EdLuo,2632
 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
@@ -75,21 +76,21 @@ iatoolkit/system_prompts/sql_rules.prompt,sha256=y4nURVnb9AyFwt-lrbMNBHHtZlhk6kC
 iatoolkit/templates/about.html,sha256=ciC08grUVz5qLzdzDDqDX31xirg5PrJIRYabWpV9oA8,294
 iatoolkit/templates/base.html,sha256=LXXB8oPrcBFkf2pLfOSyAaSh66kHbs4SEujpFL3h9Nw,2163
 iatoolkit/templates/change_password.html,sha256=DFfQSFcZ2YJZNFis2IXfzEKStxTf4i9f4eQ_6GiyNs8,2342
-iatoolkit/templates/chat.html,sha256=YZKHn0VINtAFoyAISgahMD29iNO_oHON2N5EZkoCkE4,8219
+iatoolkit/templates/chat.html,sha256=d0TCeDKEI2fZCepThqRx3xqUDfgnCSFKwdE8Cqq3t5s,8587
 iatoolkit/templates/chat_modals.html,sha256=9OtJHRMeBup8zojPnUqSl7Fd5StPdvOZMs2MiJPqVFY,5704
 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=h6qZ6d5tPYAGeHsVr-o6NFG3YMs1QeDVCxFlnPneFEA,8345
+iatoolkit/templates/home.html,sha256=HpJmz0rCMBt3XYd83hecBCESvMg85n56PV5XvE_8Q_g,8201
 iatoolkit/templates/login.html,sha256=r4hy7MsQkfDqi6pBRNkkRiFr3GPSoHCT89R5lQLUWZc,1991
 iatoolkit/templates/signup.html,sha256=J8wOjUhUe_KozyThDTWHjXpSJ1ubR2IDVobThtkSRuo,3819
 iatoolkit/templates/test.html,sha256=rwNtxC83tbCl5COZFXYvmRBxxmgFJtPNuVBd_nq9KWY,133
 iatoolkit/views/__init__.py,sha256=5JqK9sZ6jBuK83zDQokUhxQ0wuJJJ9DXB8pYCLkX7X4,102
 iatoolkit/views/change_password_view.py,sha256=rSebwecI1zwBgR2yvAhcfMwWpGDa4QbVAIllgtSOo9k,3940
 iatoolkit/views/chat_token_request_view.py,sha256=wf32_A2Sq8NHYWshCwL10Tovd1znLoD0jQjzutR3sVE,4408
-iatoolkit/views/chat_view.py,sha256=rnljdVHfe2E4E57ku6eblVgwT6SrLi1v-yqXGri68Ks,2078
+iatoolkit/views/chat_view.py,sha256=BD1N-Jo9IOoS6baX1bzdM7DKa2zepVHZnJIxiF0vNc8,2381
 iatoolkit/views/download_file_view.py,sha256=1gZ0ipqeCn39sTrJFo1-tlewlcSF7s_YNTvE4qd0HOw,2010
-iatoolkit/views/external_chat_login_view.py,sha256=uh8DQRfZpZOXYr9qoxtwSwtL3PNdIJh-R5OjGgLqcoY,3782
+iatoolkit/views/external_chat_login_view.py,sha256=h-Xr7UkQM7QxLZIDQ0fXb-BUfPbABQ1gjiP--FrXnpc,4075
 iatoolkit/views/external_login_view.py,sha256=EHxN2omqTDkWyMp0FVCNDFt-JJ2hpg8LvprKanGR1gY,1355
 iatoolkit/views/file_store_view.py,sha256=hUm5wX4E5oqJJEPEAObEj-nPiRp5EJIICULSfAWmHCs,1933
 iatoolkit/views/forgot_password_view.py,sha256=Rk8Qbe9Fz7Wlgje1rt29I15gFM-a089EBi2at4FT7kA,2715
@@ -103,7 +104,7 @@ iatoolkit/views/tasks_review_view.py,sha256=keLsLCyOTTlcoIapnB_lbuSvLwrPVZVpBiFC
 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.9.4.dist-info/METADATA,sha256=Ib0fPJJbjSJbKpKakFW2pnWBADjrZRiATXy4DlhXoX0,9300
-iatoolkit-0.9.4.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-iatoolkit-0.9.4.dist-info/top_level.txt,sha256=V_w4QvDx0b1RXiy8zTCrD1Bp7AZkFe3_O0-9fMiwogg,10
-iatoolkit-0.9.4.dist-info/RECORD,,
+iatoolkit-0.10.1.dist-info/METADATA,sha256=HQjVa4XstjTkOTiSsiQan7QknkV0o1T_NPKsLC2iv2g,9301
+iatoolkit-0.10.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+iatoolkit-0.10.1.dist-info/top_level.txt,sha256=V_w4QvDx0b1RXiy8zTCrD1Bp7AZkFe3_O0-9fMiwogg,10
+iatoolkit-0.10.1.dist-info/RECORD,,