diff --git a/app/components/Models.tsx b/app/components/Models.tsx index 7c22ab7..24ba852 100644 --- a/app/components/Models.tsx +++ b/app/components/Models.tsx @@ -154,15 +154,15 @@ const modelDropdown = { }; const selectedAIFunction = [ - 'Code', - 'Math', - 'Language', - 'Character', - 'Finance', - 'Weather', - 'Time', - 'Image', - 'Custom1', + 'Code', + 'Math', + 'Language', + 'Character', + 'Finance', + 'Weather', + 'Time', + 'Image', + 'Custom1', 'Custom2' ] @@ -172,29 +172,25 @@ const ModelSection: React.FC = () => { const [radioSelection, setRadioSelection] = useState("") const [activeSelectedAIFunction, setActiveSelectedAIFunction] = useState(''); const [currentSelectedAIFunction, setCurrentSelectedAIFunction] = useState(""); - + useEffect(() => { var temp = localStorage.getItem("activeSelectedAIFunction") || "" setActiveSelectedAIFunction(temp) if (!localStorage.getItem('selectedModelDropdown')) { - localStorage.setItem("selectedModelDropdown", "Offline Fast") + localStorage.setItem("selectedModelDropdown", "Offline Fast" ) } if (!localStorage.getItem("activeSelectedAIFunction")) { setActiveSelectedAIFunction('Code') - localStorage.setItem('activeSelectedAIFunction', 'Code') + localStorage.setItem('activeSelectedAIFunction' ,'Code') } if (!localStorage.getItem("model")) { - localStorage.setItem("model", 'starcoder2') + localStorage.setItem("model" ,'starcoder2') } if (!localStorage.getItem("radioSelection")) { - localStorage.setItem("radioSelection", 'None') - } - - if (!localStorage.getItem("type")) { - localStorage.setItem("type", 'local') + localStorage.setItem("radioSelection" ,'None') } const handleStorageChange = () => { @@ -281,7 +277,7 @@ const ModelSection: React.FC = () => { modelDropdown.offlineNonFoss.includes(model) || modelDropdown.offlineFoss.includes(model); const modelClicked = (model: string) => { - localStorage.setItem('activeSelectedAIFunction', model) + localStorage.setItem('activeSelectedAIFunction' , model) setActiveSelectedAIFunction(model) const selectedAIFunction = selectedModelDropdown as keyof typeof modelList; localStorage.setItem("model", modelList[selectedAIFunction][model as keyof typeof modelList[typeof selectedAIFunction]]) @@ -313,7 +309,7 @@ const ModelSection: React.FC = () => { (displayedCategory) => (