iatoolkit 0.63.1__py3-none-any.whl → 0.67.0__py3-none-any.whl

iatoolkit/views/init_context_api_view.py

@@ -3,7 +3,8 @@ from injector import inject
 from iatoolkit.services.query_service import QueryService
 from iatoolkit.services.profile_service import ProfileService
 from iatoolkit.services.auth_service import AuthService
-from flask import jsonify, request
+from iatoolkit.services.i18n_service import I18nService
+from flask import jsonify
 import logging
 
 
@@ -17,10 +18,12 @@ class InitContextApiView(MethodView):
     def __init__(self,
                  auth_service: AuthService,
                  query_service: QueryService,
-                 profile_service: ProfileService):
+                 profile_service: ProfileService,
+                 i18n_service: I18nService):
         self.auth_service = auth_service
         self.query_service = query_service
         self.profile_service = profile_service
+        self.i18n_service = i18n_service
 
     def post(self, company_short_name: str):
         """
@@ -28,14 +31,14 @@ class InitContextApiView(MethodView):
         an active web session or by the external_user_id in the JSON payload
         for API calls.
         """
-        # 1. Authenticate the request. This handles both session and API Key.
-        auth_result = self.auth_service.verify()
-        if not auth_result.get("success"):
-            return jsonify(auth_result), auth_result.get("status_code")
+        try:
+            # 1. Authenticate the request. This handles both session and API Key.
+            auth_result = self.auth_service.verify()
+            if not auth_result.get("success"):
+                return jsonify(auth_result), auth_result.get("status_code")
 
-        user_identifier = auth_result.get('user_identifier')
+            user_identifier = auth_result.get('user_identifier')
 
-        try:
             # 2. Execute the forced rebuild sequence using the unified identifier.
             self.query_service.session_context.clear_all_context(company_short_name, user_identifier)
             logging.info(f"Context for {company_short_name}/{user_identifier} has been cleared.")
@@ -52,11 +55,13 @@ class InitContextApiView(MethodView):
             )
 
             # 3. Respond with JSON, as this is an API endpoint.
-            return jsonify({'status': 'OK', 'message': 'El contexto se ha recargado con éxito.'}), 200
+            success_message = self.i18n_service.t('api_responses.context_reloaded_success')
+            return jsonify({'status': 'OK', 'message': success_message}), 200
 
         except Exception as e:
-            logging.exception(f"Error durante la recarga de contexto {user_identifier}: {e}")
-            return jsonify({"error_message": str(e)}), 500
+            logging.exception(f"errors while reloading context: {e}")
+            error_message = self.i18n_service.t('errors.general.unexpected_error', error=str(e))
+            return jsonify({"error_message": error_message}), 500
 
     def options(self, company_short_name):
         """

iatoolkit/views/llmquery_api_view.py

@@ -4,6 +4,7 @@ from injector import inject
 from iatoolkit.services.query_service import QueryService
 from iatoolkit.services.auth_service import AuthService
 from iatoolkit.services.profile_service import ProfileService
+from iatoolkit.services.i18n_service import I18nService
 import logging
 
 class LLMQueryApiView(MethodView):
@@ -12,34 +13,45 @@ class LLMQueryApiView(MethodView):
     """
 
     @inject
-    def __init__(self, auth_service: AuthService, query_service: QueryService, profile_service: ProfileService):
+    def __init__(self,
+                 auth_service: AuthService,
+                 query_service: QueryService,
+                 profile_service: ProfileService,
+                 i18n_service: I18nService):
         self.auth_service = auth_service
         self.query_service = query_service
         self.profile_service = profile_service
+        self.i18n_service = i18n_service
 
     def post(self, company_short_name: str):
