diff --git a/packages/components/src/api/_common/enums/WeightType.ts b/packages/components/src/api/_common/enums/WeightType.ts index 3866895148..246e7ca6b1 100644 --- a/packages/components/src/api/_common/enums/WeightType.ts +++ b/packages/components/src/api/_common/enums/WeightType.ts @@ -7,18 +7,5 @@ export enum WeightType { SVC = "so_vits_svc", LORA = "loRA", VALL_E = "vall_e", + NONE = "none", } - -export const weightTypeInfo = { - [WeightType.TT2]: { label: "TT2", color: "teal" }, - [WeightType.HIFIGAN_TT2]: { - label: "HifiGAN TT2", - color: "blue", - }, - [WeightType.RVCv2]: { label: "RVCv2", color: "orange" }, - [WeightType.SD_15]: { label: "SD 1.5", color: "green" }, - [WeightType.SDXL]: { label: "SDXL", color: "purple" }, - [WeightType.SVC]: { label: "SVC", color: "cyan" }, - [WeightType.LORA]: { label: "LoRA", color: "pink" }, - [WeightType.VALL_E]: { label: "VALL-E", color: "ultramarine" }, -}; diff --git a/packages/components/src/api/media_files/ListFeaturedMediaFiles.ts b/packages/components/src/api/media_files/ListFeaturedMediaFiles.ts new file mode 100644 index 0000000000..91585aff7c --- /dev/null +++ b/packages/components/src/api/media_files/ListFeaturedMediaFiles.ts @@ -0,0 +1,19 @@ +import MakeRequest from "../MakeRequest"; +import { MediaFile } from "./GetMedia"; + +export interface ListFeaturedMediaFilesRequest {} + +export interface ListFeaturedMediaFilesResponse { + success: boolean; + media_files: MediaFile[]; +} + +export const ListFeaturedMediaFiles = MakeRequest< + string, + ListFeaturedMediaFilesRequest, + ListFeaturedMediaFilesResponse, + {} +>({ + method: "GET", + routingFunction: () => `/v1/media_files/list_featured`, +}); diff --git a/packages/components/src/api/weights/GetWeight.ts b/packages/components/src/api/weights/GetWeight.ts index 3b30093fb3..3e12ab8b0d 100644 --- a/packages/components/src/api/weights/GetWeight.ts +++ b/packages/components/src/api/weights/GetWeight.ts @@ -26,10 +26,15 @@ export interface GetWeightRequest {} export interface GetWeightResponse { success: boolean; - media_file?: Weight; + weight?: Weight; } -export const GetWeight = MakeRequest({ +export const GetWeight = MakeRequest< + string, + GetWeightRequest, + GetWeightResponse, + {} +>({ method: "GET", routingFunction: (weightToken: string) => `/v1/weights/weight/${weightToken}`, -}); \ No newline at end of file +}); diff --git a/packages/fakeyou.com/src/components/common/Card/AudioCard/AudioCard.tsx b/packages/fakeyou.com/src/components/common/Card/AudioCard/AudioCard.tsx index f147520fd2..99557742a6 100644 --- a/packages/fakeyou.com/src/components/common/Card/AudioCard/AudioCard.tsx +++ b/packages/fakeyou.com/src/components/common/Card/AudioCard/AudioCard.tsx @@ -9,6 +9,7 @@ import FavoriteButton from "components/common/FavoriteButton"; import CreatorName from "../CreatorName"; import { faArrowRight } from "@fortawesome/pro-solid-svg-icons"; import Button from "components/common/Button"; +import useWeightTypeInfo from "hooks/useWeightTypeInfo/useWeightTypeInfo"; interface AudioCardProps { data: any; @@ -50,6 +51,9 @@ export default function AudioCard({ return true; // temporary, replace with like function }; + const { label: weightBadgeLabel, color: weightBadgeColor } = + useWeightTypeInfo(data.weights_type); + return ( {type === "media" && ( @@ -113,7 +117,7 @@ export default function AudioCard({
- +