diff --git a/apps/web/client/src/app/project/[id]/_components/right-panel/chat-tab/context-pills/image-pill.tsx b/apps/web/client/src/app/project/[id]/_components/right-panel/chat-tab/context-pills/image-pill.tsx index 02864c6f83..162cbaf459 100644 --- a/apps/web/client/src/app/project/[id]/_components/right-panel/chat-tab/context-pills/image-pill.tsx +++ b/apps/web/client/src/app/project/[id]/_components/right-panel/chat-tab/context-pills/image-pill.tsx @@ -10,14 +10,16 @@ export const ImagePill = React.forwardRef< { context: ImageMessageContext; onRemove: () => void; + onSave?: () => void; } ->(({ context, onRemove }, ref) => { +>(({ context, onRemove, onSave }, ref) => { if (context.type !== MessageContextType.IMAGE) { console.warn('ImagePill received non-image context'); return null; } const isVideo = isVideoFile(context.mimeType); + const canSave = context.source === 'external' && !!onSave; return ( - {/* Left side: Image/Video thumbnail */}
{isVideo ? (
- {/* Right side: Filename */} {getTruncatedName(context)} - {/* Hover X button */} + {canSave && ( + + )} +