Browse Source

feat(05-02): add image input to Replicate predictions

- Add image parameter to predictionInput when images array provided
- Log image input presence for debugging
- Uses 'image' param name (most common for img2img models)
handoff-20260429-1057
shrimbly 6 months ago
parent
commit
4477f6f822
  1. 9
      src/app/api/generate/route.ts

9
src/app/api/generate/route.ts

@ -236,6 +236,7 @@ async function generateWithReplicate(
console.log(`[API:${requestId}] Generating with Replicate...`);
console.log(`[API:${requestId}] - Model: ${input.model.id}`);
console.log(`[API:${requestId}] - Prompt length: ${input.prompt.length} chars`);
console.log(`[API:${requestId}] - Images count: ${input.images?.length || 0}`);
const REPLICATE_API_BASE = "https://api.replicate.com/v1";
@ -276,6 +277,14 @@ async function generateWithReplicate(
...input.parameters,
};
// Add image input if provided (for img2img workflows)
// Note: Different Replicate models use different parameter names
// Using 'image' as it's most common for img2img models
if (input.images && input.images.length > 0) {
predictionInput.image = input.images[0];
console.log(`[API:${requestId}] Added image input to prediction (${input.images[0].substring(0, 50)}...)`);
}
// Create a prediction
console.log(`[API:${requestId}] Creating Replicate prediction...`);
const createResponse = await fetch(`${REPLICATE_API_BASE}/predictions`, {

Loading…
Cancel
Save