@@ -11,7 +11,7 @@ from django.core.files import File
 from replicate.exceptions import ModelError
 
 from messages.models import Message
-from ml_model.exceptions import RequestBlocked, GenerationException
+from ml_model.exceptions import RequestBlocked, GenerationException, FileNotProvided
 from ml_model.services.base import SimpleService
 from ml_model.tasks import replicate_run
 
@@ -45,14 +45,15 @@ class Kling(SimpleService):
         duration = input_message.info.get('duration', 5)
         if (balance := PaymentPlanSelector(self.store.user).get_current_balance()) < (cost := self.TOKENS_COST[mode] * duration):
             raise InsufficientBalance(balance, cost)
+        if not input_message.file:
+            raise FileNotProvided('Image')
         callback_data = dict({'prompt': self.translate_prompt(input_message.content), 'mode': mode, **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({'start_image': image})
+        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({'start_image': image})
         start_time = time.time()
         try:
             video = replicate_run('kwaivgi/kling-v2.1', callback_data)