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<string | null>("")
   const [activeSelectedAIFunction, setActiveSelectedAIFunction] = useState('');
   const [currentSelectedAIFunction, setCurrentSelectedAIFunction] = useState<string | null>("");
-
+  
   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) => (
               <button
                 key={displayedCategory}
-                className={`${displayedCategory.toLowerCase()}-model model-box ${currentSelectedAIFunction === displayedCategory ? 'selected' : ''}`}
+                className={`${displayedCategory.toLowerCase()}-model model-box ${currentSelectedAIFunction === displayedCategory ? 'selected' : ''}`} 
                 onClick={() => modelClicked(displayedCategory)}
               >
                 <div className="overlay">