-        # 1. Authenticate the API request.
-        auth_result = self.auth_service.verify()
-        if not auth_result.get("success"):
-            return jsonify(auth_result), auth_result.get("status_code")
-
-        # 2. Get the user identifier from the payload.
-        user_identifier = auth_result.get('user_identifier')
-
-        data = request.get_json()
-        if not data:
-            return jsonify({"error": "Invalid JSON body"}), 400
-
-        # 4. Call the unified query service method.
-        result = self.query_service.llm_query(
-            company_short_name=company_short_name,
-            user_identifier=user_identifier,
-            question=data.get('question', ''),
-            prompt_name=data.get('prompt_name'),
-            client_data=data.get('client_data', {}),
-            files=data.get('files', [])
-        )
-        if 'error' in result:
-            return jsonify(result), 400
-
-        return jsonify(result), 200
+        try:
+            # 1. Authenticate the API request.
+            auth_result = self.auth_service.verify()
+            if not auth_result.get("success"):
+                return jsonify(auth_result), auth_result.get("status_code")
+
+            # 2. Get the user identifier from the payload.
+            user_identifier = auth_result.get('user_identifier')
+
+            data = request.get_json()
+            if not data:
+                return jsonify({"error": "Invalid JSON body"}), 400
+
+            # 4. Call the unified query service method.
+            result = self.query_service.llm_query(
+                company_short_name=company_short_name,
+                user_identifier=user_identifier,
+                question=data.get('question', ''),
+                prompt_name=data.get('prompt_name'),
+                client_data=data.get('client_data', {}),
+                files=data.get('files', [])
+            )
+            if 'error' in result:
+                return jsonify(result), 400
+
+            return jsonify(result), 200
+
+        except Exception as e:
+            logging.exception(
+                f"Unexpected error: {e}")
+            return jsonify({"error_message": self.i18n_service.t('errors.general.unexpected_error', error=str(e))}), 500

iatoolkit/views/login_simulation_view.py

@@ -10,17 +10,29 @@ from flask.views import MethodView
 from flask import render_template, request, Response
 from injector import inject
 from iatoolkit.services.profile_service import ProfileService
+from iatoolkit.services.branding_service import BrandingService
 
 
 class LoginSimulationView(MethodView):
     @inject
     def __init__(self,
-                 profile_service: ProfileService):
+                 profile_service: ProfileService,
+                 branding_service: BrandingService):
         self.profile_service = profile_service
+        self.branding_service = branding_service
+
 
     def get(self, company_short_name: str = None):
-        """Muestra el formulario para iniciar la simulación."""
+        company = self.profile_service.get_company_by_short_name(company_short_name)
+        if not company:
+            return render_template('error.html',
+                                   company_short_name=company_short_name,
+                                   message="Empresa no encontrada"), 404
+
+        branding_data = self.branding_service.get_company_branding(company)
+
         return render_template('login_simulation.html',
+                               branding=branding_data,
                                company_short_name=company_short_name
                                )
 

iatoolkit/views/login_view.py

@@ -4,15 +4,16 @@
 # IAToolkit is open source software.
 
 from flask.views import MethodView
-from flask import request, redirect, render_template, url_for
+from flask import (request, redirect, render_template, url_for,
+                   render_template_string, flash)
 from injector import inject
 from iatoolkit.services.profile_service import ProfileService
-from iatoolkit.services.auth_service import AuthService
 from iatoolkit.services.jwt_service import JWTService
 from iatoolkit.services.query_service import QueryService
 from iatoolkit.services.prompt_manager_service import PromptService
 from iatoolkit.services.branding_service import BrandingService
 from iatoolkit.services.onboarding_service import OnboardingService
+from iatoolkit.services.i18n_service import I18nService
 from iatoolkit.views.base_login_view import BaseLoginView
 import logging
 
@@ -25,8 +26,10 @@ class LoginView(BaseLoginView):
     def post(self, company_short_name: str):
         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('error.html',
+                                   message=self.i18n_service.t('errors.templates.company_not_found')), 404
 
+        branding_data = self.branding_service.get_company_branding(company)
         email = request.form.get('email')
         password = request.form.get('password')
 
@@ -38,15 +41,15 @@ class LoginView(BaseLoginView):
         )
 
         if not auth_response['success']:
-            branding_data = self.branding_service.get_company_branding(company)
+            flash(auth_response["message"], 'error')
+            home_template = self.utility.get_company_template(company_short_name, "home.html")
 
