Removed size and model being passed through

Author: D-K-P

product-image-generator/app/ProductImageGenerator.tsx

@@ -345,7 +345,7 @@ export default function ProductImageGenerator({
             {/* <p className="text-sm text-muted-foreground">
               Powered by{" "}
               <span className="font-medium text-primary">Trigger.dev</span> and{" "}
-              <span className="font-medium text-primary">Flux AI</span>
+              
             </p> */}
           </div>
         </div>

product-image-generator/app/actions.ts

@@ -41,8 +41,6 @@ export async function triggerGenerationTask(payload: {
   baseImageUrl: string;
   productAnalysis: ProductAnalysis;
   customPrompt?: string;
-  model?: "flux";
-  size?: "1024x1792";
 }) {
   try {
     const handle = await tasks.trigger<typeof generateAndUploadImage>(

product-image-generator/app/components/GeneratedCard.tsx

@@ -17,13 +17,6 @@ import { triggerGenerationTask } from "../actions";
 import type { generateAndUploadImage } from "../../src/trigger/generate-and-upload-image";
 import type { ProductAnalysis } from "../types/trigger";
 
-type TaskRun = {
-  id?: string;
-  status?: string;
-  output?: unknown;
-  metadata?: unknown;
-};
-
 interface GeneratedCardProps {
   baseImageUrl: string | null;
   productAnalysis: ProductAnalysis | null;
@@ -78,7 +71,6 @@ export default function GeneratedCard({
         promptStyle: promptId,
         baseImageUrl,
         productAnalysis,
-        size: "1024x1792",
       });
 
       if (result.success) {
@@ -186,7 +178,7 @@ export default function GeneratedCard({
             <div className="absolute inset-0 bg-black/50 flex items-center justify-center backdrop-blur-sm">
               <div className="flex flex-col items-center gap-2">
                 <div className="w-8 h-8 rounded-full flex items-center justify-center mb-4 transition-colors ">
-                  <div className="animate-spin rounded-full h-6 w-6">
+                  <div className="animate-spin h-6 w-6">
                     <Loader2 className="h-6 w-6 text-gray-500" />
                   </div>
                 </div>

product-image-generator/src/trigger/generate-and-upload-image.ts

@@ -15,6 +15,8 @@ const s3Client = new S3Client({
   },
 });
 
+const aiModel = "black-forest-labs/flux-dev";
+
 export const generateAndUploadImage = task({
   id: "generate-and-upload-image",
   maxDuration: 600, // 10 minutes max
@@ -47,7 +49,7 @@ export const generateAndUploadImage = task({
       baseImageUrl,
       productAnalysis,
       customPrompt,
-      model = "flux",
+      model = aiModel,
       size = "1024x1792",
       strength = 0.2,
       guidance = 7, // From your settings
@@ -127,7 +129,7 @@ export const generateAndUploadImage = task({
 
       // Use Flux with structured prompt
       const generateParams: any = {
-        model: replicate.image("black-forest-labs/flux-dev"),
+        model: replicate.image(aiModel),
         prompt: enhancedPrompt,
         image: baseImageUrl, // Reference image for img2img
         width: parseInt(size.split("x")[0]),