Spaces:
Sleeping
Sleeping
| // ── Color hex values ─────────────────────────────────────────── | |
| export const COLOR_HEX = { | |
| Red: '#e74c3c', | |
| Yellow: '#f1c40f', | |
| Purple: '#9b59b6', | |
| Green: '#2ecc71', | |
| Orange: '#e67e22', | |
| Black: '#2c3e50', | |
| Brown: '#8B5E3C', | |
| Grey: '#95a5a6', | |
| } | |
| // ── Mock episodes – 10 items each ───────────────────────────── | |
| export const MOCK_EPISODES = [ | |
| [ | |
| { color:'Red', shape:'Round', texture:'Smooth', type:'FRUIT' }, | |
| { color:'Black', shape:'Shriveled', texture:'Slimy', type:'WASTE' }, | |
| { color:'Yellow', shape:'Oval', texture:'Firm', type:'FRUIT' }, | |
| { color:'Brown', shape:'Shriveled', texture:'Mushy', type:'WASTE' }, | |
| { color:'Green', shape:'Heart', texture:'Smooth', type:'FRUIT' }, | |
| { color:'Grey', shape:'Sunken', texture:'Slimy', type:'WASTE' }, | |
| { color:'Orange', shape:'Pear', texture:'Hard', type:'FRUIT' }, | |
| { color:'Purple', shape:'Round', texture:'Mushy', type:'FRUIT' }, // ambiguous | |
| { color:'Red', shape:'Shriveled', texture:'Slimy', type:'WASTE' }, // ambiguous | |
| { color:'Yellow', shape:'Round', texture:'Juicy', type:'FRUIT' }, | |
| ], | |
| [ | |
| { color:'Brown', shape:'Sunken', texture:'Mushy', type:'WASTE' }, | |
| { color:'Green', shape:'Pear', texture:'Hard', type:'FRUIT' }, | |
| { color:'Grey', shape:'Shriveled', texture:'Slimy', type:'WASTE' }, | |
| { color:'Red', shape:'Heart', texture:'Smooth', type:'FRUIT' }, | |
| { color:'Black', shape:'Sunken', texture:'Mushy', type:'WASTE' }, | |
| { color:'Yellow', shape:'Oval', texture:'Juicy', type:'FRUIT' }, | |
| { color:'Purple', shape:'Shriveled', texture:'Mushy', type:'WASTE' }, | |
| { color:'Orange', shape:'Round', texture:'Smooth', type:'FRUIT' }, | |
| { color:'Grey', shape:'Oval', texture:'Hard', type:'WASTE' }, // ambiguous | |
| { color:'Green', shape:'Round', texture:'Smooth', type:'FRUIT' }, | |
| ], | |
| ] | |
| // ── Agent decision (mock Q-values) ───────────────────────────── | |
| // Trained agent is mostly correct but can be uncertain on ambiguous items | |
| export function getAgentDecision(item) { | |
| const isAmbiguous = | |
| (item.texture === 'Mushy' && item.type === 'FRUIT') || | |
| (item.texture === 'Slimy' && item.type === 'WASTE' && item.color === 'Red') || | |
| (item.texture === 'Hard' && item.type === 'WASTE') | |
| const correctAction = item.type === 'FRUIT' ? 'STORE' : 'CRUSH' | |
| const wrongAction = item.type === 'FRUIT' ? 'CRUSH' : 'STORE' | |
| let storeQ, crushQ | |
| if (isAmbiguous) { | |
| // Agent is uncertain | |
| storeQ = 3.2 + (Math.random() - 0.5) * 4 | |
| crushQ = 3.0 + (Math.random() - 0.5) * 4 | |
| } else if (correctAction === 'STORE') { | |
| storeQ = 7.5 + Math.random() * 2 | |
| crushQ = -4.0 + Math.random() * 2 | |
| } else { | |
| crushQ = 7.5 + Math.random() * 2 | |
| storeQ = -4.0 + Math.random() * 2 | |
| } | |
| const action = storeQ >= crushQ ? 'STORE' : 'CRUSH' | |
| const isCorrect = action === correctAction | |
| const reward = isCorrect ? 8 : -8 | |
| return { | |
| action, | |
| isCorrect, | |
| reward, | |
| qValues: { STORE: storeQ.toFixed(2), CRUSH: crushQ.toFixed(2) } | |
| } | |
| } | |
| // ── Random agent decision ────────────────────────────────────── | |
| export function getRandomDecision(item) { | |
| const action = Math.random() > 0.5 ? 'STORE' : 'CRUSH' | |
| const correctAction = item.type === 'FRUIT' ? 'STORE' : 'CRUSH' | |
| const isCorrect = action === correctAction | |
| return { action, isCorrect, reward: isCorrect ? 8 : -8, qValues: null } | |
| } | |