-            return render_template(
-                'index.html',
+            return render_template_string(
+                home_template,
                 company_short_name=company_short_name,
                 company=company,
                 branding=branding_data,
                 form_data={"email": email},
-                alert_message=auth_response["message"]
             ), 400
 
         user_identifier = auth_response['user_identifier']
@@ -60,10 +63,13 @@ class LoginView(BaseLoginView):
         try:
             return self._handle_login_path(company, user_identifier, target_url)
         except Exception as e:
-            return render_template("error.html",
-            company=company,
-            company_short_name=company_short_name,
-            message=f"Error processing login path: {str(e)}"), 500
+            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
 
 
 class FinalizeContextView(MethodView):
@@ -74,12 +80,12 @@ class FinalizeContextView(MethodView):
     @inject
     def __init__(self,
                  profile_service: ProfileService,
-                 auth_service: AuthService,
                  query_service: QueryService,
                  prompt_service: PromptService,
                  branding_service: BrandingService,
                  onboarding_service: OnboardingService,
                  jwt_service: JWTService,
+                 i18n_service: I18nService
                  ):
         self.profile_service = profile_service
         self.jwt_service = jwt_service
@@ -87,31 +93,34 @@ class FinalizeContextView(MethodView):
         self.prompt_service = prompt_service
         self.branding_service = branding_service
         self.onboarding_service = onboarding_service
+        self.i18n_service = i18n_service
 
     def get(self, company_short_name: str, token: str = None):
-        session_info = self.profile_service.get_current_session_info()
-        if session_info:
-            # session exists, internal user
-            user_identifier = session_info.get('user_identifier')
-            token = ''
-        elif token:
-            # user identified by api-key
-            payload = self.jwt_service.validate_chat_jwt(token)
-            if not payload:
-                logging.warning("Fallo crítico: No se pudo leer el auth token.")
-                return redirect(url_for('index', company_short_name=company_short_name))
-
-            user_identifier = payload.get('user_identifier')
-        else:
-            logging.warning("Fallo crítico: missing session information or auth token")
-            return redirect(url_for('index', company_short_name=company_short_name))
-
-        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
-        branding_data = self.branding_service.get_company_branding(company)
-
         try:
+            session_info = self.profile_service.get_current_session_info()
+            if session_info:
+                # session exists, internal user
+                user_identifier = session_info.get('user_identifier')
+                token = ''
+            elif token:
+                # user identified by api-key
+                payload = self.jwt_service.validate_chat_jwt(token)
+                if not payload:
+                    logging.warning("Fallo crítico: No se pudo leer el auth token.")
+                    return redirect(url_for('home', company_short_name=company_short_name))
+
+                user_identifier = payload.get('user_identifier')
+            else:
+                logging.warning("Fallo crítico: missing session information or auth token")
+                return redirect(url_for('home', company_short_name=company_short_name))
+
+            company = self.profile_service.get_company_by_short_name(company_short_name)
+            if not company:
+                return render_template('error.html',
+                            company_short_name=company_short_name,
+                            message="Empresa no encontrada"), 404
+            branding_data = self.branding_service.get_company_branding(company)
+
             # 2. Finalize the context rebuild (the heavy task).
             self.query_service.finalize_context_rebuild(
                 company_short_name=company_short_name,
@@ -122,6 +131,9 @@ class FinalizeContextView(MethodView):
             prompts = self.prompt_service.get_user_prompts(company_short_name)
             onboarding_cards = self.onboarding_service.get_onboarding_cards(company)
 
+            # 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,
@@ -129,12 +141,12 @@ class FinalizeContextView(MethodView):
                 branding=branding_data,
                 prompts=prompts,
                 onboarding_cards=onboarding_cards,
+                js_translations=js_translations,
                 redeem_token=token
             )
 
         except Exception as e:
             return render_template("error.html",
-                                   company=company,
                                    company_short_name=company_short_name,
                                    branding=branding_data,
                                    message=f"An unexpected error occurred during context loading: {str(e)}"), 500

iatoolkit/views/logout_api_view.py

@@ -9,7 +9,7 @@ from injector import inject
 from iatoolkit.services.auth_service import AuthService
 from iatoolkit.services.profile_service import ProfileService
 from iatoolkit.common.session_manager import SessionManager
-
+import logging
 
 class LogoutApiView(MethodView):
     @inject
@@ -20,26 +20,30 @@ class LogoutApiView(MethodView):
         self.auth_service = auth_service
 
     def get(self, company_short_name: str = None):
-        # 1. Get the authenticated user's
-        auth_result = self.auth_service.verify(anonymous=True)
-        if not auth_result.get("success"):
-            return jsonify(auth_result), auth_result.get("status_code", 401)
-
-        company = self.profile_service.get_company_by_short_name(company_short_name)
-        if not company:
-            return jsonify({"error": "Empresa no encontrada"}), 404
-
-        # get URL for redirection
-        url_for_redirect = company.parameters.get('external_urls', {}).get('logout_url')
-        if not url_for_redirect:
-            url_for_redirect = url_for('home', company_short_name=company_short_name)
-
-        # clear de session cookie
-        SessionManager.clear()
-
-        return {
-            'status': 'success',
-            'url': url_for_redirect,
-        }
+        try:
+            # 1. Get the authenticated user's
+            auth_result = self.auth_service.verify(anonymous=True)
+            if not auth_result.get("success"):
+                return jsonify(auth_result), auth_result.get("status_code", 401)
+
+            company = self.profile_service.get_company_by_short_name(company_short_name)
+            if not company:
+                return jsonify({"error": "company not found."}), 404
+
+            # get URL for redirection
+            url_for_redirect = company.parameters.get('external_urls', {}).get('logout_url')
+            if not url_for_redirect:
+                url_for_redirect = url_for('home', company_short_name=company_short_name)
+
+            # clear de session cookie
+            SessionManager.clear()
+
+            return {
+                'status': 'success',
+                'url': url_for_redirect,
+            }, 200
+        except Exception as e:
+            logging.exception(f"Unexpected error: {e}")
+            return {'status': 'error'}, 500
 
 

iatoolkit/views/profile_api_view.py

@@ -0,0 +1,46 @@
+# iatoolkit/views/profile_api_view.py
+from flask import request, jsonify
+from flask.views import MethodView
+from injector import inject
+from iatoolkit.services.auth_service import AuthService
+from iatoolkit.services.profile_service import ProfileService
+
+
+class UserLanguageApiView(MethodView):
+    """
+    API endpoint for managing user language preferences.
+    """
+
+    @inject
+    def __init__(self,
+                 auth_service: AuthService,
+                 profile_service: ProfileService):
+        self.auth_service = auth_service
+        self.profile_service = profile_service
+
+    def post(self):
+        """
+        Handles POST requests to update the user's preferred language.
+        Expects a JSON body with a 'language' key, e.g., {"language": "en"}.
+        """
+        # 1. Authenticate the user from the current session.
+        auth_result = self.auth_service.verify()
+        if not auth_result.get("success"):
+            return jsonify(auth_result), auth_result.get("status_code")
+
+        user_identifier = auth_result.get('user_identifier')
+
+        # 2. Validate request body
+        data = request.get_json()
+        if not data or 'language' not in data:
+            return jsonify({"error_message": "Missing 'language' field in request body"}), 400
+
+        new_lang = data.get('language')
+
+        # 3. Call the service to perform the update
+        update_result = self.profile_service.update_user_language(user_identifier, new_lang)
+
+        if not update_result.get('success'):
+            return jsonify(update_result), 400
+
+        return jsonify({"message": "Language preference updated successfully"}), 200

iatoolkit/views/prompt_api_view.py

@@ -20,12 +20,12 @@ class PromptApiView(MethodView):
         self.prompt_service = prompt_service
 
     def get(self, company_short_name):
-        # get access credentials
-        auth_result = self.auth_service.verify(anonymous=True)
-        if not auth_result.get("success"):
-            return jsonify(auth_result), auth_result.get('status_code')
-
         try:
+            # get access credentials
+            auth_result = self.auth_service.verify(anonymous=True)
+            if not auth_result.get("success"):
+                return jsonify(auth_result), auth_result.get('status_code')
+
             response = self.prompt_service.get_user_prompts(company_short_name)
             if "error" in response:
                 return {'error_message': response["error"]}, 402
@@ -33,5 +33,5 @@ class PromptApiView(MethodView):
             return response, 200
         except Exception as e:
             logging.exception(
-                f"Error inesperado al obtener el historial de consultas para company {company_short_name}: {e}")
+                f"unexpected error getting company prompts: {e}")
             return jsonify({"error_message": str(e)}), 500

