Class: GeneratedImagesController

Inherits:

CrmController

Object
ActionController::Base
ApplicationController
CrmController
GeneratedImagesController

show all

Defined in:: app/controllers/generated_images_controller.rb

Constant Summary

Constants included from Www::SeoHelper

Constants included from IconHelper

IconHelper::CUSTOM_ICON_MAP, IconHelper::CUSTOM_SVG_DIR, IconHelper::DEFAULT_FAMILY

Instance Method Summary collapse

#batch_discard ⇒ Object
POST /image_generations/batch_discard Mass-discard selected pending generated images.
#batch_import ⇒ Object
POST /image_generations/batch_import Mass-import selected pending generated images into the library with auto-derived metadata.
#create ⇒ Object
POST /image_generations.
#destroy ⇒ Object
DELETE /image_generations/:id Discard the generated image (deletes from ImageKit, marks rejected).
#index ⇒ Object
GET /image_generations.
#new ⇒ Object
GET /image_generations/new Optional: ?source_image_id= pre-populates the form with that image as a reference.
#refine ⇒ Object
POST /image_generations/:id/refine Re-enqueue generation with user feedback; feeds the previous result back as a reference.
#regenerate ⇒ Object
POST /image_generations/:id/regenerate Re-run the original generation with the same settings — no previous result used as a reference.
#show ⇒ Object
GET /image_generations/:id.
#update ⇒ Object
PATCH /image_generations/:id Import the generated image into the library.

Methods inherited from CrmController

#access_denied, #context_id, #context_object, #crm_home_path, #current_ability, #default_url_options, #download_temp, #get_tempfile_path_for_download, #initialize_crm_lazy_chunks, #record_not_found, #redirect_to_job_or_fallback, #render_edit_action, #set_context, #set_download_path, #stash_file_for_temp_download

Methods inherited from ApplicationController

#account_impersonated?, #add_to_flash, #append_token, #bypass_forgery_protection?, #chat_enabled?, #cloudflare_cleared?, #default_catalog, #default_url_options, #enable_turbo_frames, #find_publication, #fix_invalid_accept_header, #init_js_utils, #is_globals_call?, #layout_by_resource, #locale_store, #redirect_to, #require_employee_for_crm, #set_base_host, #set_real_ip, #set_report_errors_for, #should_render_layout?, #stamp_impersonation_context, #warmlyyours_canada_ip?, #warmlyyours_ip?, #y

Methods included from Controllers::ReturnPathHandling

#check_for_return_path, #redirect_to_return_path_or_default

Methods included from Controllers::AnalyticsEvents

#consume_queued_analytics_events, #track_event

Methods included from Controllers::DeviceDetection

#device_detector, #is_ie?

Methods included from Controllers::SubdomainDetection

#is_crm_request?, #is_www_request?, #json_request?

Methods included from Controllers::TrackingDetection

#bot_request?, #gdpr_country?, #gdpr_country_data, #prevent_bots, #set_tracking_cookie, #track_visitor?

Methods included from Controllers::AcceleratedFileSending

#send_file_accelerated, #send_upload_accelerated

Methods included from Controllers::ErrorRendering

#excp_string, #mail_to_for_error_reporting, #render_400, #render_404, #render_406, #render_410, #render_500, #render_invalid_authenticity_token, #render_ip_spoof_error, #safe_referer_or_fallback

Methods included from Controllers::TurnstileVerification

#load_turnstile_script_tag, #turnstile_lazy_widget, #turnstile_script_tag, #turnstile_widget, #validate_turnstile!

Methods included from Controllers::CloudflareCaching

edge_cached, #edge_cached_action?, #reset_cloudflare_cache, #set_cloudflare_cache, #skip_session

Methods included from Controllers::Webpackable

#preload_webpack_fonts, #webpack_css_include, #webpack_css_url, #webpack_js_include, #wpd_is_running?

Methods included from Controllers::Localizable

#cloudflare_country_locale, #determine_request_locale, #geocoder_locale, #guest_user_locale_check, #locale_optional_www_auth_path?, #param_locale, #set_locale, #set_request_locale, #skip_localization?, #warmlyyours_ip_locale

Methods included from Controllers::Authenticable

