mirror of
https://github.com/we-promise/sure.git
synced 2026-04-20 20:44:08 +00:00
Generalize from PDF import to just files
This commit is contained in:
@@ -38,11 +38,17 @@ class ImportsController < ApplicationController
|
||||
|
||||
def new
|
||||
@pending_import = Current.family.imports.ordered.pending.first
|
||||
@document_upload_extensions = document_upload_supported_extensions
|
||||
end
|
||||
|
||||
def create
|
||||
file = import_params[:import_file]
|
||||
|
||||
if file.present? && document_upload_request?
|
||||
create_document_import(file)
|
||||
return
|
||||
end
|
||||
|
||||
# Handle PDF file uploads - process with AI
|
||||
if file.present? && Import::ALLOWED_PDF_MIME_TYPES.include?(file.content_type)
|
||||
unless valid_pdf_file?(file)
|
||||
@@ -137,6 +143,60 @@ class ImportsController < ApplicationController
|
||||
redirect_to import_path(pdf_import), notice: t("imports.create.pdf_processing")
|
||||
end
|
||||
|
||||
def create_document_import(file)
|
||||
adapter = VectorStore.adapter
|
||||
unless adapter
|
||||
redirect_to new_import_path, alert: t("imports.create.document_provider_not_configured")
|
||||
return
|
||||
end
|
||||
|
||||
if file.size > Import::MAX_PDF_SIZE
|
||||
redirect_to new_import_path, alert: t("imports.create.document_too_large", max_size: Import::MAX_PDF_SIZE / 1.megabyte)
|
||||
return
|
||||
end
|
||||
|
||||
filename = file.original_filename.to_s
|
||||
ext = File.extname(filename).downcase
|
||||
supported_extensions = adapter.supported_extensions.map(&:downcase)
|
||||
|
||||
unless supported_extensions.include?(ext)
|
||||
redirect_to new_import_path, alert: t("imports.create.invalid_document_file_type")
|
||||
return
|
||||
end
|
||||
|
||||
if ext == ".pdf"
|
||||
unless valid_pdf_file?(file)
|
||||
redirect_to new_import_path, alert: t("imports.create.invalid_pdf")
|
||||
return
|
||||
end
|
||||
|
||||
create_pdf_import(file)
|
||||
return
|
||||
end
|
||||
|
||||
family_document = Current.family.upload_document(
|
||||
file_content: file.read,
|
||||
filename: filename
|
||||
)
|
||||
|
||||
if family_document
|
||||
redirect_to new_import_path, notice: t("imports.create.document_uploaded")
|
||||
else
|
||||
redirect_to new_import_path, alert: t("imports.create.document_upload_failed")
|
||||
end
|
||||
end
|
||||
|
||||
def document_upload_supported_extensions
|
||||
adapter = VectorStore.adapter
|
||||
return [] unless adapter
|
||||
|
||||
adapter.supported_extensions.map(&:downcase).uniq.sort
|
||||
end
|
||||
|
||||
def document_upload_request?
|
||||
params.dig(:import, :type) == "DocumentImport"
|
||||
end
|
||||
|
||||
def valid_pdf_file?(file)
|
||||
header = file.read(5)
|
||||
file.rewind
|
||||
|
||||
@@ -141,10 +141,10 @@
|
||||
</li>
|
||||
<% end %>
|
||||
|
||||
<% if (params[:type].nil? || params[:type] == "PdfImport") && Provider::Registry.get_provider(:openai)&.supports_pdf_processing? %>
|
||||
<% if (params[:type].nil? || params[:type].in?(%w[DocumentImport PdfImport])) && @document_upload_extensions.any? %>
|
||||
<li>
|
||||
<%= styled_form_with url: imports_path, scope: :import, multipart: true, class: "w-full" do |form| %>
|
||||
<%= form.hidden_field :type, value: "PdfImport" %>
|
||||
<%= form.hidden_field :type, value: "DocumentImport" %>
|
||||
<label class="flex items-center justify-between p-4 group cursor-pointer w-full">
|
||||
<div class="flex items-center gap-2">
|
||||
<div class="bg-red-500/5 rounded-md w-8 h-8 flex items-center justify-center">
|
||||
@@ -154,15 +154,15 @@
|
||||
</div>
|
||||
<div class="text-left">
|
||||
<span class="text-sm text-primary group-hover:text-secondary block">
|
||||
<%= t(".import_pdf") %>
|
||||
<%= t(".import_file") %>
|
||||
</span>
|
||||
<span class="text-xs text-secondary">
|
||||
<%= t(".import_pdf_description") %>
|
||||
<%= t(".import_file_description") %>
|
||||
</span>
|
||||
</div>
|
||||
</div>
|
||||
<%= icon("chevron-right") %>
|
||||
<%= form.file_field :import_file, accept: "application/pdf", class: "hidden", onchange: "this.form.submit()" %>
|
||||
<%= form.file_field :import_file, accept: @document_upload_extensions.join(","), class: "hidden", onchange: "this.form.submit()" %>
|
||||
</label>
|
||||
<% end %>
|
||||
|
||||
|
||||
Reference in New Issue
Block a user