iatoolkit/views/signup_view.py

@@ -4,9 +4,10 @@
 # IAToolkit is open source software.
 
 from flask.views import MethodView
-from flask import render_template, request, url_for, session, redirect
+from flask import render_template, request, url_for, redirect, flash
 from iatoolkit.services.profile_service import ProfileService
-from iatoolkit.services.branding_service import BrandingService # 1. Importar BrandingService
+from iatoolkit.services.branding_service import BrandingService
+from iatoolkit.services.i18n_service import I18nService
 from injector import inject
 from itsdangerous import URLSafeTimedSerializer
 import os
@@ -15,9 +16,12 @@ import os
 class SignupView(MethodView):
     @inject
     def __init__(self, profile_service: ProfileService,
-                 branding_service: BrandingService):
+                 branding_service: BrandingService,
+                 i18n_service: I18nService):
         self.profile_service = profile_service
         self.branding_service = branding_service # 3. Guardar la instancia
+        self.i18n_service = i18n_service
+
         self.serializer = URLSafeTimedSerializer(os.getenv("USER_VERIF_KEY"))
 
 
@@ -25,23 +29,24 @@ class SignupView(MethodView):
         # get company info
         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('error.html',
+                                   message=self.i18n_service.t('errors.templates.company_not_found')), 404
 
