From 86e587eba26971cd775d52a3c97762cc38ac3143 Mon Sep 17 00:00:00 2001
From: hayden <hayden.mike.zhang@gmail.com>
Date: Sun, 27 Oct 2024 21:23:10 +0800
Subject: [PATCH] feat: Add more information in description metadata

---
 src/hooks/download.ts | 25 +++++++++++++++++++------
 1 file changed, 19 insertions(+), 6 deletions(-)

diff --git a/src/hooks/download.ts b/src/hooks/download.ts
index 2da8fab..aa19be9 100644
--- a/src/hooks/download.ts
+++ b/src/hooks/download.ts
@@ -182,9 +182,20 @@ class Civitai extends ModelSearch {
               pathIndex: 0,
               description: [
                 '---',
-                `website: Civitai`,
-                ``,
-                `modelPage: https://civitai.com/models/${modelId}?modelVersionId=${version.id}`,
+                ...[
+                  `website: Civitai`,
+                  `modelPage: https://civitai.com/models/${modelId}?modelVersionId=${version.id}`,
+                  `author: ${resData.creator?.username}`,
+                  version.baseModel && `baseModel: ${version.baseModel}`,
+                  file.hashes && `hashes:`,
+                  ...Object.entries(file.hashes ?? {}).map(
+                    ([key, value]) => `  ${key}: ${value}`,
+                  ),
+                  file.metadata && `metadata:`,
+                  ...Object.entries(file.metadata ?? {}).map(
+                    ([key, value]) => `  ${key}: ${value}`,
+                  ),
+                ].filter(Boolean),
                 '---',
                 '',
                 '# Trigger Words',
@@ -279,9 +290,11 @@ class Huggingface extends ModelSearch {
             pathIndex: 0,
             description: [
               '---',
-              `website: HuggingFace`,
-              `author: ${resData.author}`,
-              `modelPage: https://huggingface.co/${modelId}`,
+              ...[
+                `website: HuggingFace`,
+                `modelPage: https://huggingface.co/${modelId}`,
+                `author: ${resData.author}`,
+              ].filter(Boolean),
               '---',
               '',
               '# Trigger Words',