fix combined download

lib/submissions/generate_audit_trail.rb

@@ -198,7 +198,7 @@ module Submissions
 
       composer.draw_box(divider)
 
-      documents_data = Submitters.select_attachments_for_download(last_submitter).map do |document|
+      documents_data = select_attachments(last_submitter).map do |document|
         original_documents = submission.schema_documents.select do |e|
           e.uuid == (document.metadata['original_uuid'] || document.uuid)
         end.presence
@@ -479,6 +479,16 @@ module Submissions
       'Signed with DocuSeal.com'
     end
 
+    def select_attachments(submitter)
+      original_documents = submitter.submission.schema_documents.preload(:blob)
+      is_more_than_two_images = original_documents.count(&:image?) > 1
+
+      submitter.documents.preload(:blob).reject do |attachment|
+        is_more_than_two_images &&
+          original_documents.find { |a| a.uuid == (attachment.metadata['original_uuid'] || attachment.uuid) }&.image?
+      end
+    end
+
     def maybe_add_background(_canvas, _submission, _page_size); end
 
     def show_verify?(submission)

lib/submitters.rb

@@ -94,8 +94,9 @@ module Submitters
   def select_attachments_for_download(submitter)
     if AccountConfig.exists?(account_id: submitter.submission.account_id,
                              key: AccountConfig::COMBINE_PDF_RESULT_KEY,
-                             value: true) && submitter.submission.combined_document_attachment
-      return [submitter.submission.combined_document_attachment]
+                             value: true) &&
+       submitter.submission.submitters.all?(&:completed_at?)
+      return [submitter.submission.combined_document_attachment || Submissions::EnsureCombinedGenerated.call(submitter)]
     end
 
     original_documents = submitter.submission.schema_documents.preload(:blob)