From d96811a0c398b04caef1336881c8d30dc5d23b83 Mon Sep 17 00:00:00 2001
From: bennykok <itechbenny@gmail.com>
Date: Sun, 28 Jan 2024 22:08:31 +0800
Subject: [PATCH] fix: civitai picker

---
 web/src/components/custom-form/getUrl.tsx                | 3 ++-
 web/src/components/custom-form/model-picker-url-only.tsx | 3 ++-
 2 files changed, 4 insertions(+), 2 deletions(-)

diff --git a/web/src/components/custom-form/getUrl.tsx b/web/src/components/custom-form/getUrl.tsx
index 3d0081e..13624a4 100644
--- a/web/src/components/custom-form/getUrl.tsx
+++ b/web/src/components/custom-form/getUrl.tsx
@@ -27,7 +27,8 @@ export function mapModelsList(
           description: item.description,
           reference: "",
           filename: v.files[0].name,
-          url: v.files[0].downloadUrl,
+          // Quick hack to get the download url back as normal url
+          url: `https://civitai.com/models/${v.modelId}?modelVersionId=${v.id}`, //v.files[0].downloadUrl,
         } as z.infer<typeof Model>;
       });
     }),
diff --git a/web/src/components/custom-form/model-picker-url-only.tsx b/web/src/components/custom-form/model-picker-url-only.tsx
index 136f429..ff4e50d 100644
--- a/web/src/components/custom-form/model-picker-url-only.tsx
+++ b/web/src/components/custom-form/model-picker-url-only.tsx
@@ -61,7 +61,8 @@ function ModelPickerView({
 }: Pick<AutoFormInputComponentProps, "field" | "fieldProps">) {
   const customOverride = React.useMemo(() => {
     const customOnChange = (value: z.infer<typeof ModelList>) => {
-      field.onChange(value[0]?.url);
+      const model = value[0];
+      field.onChange(model?.url);
     };
     return {
       ...field,