Merge pull request #18 from OmarMWarraich/17-develop-an-image-gallery

feat: Image Gallery

app/controllers/generated_images_controller.rb

@@ -0,0 +1,13 @@
+class GeneratedImagesController < ApplicationController
+  before_action :authenticate_user!
+
+  def show
+    @generated_image = current_user.generated_images.find(params[:id])
+
+    @generated_image.broadcast_to_main_image
+
+    respond_to do |format|
+      format.html { head :ok }
+    end
+  end
+end

app/controllers/txt2_imgs_controller.rb

@@ -16,6 +16,8 @@ def index
 
     result = api_instance.get_sd_models_sdapi_v1_sd_models_get
     @models = result.map { |model| model.model_name }
+
+    @gallery = current_user.generated_images.order(created_at: :desc).limit(30)
   end
 
   def create

app/helpers/generated_images_helper.rb

@@ -0,0 +1,2 @@
+module GeneratedImagesHelper
+end

app/models/generated_image.rb

@@ -29,6 +29,12 @@ class GeneratedImage < ApplicationRecord
   after_create_commit { broadcast_created }
 
   def broadcast_created
+    broadcast_to_main_image
+
+    broadcast_to_gallery
+  end
+
+  def broadcast_to_main_image
     broadcast_update_to(
       "#{dom_id(user)}_main_image",
       partial: "generated_images/display_main_image",
@@ -40,6 +46,18 @@ def broadcast_created
     )
   end
 
+  def broadcast_to_gallery
+    broadcast_prepend_to(
+      "#{dom_id(user)}_gallery",
+      partial: "/txt2_imgs/gallery",
+      locals: {
+        scroll_to: true,
+        gallery: self
+      },
+      target: "main_gallery"
+    )
+  end
+
   def self.create_from_sd_render_job(render_settings, user_id, result)
     user = User.find(user_id)
     result.images.each do |src_image|

app/views/generated_images/show.html.erb

@@ -0,0 +1,2 @@
+<h1>GeneratedImages#show</h1>
+<p>Find me in app/views/generated_images/show.html.erb</p>

app/views/txt2_imgs/_gallery.html.erb

@@ -0,0 +1,9 @@
+<div id="<%= dom_id(gallery)%>" class="col-4 mb-1 p-0 position-relative">
+  <div class='row'>
+    <div class='col'>
+      <%= link_to generated_image_path(gallery), method: :get, data: { turbo_frame: 'image_maker' } do %>
+        <%= image_tag gallery.sketch, class: 'img-thumbnail', alt: gallery.info, title: gallery.prompt %>
+      <% end %>
+    </div>
+  </div>
+</div>

app/views/txt2_imgs/index.html.erb

@@ -17,7 +17,10 @@
     </div>
     <div class='col-md-2'>
       <h5 class='border-bottom pb-2 mb-2'><i class="fas fa-duotone fa-images"></i> Gallery</h5>
-
+      <%= turbo_stream_from "#{dom_id(current_user)}_gallery" %>
+      <div id="main_gallery" class="row">
+        <%= render partial: '/txt2_imgs/gallery', collection: @gallery %>
+      </div>
     </div>
   </div>
 </div>

config/routes.rb

@@ -1,4 +1,5 @@
 Rails.application.routes.draw do
+  # get 'generated_images/show'
   # get 'txt2_imgs/index'
   # get 'sessions/new'
   # get 'registrations/new'
@@ -10,6 +11,7 @@
   resource :password_reset # where password reset routes are defined
   resource :password_change # where password change routes are defined
   resource :txt2_imgs, only: [:index, :create] # where the main app routes are defined
+  resources :generated_images, only: [:show] # where the generated images routes are defined
 
   # Reveal health status on /up that returns 200 if the app boots with no exceptions, otherwise 500.
   # Can be used by load balancers and uptime monitors to verify that the app is live.

test/controllers/generated_images_controller_test.rb

@@ -0,0 +1,8 @@
+require "test_helper"
+
+class GeneratedImagesControllerTest < ActionDispatch::IntegrationTest
+  test "should get show" do
+    get generated_images_show_url
+    assert_response :success
+  end
+end