diff --git a/components/layout/SidebarList.jsx b/components/layout/SidebarList.jsx index f1bddd1..809a186 100644 --- a/components/layout/SidebarList.jsx +++ b/components/layout/SidebarList.jsx @@ -27,7 +27,7 @@ export default function SidebarList({ filters, setFilters }) { /> )} - {filters?.tab === 1 && ( + {filters?.tab === 1 && filters?.model_name === 0 && ( <> )} + {filters?.tab === 1 && filters?.model_name === 1 && ( + <> + + + + + )} {filters?.tab === 2 && ( <> { }; export default function TextImagePanel({ filters }) { - const { register, handleSubmit, watch } = useForm(""); + const { register, handleSubmit, watch, setValue } = useForm(""); + const { + register: registerRokn, + handleSubmit: handleSubmitRokn, + watch: watchRokn, + setValue: setValueRokn, + } = useForm(""); + const [images, setImages] = useState([]); - const width = watch("width"); - const height = watch("height"); + const width = filters?.model_name === 0 ? watch("width") : watchRokn("width"); + const height = + filters?.model_name === 0 ? watch("height") : watchRokn("height"); const { trigger: triggerGenerateImage, isMutating: isGeneratingImage } = useSWRMutation( @@ -53,10 +61,39 @@ export default function TextImagePanel({ filters }) { }, ); + const { + trigger: triggerGenerateRoknImage, + isMutating: isGeneratingRoknImage, + } = useSWRMutation( + `/content/rokn/t2i?model_key=${filters?.tab_menu}`, + postRequest, + { + onSuccess: (data) => console.log(data), + }, + ); + const onSubmit = (data) => { - triggerGenerateImage(data); + if (filters?.model_name === 0) triggerGenerateImage(data); + else triggerGenerateRoknImage(data); }; + useEffect(() => { + if (filters?.model_name === 0) { + setValue("height", 768); + setValue("width", 768); + setValue("seed", 42); + setValue("num_inference_steps", 60); + setValue("guidance_scale", 7); + setValue("prompt", ""); + } else { + setValueRokn("height", 768); + setValueRokn("width", 768); + setValueRokn("steps", 10); + setValueRokn("preset", ""); + setValue("prompt", ""); + } + }, [filters?.model_name]); + return ( {" "} @@ -82,12 +123,15 @@ export default function TextImagePanel({ filters }) { /> طول: - + {filters?.model_name === 0 ? ( + + ) : ( + + )} @@ -102,22 +146,18 @@ export default function TextImagePanel({ filters }) { /> عرض: - - - - seed: - + {filters?.model_name === 0 ? ( + + ) : ( + + )} + {filters?.model_name === 0 && ( + + seed: + + + )} تعداد مراحل استنتاج: - - - - - - مقیاس راهنمایی: - - + ) : ( + + )} + {filters?.model_name === 0 && ( + + + + مقیاس راهنمایی: + + + + )} + {filters?.model_name === 1 && ( + + + + سبک تصویر: + + + + )} @@ -172,14 +233,18 @@ export default function TextImagePanel({ filters }) { /> پرامپت: -