error: keep style

components/main/hooks/useInputGeneration.ts

@@ -9,7 +9,6 @@ import { useCollection } from "@/components/modal/useCollection";
 export const useInputGeneration = () => {
   const { setOpen } = useCollection();
   const [myGenerationsId, setGenerationsId] = useLocalStorage<any>('my-own-generations', []);
-  const [style, setStyle] = useState<string>(list_styles[0].name)
 
   const { data: prompt } = useQuery(["prompt"], () => {
     return ''
@@ -21,6 +20,17 @@ export const useInputGeneration = () => {
   })
   const setPrompt = (str:string) => client.setQueryData(["prompt"], () => str)
 
+  const { data: style } = useQuery(["style"], () => {
+    return list_styles[0].name
+  }, {
+    refetchOnWindowFocus: false,
+    refetchOnMount: false,
+    refetchOnReconnect: false,
+    initialData: ''
+  })
+
+  const setStyle = (str:string) => client.setQueryData(["style"], () => str)
+
   const client = useQueryClient()
 
   const { mutate: submit, isLoading: loading } = useMutation(