@@ -138,8 +138,9 @@ class Chatgpt(SimpleService):
             kind = filetype.guess(input_message.file.read(20))
             mime = kind.mime if kind else 'application/octet-stream'
             normalized_image = Image.open(image)
+            format = 'jpeg' if kind.extension == 'jpg' else kind.extension
             buf = BytesIO()
-            normalized_image.save(buf, format=kind.extension.upper())
+            normalized_image.save(buf, format=format)
             image_url = (
                 f'data:{mime},base64,{base64.b64encode(buf.getvalue()).decode("utf-8")}'
             )
@@ -198,6 +199,7 @@ class Chatgpt(SimpleService):
             )
             if image:
                 response = self.llm.invoke([llm_input])
+                print('gay')
                 chat_history.chat_memory.add_ai_message(response)
             elif file:
                 human_messages = []
@@ -139,26 +139,26 @@ class Flux(SimpleService):
         versions[0].slug: ModelPaymentRule(
             strategy=ModelPaymentRule.StrategyChoices.FIXED,
             interaction_type=ModelPaymentRule.InteractionTypeChoices.INPUT,
-            cost=0.33,
+            cost=0.3,
             coefficient=5.00,
         ),
         versions[1].slug: ModelPaymentRule(
             strategy=ModelPaymentRule.StrategyChoices.FIXED,
             interaction_type=ModelPaymentRule.InteractionTypeChoices.INPUT,
-            cost=4.4,
-            coefficient=5.00,
+            cost=4.00,
+            coefficient=2.00,
         ),
         versions[2].slug: ModelPaymentRule(
             strategy=ModelPaymentRule.StrategyChoices.FIXED,
             interaction_type=ModelPaymentRule.InteractionTypeChoices.INPUT,
-            cost=2.75,
-            coefficient=5.00,
+            cost=2.5,
+            coefficient=2.00,
         ),
         versions[3].slug: ModelPaymentRule(
             strategy=ModelPaymentRule.StrategyChoices.FIXED,
             interaction_type=ModelPaymentRule.InteractionTypeChoices.INPUT,
-            cost=6.6,
-            coefficient=5.00,
+            cost=6.00,
+            coefficient=2.00,
         ),
     }
 
@@ -134,7 +134,7 @@ class Stablediffusion(SimpleService):
         ),
     ]
 
-    TOKEN_PRICE = Decimal('11')
+    TOKEN_PRICE = Decimal('2')
 
     _API_KEY = settings.STABLE_DIFFUSION_API_KEY