From c9435998a09beef311e2551157f9f3a0684a5324 Mon Sep 17 00:00:00 2001
From: wil <test@local.host>
Date: Thu, 22 Feb 2024 16:36:09 -0500
Subject: [PATCH] now profile library can show uploaded workflows

---
 .../src/api/_common/enums/WeightCategory.ts   |  1 +
 .../components/common/Card/WeightsCards.tsx   |  2 ++
 .../src/v2/view/pages/weight/WeightPage.tsx   | 23 +++++++++----------
 3 files changed, 14 insertions(+), 12 deletions(-)

diff --git a/packages/components/src/api/_common/enums/WeightCategory.ts b/packages/components/src/api/_common/enums/WeightCategory.ts
index e0c256eb67..834ed7b43f 100644
--- a/packages/components/src/api/_common/enums/WeightCategory.ts
+++ b/packages/components/src/api/_common/enums/WeightCategory.ts
@@ -4,4 +4,5 @@ export enum WeightCategory {
   VC = "voice_conversion",
   ZS = "zero_shot",
   SD = "image_generation",
+  WF = "workflow_config",
 }
diff --git a/packages/fakeyou.com/src/components/common/Card/WeightsCards.tsx b/packages/fakeyou.com/src/components/common/Card/WeightsCards.tsx
index e8879b68db..ee8ee9d52b 100644
--- a/packages/fakeyou.com/src/components/common/Card/WeightsCards.tsx
+++ b/packages/fakeyou.com/src/components/common/Card/WeightsCards.tsx
@@ -9,12 +9,14 @@ interface Props {
 }
 
 export default function WeightsCards({ props, type }: Props) {
+  console.log(props);
   switch (type) {
     case WeightCategory.TTS:
     case WeightCategory.VC:
     case WeightCategory.ZS:
       return <AudioCard { ...props} showCover />;
     case WeightCategory.SD:
+    case WeightCategory.WF:
       return <ImageCard {...props} />;
     default:
       return <div>Unsupported weight type</div>;
diff --git a/packages/fakeyou.com/src/v2/view/pages/weight/WeightPage.tsx b/packages/fakeyou.com/src/v2/view/pages/weight/WeightPage.tsx
index 11e051d2c3..667c9a7a63 100644
--- a/packages/fakeyou.com/src/v2/view/pages/weight/WeightPage.tsx
+++ b/packages/fakeyou.com/src/v2/view/pages/weight/WeightPage.tsx
@@ -73,13 +73,7 @@ export default function WeightPage({
   const history = useHistory();
   const bookmarks = useBookmarks();
   const ratings = useRatings();
-  const {
-    data: weight,
-    fetchError,
-    isLoading,
-    title,
-    remove,
-  } = useWeightFetch({
+  const fetchedWeight = useWeightFetch({
     onRemove: () => {
       history.push(source || "");
     },
@@ -89,7 +83,14 @@ export default function WeightPage({
     },
     token: weight_token,
   });
-
+  const {
+    data: weight,
+    fetchError,
+    isLoading,
+    title,
+    remove,
+  } = fetchedWeight;
+  console.log(fetchedWeight);
   const timeUpdated = moment(weight?.updated_at || "").fromNow();
   const dateUpdated = moment(weight?.updated_at || "").format("LLL");
   const dateCreated = moment(weight?.created_at || "").format("LLL");
@@ -287,6 +288,7 @@ export default function WeightPage({
       [WeightCategory.SD]: { weightCategory: "Image Generation" },
       [WeightCategory.ZS]: { weightCategory: "Voice Designer" },
       [WeightCategory.VOCODER]: { weightCategory: "Vocoder" },
+      [WeightCategory.WF]: { weightCategory: "Workflow Config" },
     };
 
   let { weightCategory } = weightCategoryMap[weight.weight_category] || {
@@ -321,14 +323,11 @@ export default function WeightPage({
 
   switch (weight.weight_category) {
     case WeightCategory.TTS:
-      weightDetails = <DataTable data={voiceDetails} />;
-      break;
     case WeightCategory.VC:
-      weightDetails = <DataTable data={voiceDetails} />;
-      break;
     case WeightCategory.ZS:
       weightDetails = <DataTable data={voiceDetails} />;
       break;
+    case WeightCategory.WF:
     case WeightCategory.SD:
       weightDetails = <DataTable data={imageDetails} />;
       break;