Refactor CreditBalanceViewSet for improved readability and error handling. Update SiteDefinitionView to allow public access without authentication. Enhance Vite configuration for shared component paths and debugging logs. Remove inaccessible shared CSS imports in main.tsx.

backend/igny8_core/modules/billing/views.py

@@ -41,70 +41,48 @@ class CreditBalanceViewSet(viewsets.ViewSet):
     @action(detail=False, methods=['get'])
     def balance(self, request):
         """Get current credit balance and usage"""
-        try:
-            account = getattr(request, 'account', None)
-            if not account:
-                user = getattr(request, 'user', None)
-                if user and user.is_authenticated:
-                    # Try to get account from user - refresh from DB to ensure we have latest
-                    try:
-                        from igny8_core.auth.models import User as UserModel
-                        # Refresh user from DB to get account relationship
-                        user = UserModel.objects.select_related('account', 'account__plan').get(id=user.id)
-                        account = user.account
-                        # Also set it on request for future use
-                        request.account = account
-                    except (AttributeError, UserModel.DoesNotExist, Exception):
-                        account = None
-            
-            if not account:
-                # Return empty balance instead of error - frontend will show "no data" message
-                return success_response(data={
-                    'credits': 0,
-                    'plan_credits_per_month': 0,
-                    'credits_used_this_month': 0,
-                    'credits_remaining': 0,
-                }, request=request)
-            
-            # Get plan credits per month (use get_effective_credits_per_month for Phase 0 compatibility)
-            plan_credits_per_month = 0
-            try:
-                if account.plan:
-                    plan_credits_per_month = account.plan.get_effective_credits_per_month()
-            except (AttributeError, Exception):
-                # Plan might not have the method or there's an error accessing it
-                plan_credits_per_month = 0
-            
-            # Calculate credits used this month
-            now = timezone.now()
-            start_of_month = now.replace(day=1, hour=0, minute=0, second=0, microsecond=0)
-            credits_used_this_month = CreditUsageLog.objects.filter(
-                account=account,
-                created_at__gte=start_of_month
-            ).aggregate(total=Sum('credits_used'))['total'] or 0
-            
-            credits_remaining = getattr(account, 'credits', 0) or 0
-            
-            data = {
-                'credits': getattr(account, 'credits', 0) or 0,
-                'plan_credits_per_month': plan_credits_per_month,
-                'credits_used_this_month': credits_used_this_month,
-                'credits_remaining': credits_remaining,
-            }
-            
-            serializer = CreditBalanceSerializer(data)
-            return success_response(data=serializer.data, request=request)
-        except Exception as e:
-            import logging
-            logger = logging.getLogger(__name__)
-            logger.error(f"Error in credit balance endpoint: {str(e)}", exc_info=True)
-            # Return empty balance instead of error
+        account = getattr(request, 'account', None)
+        if not account:
+            user = getattr(request, 'user', None)
+            if user and user.is_authenticated:
+                from igny8_core.auth.models import User as UserModel
+                user = UserModel.objects.select_related('account', 'account__plan').get(id=user.id)
+                account = user.account
+                request.account = account
+        
+        if not account:
             return success_response(data={
                 'credits': 0,
                 'plan_credits_per_month': 0,
                 'credits_used_this_month': 0,
                 'credits_remaining': 0,
             }, request=request)
+        
+        # Get plan credits - plan is already associated
+        plan_credits_per_month = 0
+        if account.plan:
+            plan_credits_per_month = account.plan.get_effective_credits_per_month()
+        
+        # Calculate credits used this month
+        now = timezone.now()
+        start_of_month = now.replace(day=1, hour=0, minute=0, second=0, microsecond=0)
+        credits_used_this_month = CreditUsageLog.objects.filter(
+            account=account,
+            created_at__gte=start_of_month
+        ).aggregate(total=Sum('credits_used'))['total'] or 0
+        
+        credits = account.credits or 0
+        credits_remaining = credits
+        
+        data = {
+            'credits': credits,
+            'plan_credits_per_month': plan_credits_per_month,
+            'credits_used_this_month': credits_used_this_month,
+            'credits_remaining': credits_remaining,
+        }
+        
+        serializer = CreditBalanceSerializer(data)
+        return success_response(data=serializer.data, request=request)
 
 
 @extend_schema_view(