@@ -2,7 +2,7 @@ import logging from django.conf import settings from django.core.mail import EmailMessage, send_mail -from django.utils.html import format_html +from django.utils.html import format_html, strip_tags from django.utils.translation import gettext_lazy as _ from authentication.models import BusinessAccount, BusinessUserHost @@ -27,7 +27,8 @@ class EmailService: try: send_mail( subject=subject, - message=message, + message=strip_tags(message), + html_message=message, from_email=settings.EMAIL_HOST_USER, recipient_list=(user_email,), auth_password=settings.EMAIL_HOST_PASSWORD, @@ -140,11 +140,13 @@ class Stablediffusion(SimpleService): def calculate_price(self, input_message: Message) -> Decimal: if ( - input_message.info.get('version') or input_message.info.get('engine', 'sd3') + input_message.info.get('version') + or input_message.info.get('engine', 'sd3-turbo') ) == 'sd3': return Decimal('13') elif ( - input_message.info.get('version') or input_message.info.get('engine', 'sd3') + input_message.info.get('version') + or input_message.info.get('engine', 'sd3-turbo') ) == 'sd3-turbo': return Decimal('8') steps = input_message.info.get('steps', 30) @@ -184,7 +186,7 @@ class Stablediffusion(SimpleService): } start_time = time.time() info = input_message.info.copy() - model_name = info.get('version') or info.get('engine', 'sd3') + model_name = info.get('version') or info.get('engine', 'sd3-turbo') formatter = { 'width': int( aspect_rations[input_message.info.get('aspect_ratio', '1:1')][0]