add widget count

app/controllers/galleries_controller.rb

@@ -36,6 +36,7 @@ class GalleriesController < ApplicationController
   def widget
     albums = Album.filter_by_widget_categories
     params = OrbitHelper.params
+    counts = OrbitHelper.widget_data_count
     
     images = []
     total_images = 0
@@ -43,8 +44,8 @@ class GalleriesController < ApplicationController
       albums.each do |album|
         total_images = total_images + album.album_images.count
       end
-      total_images = total_images > 9 ? 9 : total_images
-      while images.count < total_images and images.count < AlbumImage.count
+      total_images = counts > 9 ? 9 : counts
+      while images.count < counts and images.count < AlbumImage.count
         albums.each do |album|
           img = album.album_images.sample
           if !images.include?(img) && img != nil