From 78c9cd38e076d08b00beb23caa290f44c229ec5f Mon Sep 17 00:00:00 2001
From: "IGNY8 VPS (Salman)" <vps@igny8.com>
Date: Tue, 13 Jan 2026 00:32:49 +0000
Subject: [PATCH] fixed site id issue for credits in wdigets for specific site

---
 backend/igny8_core/ai/tasks.py | 9 +++++++--
 1 file changed, 7 insertions(+), 2 deletions(-)

diff --git a/backend/igny8_core/ai/tasks.py b/backend/igny8_core/ai/tasks.py
index f50ba64d..cdd50140 100644
--- a/backend/igny8_core/ai/tasks.py
+++ b/backend/igny8_core/ai/tasks.py
@@ -765,6 +765,9 @@ def process_image_generation_queue(self, image_ids: list, account_id: int = None
                     cost_usd = result.get('cost', 0) or result.get('cost_usd', 0)  # generate_image returns 'cost'
                     if account:
                         try:
+                            # Get site_id from content if available
+                            site_id = content.site_id if content else None
+                            
                             credits_deducted = CreditService.deduct_credits_for_image(
                                 account=account,
                                 model_name=model,
@@ -773,6 +776,7 @@ def process_image_generation_queue(self, image_ids: list, account_id: int = None
                                 metadata={
                                     'image_id': image_id,
                                     'content_id': content_id,
+                                    'site_id': site_id,
                                     'provider': provider,
                                     'model': model,
                                     'image_type': image.image_type if image else 'unknown',
@@ -780,9 +784,10 @@ def process_image_generation_queue(self, image_ids: list, account_id: int = None
                                 },
                                 cost_usd=cost_usd,
                                 related_object_type='image',
-                                related_object_id=image_id
+                                related_object_id=image_id,
+                                site=site_id  # Pass site_id for credit logging
                             )
-                            logger.info(f"[process_image_generation_queue] Credits deducted for image {image_id}: account balance now {credits_deducted}")
+                            logger.info(f"[process_image_generation_queue] Credits deducted for image {image_id}: account balance now {credits_deducted}, site_id={site_id}")
                         except Exception as credit_error:
                             logger.error(f"[process_image_generation_queue] Failed to deduct credits for image {image_id}: {credit_error}")
                             # Don't fail the image generation if credit deduction fails