#access_denied, #authenticate_account, #authenticate_account!, #authenticate_account_from_login_token!, #authenticate_account_from_token!, #check_is_a_manager, #check_is_a_sales_manager, #check_is_an_admin, #check_is_an_employee, #check_party, #clear_mismatched_guest_user, #create_guest_user, #credentials?, #current_or_guest_user, #current_or_guest_user_id_read_only, #current_user, #devise_mapping, #fully_logged_in?, #generate_bot_id, #guest_user, #identifiable?, #init_current_user, #initialize_guest, #load_context_user, #logging_in, #resource, #resource_name, #restrict_access_for_non_employees, #scrubbed_request_path, #user_object, #warn_on_session_guest_id_leak

Methods included from ApplicationHelper

#better_number_to_currency, #check_force_logout, #check_or_cross, #check_or_times, #error_messages, #general_disclaimer_on_product_installation_and_local_codes, #gridjs_from_html_table, #gridjs_table, #is_wy_ip, #line_break, #parent_layout, #pass_or_fail, #render_error_messages_list, #render_video_card, #resolved_auth_form_turbo_frame, #return_path_or, #safe_css_color, #set_return_path_if_present, #set_section_if_present, #tab_frame_id, #to_underscore, #track_page?, #turbo_section_wrapper, #turbo_tabs_request?, #url_on_same_domain_as_request, #widget_index_daily_focus_index_path, #working_hours?, #yes_or_no, #yes_or_no_highlighted, #yes_or_no_with_check_or_cross, #youtube_video

Methods included from UppyUploaderHelper

#file_uploader, #image_uploader, #large_file_uploader_s3, #lead_sketch_uploader, #rma_image_uploader, #rma_image_uploader_s3, #uppy_uploader, #video_uploader

Methods included from Www::ImagesHelper

#image_asset_tag, #image_asset_url

Methods included from Www::SeoHelper

#add_page_schema, #canada?, #company_social_links, #ensure_context_json, #json_ld_script_tag, #local_business_schema, #online_store_id, #online_store_schema, #page_main_entity, #page_main_entity_json, #render_auto_collection_page_schema, #render_collection_page_schema, #render_local_business_schema, #render_online_store_schema, #render_page_schemas, #render_page_video_schemas, #render_webpage_schema, #render_webpage_schema_with_collections, #usa?

Methods included from UrlsHelper

#catalog_breadcrumb_links, #catalog_link, #catalog_link_for_product_line, #catalog_link_for_sku, #cms_link, #delocalized_path, #path_to_sales_product_sku, #path_to_sales_product_sku_for_product_line, #path_to_sales_product_sku_for_product_line_slug, #product_line_from_catalog_link, #protocol_neutral_url, #sanitize_external_url, #valid_external_url?

Methods included from IconHelper

#account_nav_icon, #fa_icon, #star_rating_html

Instance Method Details

#batch_discard ⇒ `Object`

POST /image_generations/batch_discard
Mass-discard selected pending generated images.

# File 'app/controllers/generated_images_controller.rb', line 216

def batch_discard
  authorize!(:create, Image)

  ids = params[:ids].to_s.split(',').map(&:to_i).reject(&:zero?)
  if ids.empty?
    flash[:warning] = 'No images selected.'
    redirect_to generated_images_path and return
  end

  discarded = 0
  GeneratedImage.pending.where(id: ids).find_each do |gen|
    gen.delete_ik_asset!
    gen.update!(status: 'rejected')
    discarded += 1
  end

  flash[:info] = "Discarded #{discarded} image(s)."
  redirect_to generated_images_path(q: params[:q])
end

#batch_import ⇒ `Object`

POST /image_generations/batch_import
Mass-import selected pending generated images into the library with auto-derived metadata.

# File 'app/controllers/generated_images_controller.rb', line 180

def batch_import
  authorize!(:create, Image)

  ids = params[:ids].to_s.split(',').map(&:to_i).reject(&:zero?)
  if ids.empty?
    flash[:warning] = 'No images selected.'
    redirect_to generated_images_path and return
  end

  imported_image_ids = []
  failed = 0

  GeneratedImage.pending.where(id: ids).find_each do |gen|
    meta      = auto_metadata_for(gen)
    new_image = promote_to_image(gen, **meta)

    if new_image
      gen.update!(status: 'imported')
      imported_image_ids << new_image.id
    else
      failed += 1
    end
  end

  if imported_image_ids.any?
    ImageFullAnalysisWorker.perform_bulk(imported_image_ids.map { |id| [id] })
  end

  flash[:success] = "Imported #{imported_image_ids.size} image(s) into the library." if imported_image_ids.any?
  flash[:error]   = "#{failed} image(s) failed to import." if failed > 0
  redirect_to generated_images_path(q: params[:q])
