some changes

backend/igny8_core/modules/writer/views.py

@@ -551,12 +551,24 @@ class ImagesViewSet(SiteSectorModelViewSet):
                     account_id=account_id
                 )
                 if result.get('success'):
-                    return Response({
+                    # Include queued prompts in response for TEST MODE
+                    response_data = {
                         'success': True,
-                        'images_generated': result.get('count', 0),
+                        'images_generated': result.get('images_generated', 0),
                         'images_failed': result.get('images_failed', 0),
+                        'count': result.get('count', 0),
+                        'total_images': result.get('total_images', 0),
                         'message': 'Images generated successfully'
-                    }, status=status.HTTP_200_OK)
+                    }
+                    # Add test mode data if available
+                    if result.get('queued_prompts'):
+                        response_data['queued_prompts'] = result.get('queued_prompts')
+                        response_data['test_mode'] = result.get('test_mode', False)
+                        response_data['provider'] = result.get('provider')
+                        response_data['model'] = result.get('model')
+                    
+                    logger.info(f"[generate_images] Response: {response_data}")
+                    return Response(response_data, status=status.HTTP_200_OK)
                 else:
                     return Response({
                         'error': result.get('error', 'Image generation failed'),