diff --git a/packages/fakeyou.com/src/hooks/useInferenceJobs/useInferenceJobs.tsx b/packages/fakeyou.com/src/hooks/useInferenceJobs/useInferenceJobs.tsx index d614de5ed6..3ceb1801c2 100644 --- a/packages/fakeyou.com/src/hooks/useInferenceJobs/useInferenceJobs.tsx +++ b/packages/fakeyou.com/src/hooks/useInferenceJobs/useInferenceJobs.tsx @@ -5,6 +5,7 @@ import { InferenceJobsContext } from 'context'; export default function useInferenceJobs(jobType: FrontendInferenceJobType) { const { byCategory, enqueue } = useContext(InferenceJobsContext); + return { enqueue, inferenceJobs: (byCategory?.get(jobType) || []), diff --git a/packages/fakeyou.com/src/hooks/useLoraUpload/useLoraUpload.tsx b/packages/fakeyou.com/src/hooks/useLoraUpload/useLoraUpload.tsx index 14a84a13cf..4b8b96624a 100644 --- a/packages/fakeyou.com/src/hooks/useLoraUpload/useLoraUpload.tsx +++ b/packages/fakeyou.com/src/hooks/useLoraUpload/useLoraUpload.tsx @@ -1,7 +1,8 @@ import { useState } from "react"; import { UploadModel } from "@storyteller/components/src/api/image_generation/UploadModel"; import { FetchStatus } from "@storyteller/components/src/api/_common/SharedFetchTypes"; -import { useCoverImgUpload } from "hooks"; +import { useCoverImgUpload, useInferenceJobs } from "hooks"; +import { FrontendInferenceJobType } from "@storyteller/components/src/jobs/InferenceJob"; import { v4 as uuidv4 } from "uuid"; // this hook is mostly for organizational purposes while I work -V @@ -13,6 +14,7 @@ export default function useLoraUpload() { const [descriptionMD, descriptionMDSet] = useState(""); const [writeStatus, writeStatusSet] = useState(FetchStatus.paused); const coverImg = useCoverImgUpload(); + const { enqueue } = useInferenceJobs(FrontendInferenceJobType.ImageGeneration); const onChange = ({ target }: { target: { name: string; value: any } }) => { const todo: { [key: string]: (x: any) => void } = { @@ -38,9 +40,11 @@ export default function useLoraUpload() { visibility, }) .then((res: any) => { - writeStatusSet(FetchStatus.success); - console.log("🌠", res); - // history.replace(`/weight/${token}`); + if (res.success && res.inference_job_token) { + console.log("🌠", res); + writeStatusSet(FetchStatus.success); + enqueue(res.inference_job_token, FrontendInferenceJobType.ImageGeneration); + } }) .catch(err => { writeStatusSet(FetchStatus.error); diff --git a/packages/fakeyou.com/src/hooks/useWeightFetch/useWeightFetch.tsx b/packages/fakeyou.com/src/hooks/useWeightFetch/useWeightFetch.tsx index e3d6790503..887cdc8dc0 100644 --- a/packages/fakeyou.com/src/hooks/useWeightFetch/useWeightFetch.tsx +++ b/packages/fakeyou.com/src/hooks/useWeightFetch/useWeightFetch.tsx @@ -71,17 +71,16 @@ export default function useWeightFetch({ onRemove = n, onSuccess = n, token }: P statusSet(FetchStatus.in_progress); GetWeight(token, {}) .then((res: any) => { - let { - creator_set_visibility, - description_markdown, - title: resTitle, - } = res; - statusSet(FetchStatus.success); - titleSet(resTitle); - descriptionMDSet(description_markdown); - visibilitySet(creator_set_visibility); - onSuccess(res); - setData(res); + if (res.success) { + let { creator_set_visibility, description_markdown, title: resTitle } = res; + + statusSet(FetchStatus.success); + titleSet(resTitle); + descriptionMDSet(description_markdown); + visibilitySet(creator_set_visibility); + onSuccess(res); + setData(res); + } else { statusSet(FetchStatus.error); } }) .catch(err => { statusSet(FetchStatus.error); diff --git a/packages/fakeyou.com/src/v2/view/pages/upload/UploadLoraWeightPage.tsx b/packages/fakeyou.com/src/v2/view/pages/upload/UploadLoraWeightPage.tsx index e13037dfa9..eaeef3483a 100644 --- a/packages/fakeyou.com/src/v2/view/pages/upload/UploadLoraWeightPage.tsx +++ b/packages/fakeyou.com/src/v2/view/pages/upload/UploadLoraWeightPage.tsx @@ -15,6 +15,8 @@ import { TempSelect, TempTextArea, } from "components/common"; +import { FrontendInferenceJobType } from "@storyteller/components/src/jobs/InferenceJob"; +import InferenceJobsList from "components/layout/InferenceJobsList"; import { SessionWrapper } from "@storyteller/components/src/session/SessionWrapper"; import useLoraUpload from "hooks/useLoraUpload"; @@ -32,6 +34,14 @@ export default function UploadLoraWeightPage({ { label: "Private", value: "private" }, ]; + + const failures = (fail = "") => { + switch (fail) { + default: + return "Uknown failure"; + } + }; + const { coverImg, descriptionMD, @@ -66,6 +76,12 @@ export default function UploadLoraWeightPage({ return ( + +
+ + + + +
+ + +
+
+ ; + } else { + if (!weightToken) { return ( + +