end

#create ⇒ `Object`

POST /image_generations

# File 'app/controllers/generated_images_controller.rb', line 35

def create
  authorize!(:create, Image)

  prompt = params[:prompt].to_s.strip

  if prompt.blank?
    flash[:error] = 'A prompt is required to generate an image.'
    redirect_back_or_to new_generated_image_path and return
  end

  unless ImageGeneration::Service.available?
    flash[:error] = 'No image generation provider is configured. Contact an administrator.'
    redirect_back_or_to new_generated_image_path and return
  end

  reference_ids       = Array(params[:reference_image_ids]).map(&:to_i).uniq.reject(&:zero?)
  source_image_id     = params[:source_image_id].presence&.to_i
  source_record_type  = params[:source_record_type].presence
  source_record_id    = params[:source_record_id].presence&.to_i
  return_to           = request.referer.presence || new_generated_image_path

  # Resolve externally uploaded reference images into fetchable S3 URLs
  upload_ids = Array(params[:extra_reference_upload_ids]).map(&:to_i).uniq.reject(&:zero?)
  extra_urls = Upload.where(id: upload_ids).filter_map { |u| u.presigned_url(expires_in: 1.hour) }

  jid = ImageGenerationWorker.perform_async(
    prompt:               prompt,
    model:                params[:model].presence || ImageGeneration::Service::DEFAULT_MODEL,
    aspect_ratio:         params[:aspect_ratio].presence || '1:1',
    image_size:           params[:image_size].presence || '1K',
    reference_image_ids:  reference_ids,
    extra_reference_urls: extra_urls,
    source_image_id:      source_image_id,
    source_record_type:   source_record_type,
    source_record_id:     source_record_id,
    created_by_id:        current_account.id,
    return_to:            return_to
  )

  Rails.logger.info "[GeneratedImagesController#create] Queued #{jid} " \
                    "(source: #{source_image_id}, refs: #{reference_ids.size}, uploads: #{extra_urls.size})"

  if jid
    redirect_to job_path(jid)
  else
    flash[:error] = 'Failed to queue image generation. Please try again.'
    redirect_back_or_to new_generated_image_path
  end
end

#destroy ⇒ `Object`

DELETE /image_generations/:id
Discard the generated image (deletes from ImageKit, marks rejected).

# File 'app/controllers/generated_images_controller.rb', line 165

def destroy
  authorize!(:create, Image)
  guard_pending_or_redirect and return

  @generated_image.delete_ik_asset!
  @generated_image.update!(status: 'rejected')
  flash[:info] = 'Generated image discarded.'

  source = @generated_image.source_image
  redirect_to(source ? new_generated_image_path(source_image_id: source.id) : new_generated_image_path)
end

#index ⇒ `Object`

GET /image_generations

# File 'app/controllers/generated_images_controller.rb', line 7

def index
  authorize!(:create, Image)
  load_generated_images
end

#new ⇒ `Object`

GET /image_generations/new
Optional: ?source_image_id= pre-populates the form with that image as a reference.

# File 'app/controllers/generated_images_controller.rb', line 15

def new
  authorize!(:create, Image)
  @source_image              = Image.find_by(id: params[:source_image_id])
  @preselected_reference_ids = @source_image ? [@source_image.id] : []
  @brand_logo_images         = ImageGeneration::BrandStyleGuide.logo_images.limit(6)
  @brand_prompt_snippet      = ImageGeneration::BrandStyleGuide.prompt_snippet

  # Pre-fill form fields when linked from elsewhere (e.g. "Generate Avatar" on a review)
  @prefill_prompt        = params[:prompt].presence
  @prefill_aspect_ratio  = params[:aspect_ratio].presence
  @prefill_image_size    = params[:image_size].presence
  @prefill_model         = params[:model].presence
  @prefill_tags          = Array(params[:prefill_tags]).map(&:presence).compact
  @source_record_type    = params[:source_record_type].presence
  @source_record_id      = params[:source_record_id].presence&.to_i
  @source_record         = resolve_source_record(@source_record_type, @source_record_id)
