custom pipeline

oaknational · Nov 29, 2024 · 8233576 · 8233576
1 parent 168ae75
commit 8233576
Show file tree

Hide file tree

Showing 4 changed files with 483 additions and 62 deletions.
diff --git a/apps/nextjs/src/app/image-spike/[slug]/images.tsx b/apps/nextjs/src/app/image-spike/[slug]/images.tsx
@@ -76,6 +76,7 @@ const ImagesPage = ({ pageData }: { pageData: PageData }) => {
   };
 
   const trpcMutations = {
+    "Custom Pipeline": trpc.imageGen.customPipeline.useMutation(),
     "Stable Diffusion Ultra": trpc.imageGen.stableDifUltra.useMutation(),
     "Stable Diffusion 3.0 & 3.5": trpc.imageGen.stableDif3.useMutation(),
     "Stable Diffusion Core": trpc.imageGen.stableDifCore.useMutation(),
@@ -110,7 +111,8 @@ const ImagesPage = ({ pageData }: { pageData: PageData }) => {
         if (
           source === "Stable Diffusion Ultra" ||
           source === "Stable Diffusion 3.0 & 3.5" ||
-          source === "Stable Diffusion Core"
+          source === "Stable Diffusion Core" ||
+          source === "Custom Pipeline"
         ) {
           updateColumn(columnId, {
             imageSearchBatch: [

diff --git a/apps/nextjs/src/app/image-spike/image-spike.tsx b/apps/nextjs/src/app/image-spike/image-spike.tsx
@@ -6,6 +6,10 @@ const ImageSpike = ({ lessons }: { lessons: any }) => {
   const lessonTitles = lessons.map((lesson: any) => ({
     sessionId: lesson.id,
     lessonTitle: lesson?.output?.lessonPlan?.title || "",
+    hasThreeCycles:
+      !!lesson?.output?.lessonPlan?.cycle1 &&
+      !!lesson?.output?.lessonPlan?.cycle2 &&
+      !!lesson?.output?.lessonPlan?.cycle3,
   }));
 
   return (
@@ -16,11 +20,13 @@ const ImageSpike = ({ lessons }: { lessons: any }) => {
           ({
             sessionId,
             lessonTitle,
+            hasThreeCycles,
           }: {
             sessionId: string;
             lessonTitle: string;
+            hasThreeCycles: boolean;
           }) => {
-            if (!lessonTitle) {
+            if (!lessonTitle || !hasThreeCycles) {
               return null;
             }
             return (