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 }) {
/>
پرامپت:
-
+ {filters?.model_name === 0 ? (
+
+ ) : (
+
+ )}
}
type="submit"
- isLoading={isGeneratingImage}
+ isLoading={isGeneratingImage || isGeneratingRoknImage}
alignSelf={"end"}
>
اجرا
diff --git a/pages/index.js b/pages/index.js
index 24e1149..3d991d5 100644
--- a/pages/index.js
+++ b/pages/index.js
@@ -40,9 +40,10 @@ export default function Home() {
useEffect(() => {
if (filters?.tab == 0) setFilters({ tab_menu: "gemma_base" });
- else if (filters?.tab == 1) setFilters({ tab_menu: "sdxl" });
+ else if (filters?.tab == 1 && filters?.model_name == 0) setFilters({ tab_menu: "sdxl" });
+ else if (filters?.tab == 1 && filters?.model_name == 1) setFilters({ tab_menu: "qwen" });
else if (filters?.tab == 2) setFilters({ tab_menu: "cogvideo_base" });
- }, [filters?.tab]);
+ }, [filters?.tab, filters?.model_name]);
return (