end

#refine ⇒ `Object`

POST /image_generations/:id/refine
Re-enqueue generation with user feedback; feeds the previous result back as a reference.

# File 'app/controllers/generated_images_controller.rb', line 239

def refine
  authorize!(:create, Image)
  guard_pending_or_redirect and return

  refinement_notes = params[:refinement_notes].to_s.strip

  if refinement_notes.blank?
    flash[:error] = 'Please describe what you would like to change.'
    redirect_to generated_image_path(@generated_image) and return
  end

  unless ImageGeneration::Service.available?
    flash[:error] = 'No image generation provider is configured. Contact an administrator.'
    redirect_to generated_image_path(@generated_image) and return
  end

  refined_prompt = "#{@generated_image.prompt}\n\nPlease revise with these changes: #{refinement_notes}"

  jid = ImageGenerationWorker.perform_async(
    prompt:               refined_prompt,
    model:                @generated_image.model,
    aspect_ratio:         @generated_image.aspect_ratio,
    image_size:           @generated_image.image_size,
    reference_image_ids:  @generated_image.reference_image_ids,
    extra_reference_urls: [@generated_image.ik_url].compact,
    source_image_id:      @generated_image.source_image_id,
    created_by_id:        current_account.id,
    return_to:            generated_image_path(@generated_image)
  )

  Rails.logger.info "[GeneratedImagesController#refine] Queued #{jid} from GeneratedImage #{@generated_image.id}"

  if jid
    redirect_to job_path(jid)
  else
    flash[:error] = 'Failed to queue image generation. Please try again.'
    redirect_to generated_image_path(@generated_image)
  end
end

#regenerate ⇒ `Object`

POST /image_generations/:id/regenerate
Re-run the original generation with the same settings — no previous result used as a reference.
Useful when the current result is close but the user wants a fresh attempt.

# File 'app/controllers/generated_images_controller.rb', line 132

def regenerate
  authorize!(:create, Image)
  guard_pending_or_redirect and return

  unless ImageGeneration::Service.available?
    flash[:error] = 'No image generation provider is configured. Contact an administrator.'
    redirect_to generated_image_path(@generated_image) and return
  end

  jid = ImageGenerationWorker.perform_async(
    prompt:              @generated_image.prompt,
    model:               @generated_image.model,
    aspect_ratio:        @generated_image.aspect_ratio,
    image_size:          @generated_image.image_size,
    reference_image_ids: @generated_image.reference_image_ids,
    source_image_id:     @generated_image.source_image_id,
    created_by_id:       current_account.id,
    return_to:           generated_image_path(@generated_image)
  )

  Rails.logger.info "[GeneratedImagesController#regenerate] Queued #{jid} from GeneratedImage #{@generated_image.id}"

  if jid
    redirect_to job_path(jid)
  else
    flash[:error] = 'Failed to queue image generation. Please try again.'
    redirect_to generated_image_path(@generated_image)
  end
end

#show ⇒ `Object`

GET /image_generations/:id

# File 'app/controllers/generated_images_controller.rb', line 87

def show
  authorize!(:create, Image)
  guard_pending_or_redirect and return
  @source_image  = @generated_image.source_image
  @source_record = @generated_image.source_record
end

#update ⇒ `Object`

PATCH /image_generations/:id
Import the generated image into the library.

# File 'app/controllers/generated_images_controller.rb', line 97

def update
  authorize!(:create, Image)
  guard_pending_or_redirect and return

  # params[:tags] is an array from the multi-select; fall back to comma-string split for safety
  tags_raw = Array(params[:tags]).flat_map { |t| t.to_s.split(',') }.map(&:strip).reject(&:blank?)

  result = GeneratedImageImporter.call(
    @generated_image,
    title:    params[:title],
    tags:     tags_raw,
    category: params[:category],
    notes:    params[:notes]
  )

  if result.success?
    ImageFullAnalysisWorker.perform_async(result.image.id) if params[:run_analysis] == '1'

    flash[:success] = result.notice || 'Generated image imported into the library!'

    if params[:after_import] == 'upscale'
      redirect_to upscale_preview_image_path(result.image)
    else
      redirect_to image_path(result.image)
    end
  else
    flash[:error] = result.error || 'Failed to import image. Please try again.'
    redirect_to generated_image_path(@generated_image)
  end
end