FEATURE: Handle secure uploads in image caption (#476)

app/controllers/discourse_ai/ai_helper/assistant_controller.rb

@@ -109,11 +109,16 @@ module DiscourseAi
         image_url = params[:image_url]
         raise Discourse::InvalidParameters.new(:image_url) if !image_url
 
-        image = Upload.where(url: params[:image_url])
+        image = Upload.find_by(sha1: Upload.sha1_from_long_url(image_url))
+
+        if image&.secure?
+          url = Upload.signed_url_from_secure_uploads_url(image_url)
+        else
+          url = UrlHelper.absolute(image_url)
+        end
 
         hijack do
-          caption =
+          caption = DiscourseAi::AiHelper::Assistant.new.generate_image_caption(url, current_user)
-            DiscourseAi::AiHelper::Assistant.new.generate_image_caption(image_url, current_user)
           render json: { caption: caption }, status: 200
         end
       rescue DiscourseAi::Completions::Endpoints::Base::CompletionFailed, Net::HTTPBadResponse

assets/javascripts/initializers/ai-image-caption.js

@@ -10,7 +10,11 @@ export default apiInitializer("1.25.0", (api) => {
     class: "generate-caption",
   };
   const imageCaptionPopup = api.container.lookup("service:imageCaptionPopup");
+  const settings = api.container.lookup("service:site-settings");
 
+  if (!settings.ai_helper_enabled_features.includes("image_caption")) {
+    return;
+  }
   api.addComposerImageWrapperButton(
     buttonAttrs.label,
     buttonAttrs.class,

config/settings.yml

@@ -224,6 +224,7 @@ discourse_ai:
     choices:
       - "suggestions"
       - "context_menu"
+      - "image_caption"
   ai_helper_image_caption_model:
     default: "llava"
     type: enum