From c3b8e0cfa8d70fe57d73a822ef9d902177dcfb51 Mon Sep 17 00:00:00 2001
From: GeorgFrobenius <mplaughlin@live.com>
Date: Wed, 16 Oct 2024 12:53:17 -0400
Subject: [PATCH] Model Availability and Publish Date (#348)

---
 scripts/civitai_api.py | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/scripts/civitai_api.py b/scripts/civitai_api.py
index 0ddefd9..7143584 100644
--- a/scripts/civitai_api.py
+++ b/scripts/civitai_api.py
@@ -637,6 +637,9 @@ def update_model_info(model_string=None, model_version=None, only_html=False, in
                             selected_version = model
                             break
                     
+                model_availability = selected_version.get('availability', 'Unknown')
+                model_date_published = selected_version.get('publishedAt', '').split('T')[0]
+
                 if selected_version['trainedWords']:
                     output_training = ",".join(selected_version['trainedWords'])
                     output_training = re.sub(r'<[^>]*:[^>]*>', '', output_training)
@@ -818,6 +821,10 @@ def update_model_info(model_string=None, model_version=None, only_html=False, in
                             <dd>{escape(str(model_version))}</dd>
                             <dt>Base Model</dt>
                             <dd>{escape(str(output_basemodel))}</dd>
+                            <dt>Published</dt>
+                            <dd>{model_date_published}</dd>
+                            <dt>Availability</dt>
+                            <dd>{model_availability}</dd>
                             <dt>CivitAI Tags</dt>
                             <dd>
                                 <div class="civitai-tags-container">