File size: 4,909 Bytes
a94a061 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 |
import React, { createContext, useContext, useState, useCallback } from 'react'
import { Section } from '../types'
interface ZeroShotClassificationConfig {
threshold: number
}
interface ZeroShotClassificationContextType {
config: ZeroShotClassificationConfig
setConfig: React.Dispatch<React.SetStateAction<ZeroShotClassificationConfig>>
text: string
setText: React.Dispatch<React.SetStateAction<string>>
sections: Section[]
setSections: React.Dispatch<React.SetStateAction<Section[]>>
addCategory: () => void
removeCategory: () => void
clearResults: () => void
updateSectionTitle: (index: number, newTitle: string) => void
}
const ZeroShotClassificationContext = createContext<ZeroShotClassificationContextType | undefined>(undefined)
export const useZeroShotClassification = () => {
const context = useContext(ZeroShotClassificationContext)
if (!context) {
throw new Error('useZeroShotClassification must be used within a ZeroShotClassificationProvider')
}
return context
}
const PLACEHOLDER_REVIEWS: string[] = [
// battery/charging problems
'Disappointed with the battery life! The phone barely lasts half a day with regular use. Considering how much I paid for it, I expected better performance in this department.',
"I bought this phone a week ago, and I'm already frustrated with the battery life. It barely lasts half a day with normal usage. I expected more from a supposedly high-end device",
"The charging port is so finicky. Sometimes it takes forever to charge, and other times it doesn't even recognize the charger. Frustrating experience!",
// overheating
"This phone heats up way too quickly, especially when using demanding apps. It's uncomfortable to hold, and I'm concerned it might damage the internal components over time. Not what I expected",
"This phone is like holding a hot potato. Video calls turn it into a scalding nightmare. Seriously, can't it keep its cool?",
"Forget about a heatwave outside; my phone's got its own. It's like a little portable heater. Not what I signed up for.",
// poor build quality
'I dropped the phone from a short distance, and the screen cracked easily. Not as durable as I expected from a flagship device.',
'Took a slight bump in my bag, and the frame got dinged. Are we back in the flip phone era?',
"So, my phone's been in my pocket with just keys – no ninja moves or anything. Still, it managed to get some scratches. Disappointed with the build quality.",
// software
'The software updates are a nightmare. Each update seems to introduce new bugs, and it takes forever for them to be fixed.',
'Constant crashes and freezes make me want to throw it into a black hole.',
"Every time I open Instagram, my phone freezes and crashes. It's so frustrating!",
// other
"I'm not sure what to make of this phone. It's not bad, but it's not great either. I'm on the fence about it.",
"I hate the color of this phone. It's so ugly!",
"This phone sucks! I'm returning it."
].sort(() => Math.random() - 0.5)
const PLACEHOLDER_SECTIONS: string[] = [
'Battery and charging problems',
'Overheating',
'Poor build quality',
'Software issues',
'Other'
]
export const ZeroShotClassificationProvider: React.FC<{ children: React.ReactNode }> = ({ children }) => {
const [config, setConfig] = useState<ZeroShotClassificationConfig>({
threshold: 0.5
})
const [text, setText] = useState<string>(PLACEHOLDER_REVIEWS.join('\n'))
const [sections, setSections] = useState<Section[]>(
PLACEHOLDER_SECTIONS.map((title) => ({ title, items: [] }))
)
const addCategory = useCallback(() => {
setSections((sections) => {
const newSections = [...sections]
// add at position 2 from the end
newSections.splice(newSections.length - 1, 0, {
title: 'New Category',
items: []
})
return newSections
})
}, [])
const removeCategory = useCallback(() => {
setSections((sections) => {
const newSections = [...sections]
newSections.splice(newSections.length - 2, 1) // Remove second last element
return newSections
})
}, [])
const clearResults = useCallback(() => {
setSections((sections) =>
sections.map((section) => ({
...section,
items: []
}))
)
}, [])
const updateSectionTitle = useCallback((index: number, newTitle: string) => {
setSections((sections) => {
const newSections = [...sections]
newSections[index].title = newTitle
return newSections
})
}, [])
const value: ZeroShotClassificationContextType = {
config,
setConfig,
text,
setText,
sections,
setSections,
addCategory,
removeCategory,
clearResults,
updateSectionTitle
}
return (
<ZeroShotClassificationContext.Provider value={value}>
{children}
</ZeroShotClassificationContext.Provider>
)
}
|