@@ -31,5 +31,6 @@ from ml_model.services.recraft import Recraft
 from ml_model.services.sdxlemoji import Sdxlemoji
 from ml_model.services.stablediffusion import Stablediffusion
 from ml_model.services.upscaleai import Upscaleai
+from ml_model.services.veo import Veo
 from ml_model.services.vicuna import Vicuna
 from ml_model.services.whisper import Whisper
@@ -0,0 +1,51 @@
+import base64
+import time
+from datetime import timedelta
+from decimal import Decimal
+from io import BytesIO
+
+import filetype
+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
+
+
+class Veo(SimpleService):
+    TOKENS_COST = {
+        'veo-3': Decimal('640'),
+        'veo-3-fast': Decimal('240')
+    }
+
+    def calculate_price(self, version: str) -> Decimal:
+        return self.TOKENS_COST[version]
+
+    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]:
+        version = input_message.info.pop('version', 'veo-3-fast')
+        callback_data = dict({'prompt': input_message.content, **input_message.info})
+        if input_message.file:
+            kind = filetype.guess(input_message.file.read(20))
+            mime = kind.mime if kind else 'application/octet-stream'
+            input_message.file.seek(0)
+            image = f'data:{mime};base64,{base64.b64encode(input_message.file.read()).decode("utf-8")}'
+            input_message.file.close()
+            callback_data.update({'image': image})
+        start_time = time.time()
+        video = replicate_run(f'google/{version}', callback_data)
+        process_time = timedelta(seconds=(time.time() - start_time))
+        self.handle_invoice(input_message.content_object.model, version)
+        msgs = self.save_results(input_message.content, process_time, video, save)
+        return msgs