@@ -13,6 +13,7 @@ from ml_model.services.gemini import Gemini from ml_model.services.geminiimage import Geminiimage from ml_model.services.gptimage import Gptimage from ml_model.services.hailuo import Hailuo +from ml_model.services.hunyuan import Hunyuan from ml_model.services.sora import Sora from ml_model.services.speedance import Speedance from ml_model.services.granite import Granite @@ -0,0 +1,53 @@ +import time +from datetime import timedelta +from decimal import Decimal +from io import BytesIO + +import requests +from django.core.files import File + +from messages.models import Message +from ml_model.services.base import SimpleService +from ml_model.tasks import replicate_run + +from payments.exceptions.insufficient_balance import InsufficientBalance +from payments.selectors.payment_plan_selector import PaymentPlanSelector + + +class Hunyuan(SimpleService): + + PRICE = Decimal('1.575') + + _CALLBACK = 'tencent/hunyuan-video:6c9132aee14409cd6568d030453f1ba50f5f3412b844fe67f78a9eb62d55664f' + + def calculate_price(self, process_time: timedelta) -> Decimal: + price = self.PRICE * Decimal(process_time.total_seconds()) + return price.quantize(Decimal('0.1'), rounding='ROUND_UP') + + def save_results(self, content: str, t: timedelta, video: str, save: bool = True) -> list[Message]: + msg = Message( + content=content, + content_object=self.store, + elapsed_time=t, + file=File(BytesIO(requests.get(video).content), '.mp4'), + ) + if save: + return Message.objects.bulk_create([msg]) + return [msg] + + def make(self, input_message: Message, save: bool = True) -> list[Message]: + if ( + (balance := PaymentPlanSelector(self.store.user).get_current_balance()) + < (cost := Decimal('378')) + ): + raise InsufficientBalance(balance, cost) + callback_data = { + 'prompt': self.translate_prompt(input_message.content), + **input_message.info, + } + start_time = time.time() + video = replicate_run(self._CALLBACK, callback_data) + process_time = timedelta(seconds=(time.time() - start_time)) + self.handle_invoice(input_message.content_object.model, process_time=process_time) + msgs = self.save_results(input_message.content, process_time, video, save) + return msgs