-        # Obtener los datos de branding
         branding_data = self.branding_service.get_company_branding(company)
-
         return render_template('signup.html',
                                company=company,
                                company_short_name=company_short_name,
                                branding=branding_data)
 
     def post(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="Empresa no encontrada"), 404
-
         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)
+
             first_name = request.form.get('first_name')
             last_name = request.form.get('last_name')
             email = request.form.get('email')
@@ -62,7 +67,7 @@ class SignupView(MethodView):
                 verification_url=verification_url)
 
             if "error" in response:
-                branding_data = self.branding_service.get_company_branding(company)
+                flash(response["error"], 'error')
                 return render_template(
                     'signup.html',
                     company=company,
@@ -74,19 +79,16 @@ class SignupView(MethodView):
                         "email": email,
                         "password": password,
                         "confirm_password": confirm_password
-                    },
-                    alert_message=response["error"]), 400
-
-            # Guardamos el mensaje de éxito en la sesión
-            session['alert_message'] = response["message"]
-            session['alert_icon'] = 'success'
+                    }), 400
 
-            # Redirigimos al usuario a la página de login
+            flash(response["message"], 'success')
             return redirect(url_for('home', company_short_name=company_short_name))
 
         except Exception as e:
-            return render_template("error.html",
-                                   company=company,
-                                   company_short_name=company_short_name,
-                                   message=f"Ha ocurrido un error inesperado: {str(e)}"), 500
-
+            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/user_feedback_api_view.py

@@ -20,26 +20,28 @@ class UserFeedbackApiView(MethodView):
         self.user_feedback_service = user_feedback_service
 
     def post(self, company_short_name):
-        # get access credentials
-        auth_result = self.auth_service.verify()
-        if not auth_result.get("success"):
-            return jsonify(auth_result), auth_result.get("status_code")
+        try:
+            # get access credentials
+            auth_result = self.auth_service.verify()
+            if not auth_result.get("success"):
+                return jsonify(auth_result), auth_result.get("status_code")
 
-        user_identifier = auth_result.get('user_identifier')
+            user_identifier = auth_result.get('user_identifier')
 
-        data = request.get_json()
-        if not data:
-            return jsonify({"error_message": "Cuerpo de la solicitud JSON inválido o faltante"}), 402
+            data = request.get_json()
+            if not data:
+                return jsonify({"error_message": "invalid json body"}), 402
 
-        message = data.get("message")
-        if not message:
-            return jsonify({"error_message": "Falta el mensaje de feedback"}), 400
-        
-        rating = data.get("rating")
-        if not rating:
-            return jsonify({"error_message": "Falta la calificación"}), 400
+            # these validations are performed also in the frontend
+            # the are localized in the front
+            message = data.get("message")
+            if not message:
+                return jsonify({"error_message": "missing feedback message"}), 400
+
+            rating = data.get("rating")
+            if not rating:
+                return jsonify({"error_message": "missing rating"}), 400
 
-        try:
             response = self.user_feedback_service.new_feedback(
                 company_short_name=company_short_name,
                 message=message,
@@ -53,7 +55,6 @@ class UserFeedbackApiView(MethodView):
             return response, 200
         except Exception as e:
             logging.exception(
-                f"Error inesperado al procesar feedback para company {company_short_name}: {e}")
-
+                f"unexpected error processing feedback for {company_short_name}: {e}")
             return jsonify({"error_message": str(e)}), 500