| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| import React, { useState, useEffect, useCallback, useRef, useMemo } from 'react'; |
| import { ArrowLeft, Search, Shirt, Image as ImageIcon, FileText, Loader2, Eye, Lock, Package, AlertCircle, Trash2, RotateCcw, } from 'lucide-react'; |
| import { fetchInventoryCategories, searchInventory, deleteInventoryItem, } from '../inventoryApi'; |
| import { PersonaDocumentsPanel } from './PersonaDocumentsPanel'; |
| |
| |
| |
| const SENS_ORDER = { safe: 0, sensitive: 1, explicit: 2 }; |
| function classifySensitivity(label) { |
| const low = (label || '').trim().toLowerCase(); |
| const keywords = ['lingerie', 'intimate', 'underwear', 'bra', 'panties', 'sexy', 'bikini']; |
| return keywords.some(k => low.includes(k)) ? 'sensitive' : 'safe'; |
| } |
| function allowedBySensitivity(itemSens, maxSens) { |
| return (SENS_ORDER[itemSens] ?? 0) <= (SENS_ORDER[maxSens] ?? 0); |
| } |
| |
| |
| |
| function buildDraftItems(draft, sensitivityMax) { |
| const out = []; |
| const active = draft.selected; |
| |
| for (const s of draft.sets) { |
| for (const img of s.images) { |
| if (!img.url) |
| continue; |
| if (!allowedBySensitivity('safe', sensitivityMax)) |
| continue; |
| out.push({ |
| id: img.id, |
| type: 'image', |
| label: 'Portrait', |
| tags: ['portrait', 'set'], |
| sensitivity: 'safe', |
| url: img.url, |
| set_id: s.set_id, |
| image_id: img.id, |
| is_active_look: !!(active && s.set_id === active.set_id && img.id === active.image_id), |
| }); |
| } |
| } |
| |
| for (const outfit of draft.outfits) { |
| const sens = classifySensitivity(outfit.label); |
| if (!allowedBySensitivity(sens, sensitivityMax)) |
| continue; |
| const firstImg = outfit.images[0]; |
| out.push({ |
| id: outfit.id, |
| type: 'outfit', |
| label: outfit.label, |
| tags: [outfit.label.toLowerCase()], |
| sensitivity: sens, |
| description: outfit.outfit_prompt, |
| asset_ids: outfit.images.map(img => img.id), |
| preview_asset_id: firstImg?.id, |
| url: firstImg?.url, |
| |
| set_id: firstImg ? outfit.id : undefined, |
| image_id: firstImg?.id, |
| is_active_look: !!(active && firstImg && outfit.id === active.set_id && firstImg.id === active.image_id), |
| }); |
| } |
| return out; |
| } |
| |
| |
| |
| const CATEGORY_ICONS = { |
| outfit: Shirt, |
| image: ImageIcon, |
| file: FileText, |
| all: Package, |
| }; |
| const CATEGORY_COLORS = { |
| outfit: 'text-amber-400', |
| image: 'text-pink-400', |
| file: 'text-blue-400', |
| all: 'text-purple-400', |
| }; |
| |
| |
| |
| export function InventoryView({ projectId, backendUrl, apiKey, onBack, onShowInChat, activeSelection, onSetActiveLook, draftAppearance }) { |
| |
| const [categories, setCategories] = useState([]); |
| const [items, setItems] = useState([]); |
| const [totalCount, setTotalCount] = useState(0); |
| const [loading, setLoading] = useState(true); |
| const [searchLoading, setSearchLoading] = useState(false); |
| const [error, setError] = useState(null); |
| const [selectedCategory, setSelectedCategory] = useState('all'); |
| const [query, setQuery] = useState(''); |
| const [debouncedQuery, setDebouncedQuery] = useState(''); |
| const [limit] = useState(30); |
| const [refreshKey, setRefreshKey] = useState(0); |
| const [deleting, setDeleting] = useState(null); |
| const searchTimerRef = useRef(); |
| |
| const isSpicy = (() => { |
| try { |
| return localStorage.getItem('homepilot_nsfw_mode') === 'true'; |
| } |
| catch { |
| return false; |
| } |
| })(); |
| const sensitivityMax = isSpicy ? 'explicit' : 'safe'; |
| |
| |
| |
| const draftItems = useMemo(() => { |
| if (!draftAppearance) |
| return []; |
| return buildDraftItems(draftAppearance, sensitivityMax); |
| }, [draftAppearance, sensitivityMax]); |
| |
| const mergedItems = useMemo(() => { |
| const backendIds = new Set(items.map(i => i.id)); |
| const q = debouncedQuery.toLowerCase(); |
| const extras = draftItems.filter(d => { |
| if (backendIds.has(d.id)) |
| return false; |
| if (selectedCategory !== 'all' && d.type !== selectedCategory) |
| return false; |
| if (q) { |
| const label = (d.label || '').toLowerCase(); |
| const desc = (d.description || '').toLowerCase(); |
| const tags = (d.tags || []).join(' ').toLowerCase(); |
| if (!label.includes(q) && !desc.includes(q) && !tags.includes(q)) |
| return false; |
| } |
| return true; |
| }); |
| return [...items, ...extras]; |
| }, [items, draftItems, selectedCategory, debouncedQuery]); |
| |
| |
| const mergedCategories = useMemo(() => { |
| const extrasByType = {}; |
| const backendIds = new Set(items.map(i => i.id)); |
| for (const d of draftItems) { |
| if (backendIds.has(d.id)) |
| continue; |
| |
| const url = d.url || ''; |
| if (url.includes('/files/')) |
| continue; |
| if (!url) |
| continue; |
| extrasByType[d.type] = (extrasByType[d.type] || 0) + 1; |
| } |
| if (categories.length > 0) { |
| return categories.map(cat => ({ |
| ...cat, |
| count: (cat.count ?? 0) + (extrasByType[cat.type] ?? 0), |
| })); |
| } |
| return [ |
| { type: 'outfit', label: 'Outfits', count: extrasByType.outfit ?? 0 }, |
| { type: 'image', label: 'Photos', count: extrasByType.image ?? 0 }, |
| { type: 'file', label: 'Documents', count: extrasByType.file ?? 0 }, |
| ]; |
| }, [categories, items, draftItems]); |
| |
| |
| |
| useEffect(() => { |
| if (searchTimerRef.current) |
| clearTimeout(searchTimerRef.current); |
| searchTimerRef.current = setTimeout(() => setDebouncedQuery(query), 300); |
| return () => { if (searchTimerRef.current) |
| clearTimeout(searchTimerRef.current); }; |
| }, [query]); |
| |
| |
| |
| useEffect(() => { |
| let cancelled = false; |
| (async () => { |
| try { |
| const data = await fetchInventoryCategories(backendUrl, projectId, { |
| apiKey, |
| includeTags: true, |
| sensitivityMax, |
| }); |
| if (!cancelled) |
| setCategories(data.categories || []); |
| } |
| catch (e) { |
| if (!cancelled) |
| setError(e.message); |
| } |
| })(); |
| return () => { cancelled = true; }; |
| }, [backendUrl, projectId, apiKey, sensitivityMax, refreshKey]); |
| |
| |
| |
| useEffect(() => { |
| let cancelled = false; |
| (async () => { |
| setSearchLoading(true); |
| try { |
| const types = selectedCategory === 'all' ? undefined : [selectedCategory]; |
| const data = await searchInventory(backendUrl, projectId, { |
| apiKey, |
| query: debouncedQuery || undefined, |
| types, |
| limit, |
| sensitivityMax, |
| }); |
| if (!cancelled) { |
| setItems(data.items || []); |
| setTotalCount(data.total_count || 0); |
| setError(null); |
| } |
| } |
| catch (e) { |
| if (!cancelled) |
| setError(e.message); |
| } |
| finally { |
| if (!cancelled) { |
| setSearchLoading(false); |
| setLoading(false); |
| } |
| } |
| })(); |
| return () => { cancelled = true; }; |
| }, [backendUrl, projectId, apiKey, selectedCategory, debouncedQuery, limit, sensitivityMax, refreshKey]); |
| |
| |
| |
| const resolveImageUrl = useCallback((item) => { |
| const url = item.url; |
| if (!url) |
| return undefined; |
| const tok = (() => { try { |
| return localStorage.getItem('homepilot_auth_token') || ''; |
| } |
| catch { |
| return ''; |
| } })(); |
| let full = url.startsWith('http') ? url : `${backendUrl}${url}`; |
| if (tok && full.includes('/files/')) { |
| const sep = full.includes('?') ? '&' : '?'; |
| full = `${full}${sep}token=${encodeURIComponent(tok)}`; |
| } |
| return full; |
| }, [backendUrl]); |
| |
| |
| |
| const handleDelete = useCallback(async (item) => { |
| if (deleting) |
| return; |
| |
| if (item.is_active_look) { |
| setError('Cannot delete the active look. Change the active look first.'); |
| return; |
| } |
| if (!confirm(`Delete "${item.label}"? This cannot be undone.`)) |
| return; |
| setDeleting(item.id); |
| try { |
| await deleteInventoryItem(backendUrl, projectId, item.id, { apiKey }); |
| setRefreshKey((k) => k + 1); |
| } |
| catch (e) { |
| const msg = e.message || ''; |
| if (msg.includes('409') || msg.includes('active look')) { |
| setError('Cannot delete the active look. Change the active look first.'); |
| } |
| else { |
| setError(`Delete failed: ${msg}`); |
| } |
| } |
| finally { |
| setDeleting(null); |
| } |
| }, [backendUrl, projectId, apiKey, deleting]); |
| |
| |
| |
| const handleSetActiveLook = useCallback((item) => { |
| if (!item.set_id || !item.image_id) |
| return; |
| onSetActiveLook?.({ set_id: item.set_id, image_id: item.image_id }); |
| }, [onSetActiveLook]); |
| |
| |
| |
| const totalItems = mergedCategories.reduce((sum, c) => sum + (c.count || 0), 0); |
| const renderSidebar = () => (<div className="w-44 shrink-0 border-r border-white/10 py-3 space-y-0.5"> |
| {/* All items */} |
| <SidebarItem icon={Package} label="All Items" count={totalItems} color="text-purple-400" active={selectedCategory === 'all'} onClick={() => setSelectedCategory('all')}/> |
| {/* Per-category */} |
| {mergedCategories.map((cat) => (<SidebarItem key={cat.type} icon={CATEGORY_ICONS[cat.type] || Package} label={cat.label} count={cat.count} color={CATEGORY_COLORS[cat.type] || 'text-white/50'} active={selectedCategory === cat.type} onClick={() => setSelectedCategory(cat.type)}/>))} |
| </div>); |
| |
| |
| |
| const renderItemCard = (item) => { |
| const isImage = item.type === 'image'; |
| const isOutfit = item.type === 'outfit'; |
| const isFile = item.type === 'file'; |
| const isSensitive = item.sensitivity === 'sensitive' || item.sensitivity === 'explicit'; |
| const imgUrl = (isImage || isOutfit) ? resolveImageUrl(item) : undefined; |
| const canSetActive = !!(item.set_id && item.image_id && onSetActiveLook); |
| |
| const isSelected = !!(activeSelection |
| && item.set_id === activeSelection.set_id |
| && item.image_id === activeSelection.image_id); |
| const isDeleting = deleting === item.id; |
| return (<div key={item.id} className={[ |
| 'group relative rounded-xl overflow-hidden transition-all', |
| isSelected |
| ? 'border-2 border-amber-500/50 ring-2 ring-amber-500/20 shadow-[0_0_12px_rgba(245,158,11,0.15)]' |
| : 'border border-white/[0.06] bg-white/[0.02] hover:border-white/15', |
| isDeleting ? 'opacity-40 pointer-events-none' : '', |
| ].join(' ')}> |
| {/* Equipped badge — Wardrobe style (top-right, always visible when selected or outfit is equipped) */} |
| {(isSelected || (isOutfit && item.equipped)) && (<div className="absolute top-1.5 right-1.5 px-1.5 py-0.5 rounded-md bg-amber-500/80 text-[7px] text-white font-bold uppercase tracking-wider z-10"> |
| Equipped |
| </div>)} |
| |
| {/* Delete button — top-right on hover (shifts left when Equipped badge is showing) */} |
| <div className={[ |
| 'absolute top-1.5 z-10 opacity-0 group-hover:opacity-100 transition-opacity', |
| isSelected ? 'left-1.5' : 'right-1.5', |
| ].join(' ')}> |
| <ActionButton icon={Trash2} label="Delete" onClick={() => handleDelete(item)} variant="danger"/> |
| </div> |
| |
| {/* Image preview — primary click selects (like Wardrobe) */} |
| {(isImage || isOutfit) && (<div className="aspect-[3/4] bg-black/30 relative overflow-hidden cursor-pointer" onClick={() => { |
| if (canSetActive) { |
| handleSetActiveLook(item); |
| } |
| }}> |
| {isSensitive && !isSpicy ? (<div className="absolute inset-0 flex flex-col items-center justify-center gap-2 bg-black/60"> |
| <Lock size={20} className="text-white/30"/> |
| <span className="text-[10px] text-white/30">Locked</span> |
| </div>) : imgUrl ? (<img src={imgUrl} alt={item.label} className="w-full h-full object-cover" loading="lazy" onError={(e) => { e.target.style.display = 'none'; }}/>) : (<div className="absolute inset-0 flex items-center justify-center"> |
| <ImageIcon size={24} className="text-white/20"/> |
| </div>)} |
| |
| {/* Hover actions — bottom center (only Show in chat) */} |
| <div className="absolute inset-0 bg-black/0 group-hover:bg-black/40 transition-all flex items-end justify-center pb-2 gap-1.5 opacity-0 group-hover:opacity-100"> |
| {onShowInChat && imgUrl && (<ActionButton icon={Eye} label="Show in chat" onClick={() => onShowInChat(item, imgUrl)}/>)} |
| </div> |
| </div>)} |
| |
| {/* File icon for documents */} |
| {isFile && (<div className="aspect-[4/3] bg-black/20 flex items-center justify-center relative"> |
| <div className="flex flex-col items-center gap-1.5"> |
| <FileText size={28} className="text-blue-400/60"/> |
| <span className="text-[9px] text-white/30 uppercase tracking-wider"> |
| {(item.mime || '').split('/').pop() || 'file'} |
| </span> |
| </div> |
| </div>)} |
| |
| {/* Label + metadata */} |
| <div className="px-2.5 py-2"> |
| <div className="text-xs text-white truncate font-medium">{item.label}</div> |
| <div className="flex items-center gap-1 mt-1 flex-wrap"> |
| <span className={`text-[9px] uppercase tracking-wider ${CATEGORY_COLORS[item.type] || 'text-white/40'}`}> |
| {item.type} |
| </span> |
| {isSensitive && (<span className={`text-[9px] flex items-center gap-0.5 ${item.sensitivity === 'explicit' ? 'text-red-400/60' : 'text-orange-400/60'}`}> |
| <Lock size={8}/> {item.sensitivity} |
| </span>)} |
| {/* 360 preview badge for outfits with view packs */} |
| {isOutfit && item.interactive_preview && (<span className="text-[9px] flex items-center gap-0.5 text-violet-300/60"> |
| <RotateCcw size={8}/> 360 |
| </span>)} |
| </div> |
| {/* Tags */} |
| {item.tags && item.tags.length > 0 && (<div className="flex flex-wrap gap-1 mt-1.5"> |
| {item.tags.slice(0, 3).map((tag) => (<span key={tag} className="text-[9px] px-1.5 py-0.5 rounded-full bg-white/5 text-white/30 border border-white/5"> |
| {tag} |
| </span>))} |
| </div>)} |
| {/* View angle chips for outfits with view packs */} |
| {isOutfit && item.available_views && item.available_views.length > 0 && (<div className="flex flex-wrap gap-1 mt-1.5"> |
| {item.available_views.map((angle) => (<span key={angle} className="text-[8px] px-1.5 py-0.5 rounded-full bg-violet-500/8 text-violet-300/50 border border-violet-500/15 capitalize"> |
| {angle} |
| </span>))} |
| </div>)} |
| {isFile && item.size_bytes != null && item.size_bytes > 0 && (<div className="text-[9px] text-white/25 mt-1"> |
| {item.size_bytes > 1048576 |
| ? `${(item.size_bytes / 1048576).toFixed(1)} MB` |
| : `${Math.round(item.size_bytes / 1024)} KB`} |
| </div>)} |
| </div> |
| </div>); |
| }; |
| |
| |
| |
| const activeLookItem = mergedItems.find((i) => i.is_active_look); |
| const outfitCount = mergedCategories.find((c) => c.type === 'outfit')?.count || 0; |
| const photoCount = mergedCategories.find((c) => c.type === 'image')?.count || 0; |
| const docCount = mergedCategories.find((c) => c.type === 'file')?.count || 0; |
| const renderLoadoutSummary = () => (<div className="flex items-center gap-4 px-4 py-2.5 bg-white/[0.03] border-b border-white/5"> |
| {/* Mini Active Look */} |
| {activeLookItem?.url && (<div className="w-8 h-8 rounded-lg overflow-hidden border border-amber-500/30 shrink-0"> |
| <img src={resolveImageUrl(activeLookItem) || ''} alt="Active Look" className="w-full h-full object-cover"/> |
| </div>)} |
| <div className="flex items-center gap-3 text-[10px] text-white/40"> |
| <span className="flex items-center gap-1"><Shirt size={10} className="text-amber-400/60"/> {outfitCount} outfits</span> |
| <span className="flex items-center gap-1"><ImageIcon size={10} className="text-pink-400/60"/> {photoCount} photos</span> |
| <span className="flex items-center gap-1"><FileText size={10} className="text-blue-400/60"/> {docCount} docs</span> |
| </div> |
| </div>); |
| |
| |
| |
| if (loading && items.length === 0) { |
| return (<div className="flex-1 flex items-center justify-center py-20"> |
| <Loader2 size={24} className="animate-spin text-white/30"/> |
| </div>); |
| } |
| return (<div className="flex flex-col flex-1 min-h-0"> |
| {/* Top bar: back + search */} |
| <div className="flex items-center gap-3 px-4 py-3 border-b border-white/10 bg-white/[0.02]"> |
| <button onClick={onBack} className="flex items-center gap-1.5 text-xs text-white/50 hover:text-white transition-colors shrink-0"> |
| <ArrowLeft size={14}/> |
| <span>Back</span> |
| </button> |
| <div className="flex-1 relative"> |
| <Search size={14} className="absolute left-3 top-1/2 -translate-y-1/2 text-white/30"/> |
| <input type="text" value={query} onChange={(e) => setQuery(e.target.value)} placeholder="Search inventory..." className="w-full pl-9 pr-3 py-1.5 text-xs bg-white/5 border border-white/10 rounded-lg text-white placeholder:text-white/25 focus:outline-none focus:border-white/20 transition-colors"/> |
| </div> |
| <div className="text-[10px] text-white/30 shrink-0"> |
| {searchLoading ? (<Loader2 size={12} className="animate-spin"/>) : (`${mergedItems.length} items`)} |
| </div> |
| </div> |
| |
| {/* Loadout summary */} |
| {renderLoadoutSummary()} |
| |
| {/* Two-pane body */} |
| <div className="flex flex-1 min-h-0"> |
| {/* Sidebar */} |
| {renderSidebar()} |
| |
| {/* Item grid */} |
| <div className="flex-1 overflow-y-auto custom-scrollbar p-3"> |
| {error && (<div className="flex items-center gap-2 px-3 py-2 mb-3 rounded-lg bg-red-500/10 border border-red-500/20 text-xs text-red-400"> |
| <AlertCircle size={14}/> |
| {error} |
| </div>)} |
| |
| {/* Knowledge Base panel (shown when Documents category is selected) */} |
| {selectedCategory === 'file' && (<div className="mb-4 p-3 rounded-xl bg-white/[0.02] border border-white/5"> |
| <PersonaDocumentsPanel projectId={projectId} backendUrl={backendUrl} apiKey={apiKey} onChanged={() => setRefreshKey((k) => k + 1)}/> |
| </div>)} |
| |
| {mergedItems.length === 0 && !searchLoading && selectedCategory !== 'file' && (<div className="flex flex-col items-center justify-center py-16 gap-2 text-white/30"> |
| <Package size={32} className="text-white/15"/> |
| <p className="text-xs"> |
| {debouncedQuery |
| ? 'No items match your search.' |
| : totalItems > 0 && mergedItems.length === 0 |
| ? 'No items returned. Check sensitivity filters or try refreshing.' |
| : 'Inventory is empty.'} |
| </p> |
| </div>)} |
| |
| {/* Hide inventory grid when Documents tab is active — the Knowledge Base panel above is the single source of truth */} |
| {selectedCategory !== 'file' && (<div className="grid grid-cols-2 sm:grid-cols-3 gap-2.5"> |
| {mergedItems.map(renderItemCard)} |
| </div>)} |
| |
| {selectedCategory !== 'file' && items.length < totalCount && !searchLoading && (<div className="text-center py-4"> |
| <span className="text-[10px] text-white/25"> |
| Showing {mergedItems.length} of {totalCount + (mergedItems.length - items.length)} items |
| </span> |
| </div>)} |
| </div> |
| </div> |
| </div>); |
| } |
| |
| |
| |
| function SidebarItem({ icon: Icon, label, count, color, active, onClick, }) { |
| return (<button onClick={onClick} className={[ |
| 'w-full flex items-center gap-2.5 px-3 py-2 text-xs transition-colors', |
| active |
| ? 'bg-white/10 text-white border-r-2 border-pink-500' |
| : 'text-white/50 hover:text-white/70 hover:bg-white/5', |
| ].join(' ')}> |
| <Icon size={14} className={active ? color : 'text-white/30'}/> |
| <span className="flex-1 text-left truncate">{label}</span> |
| {count != null && (<span className="text-[10px] text-white/30 tabular-nums">{count}</span>)} |
| </button>); |
| } |
| |
| |
| |
| function ActionButton({ icon: Icon, label, onClick, variant, }) { |
| const isDanger = variant === 'danger'; |
| return (<button onClick={(e) => { e.stopPropagation(); onClick(); }} title={label} className={[ |
| 'p-1.5 rounded-lg transition-all border', |
| isDanger |
| ? 'bg-red-900/60 text-red-300/70 hover:text-red-200 hover:bg-red-800/80 border-red-500/20' |
| : 'bg-black/60 text-white/70 hover:text-white hover:bg-black/80 border-white/10', |
| ].join(' ')}> |
| <Icon size={12}/> |
| </button>); |
| } |
|
|