From 10de19bd055c1a5129fcfecc09bee2791f3a5ca9 Mon Sep 17 00:00:00 2001
From: Vladimir Mandic <mandic00@live.com>
Date: Sun, 16 Jun 2024 12:02:31 -0400
Subject: [PATCH] extra networks add link to source url

---
 CHANGELOG.md                 | 2 ++
 modules/ui_extra_networks.py | 9 ++++++++-
 2 files changed, 10 insertions(+), 1 deletion(-)

diff --git a/CHANGELOG.md b/CHANGELOG.md
index 2affef04b..42bf77ff8 100644
--- a/CHANGELOG.md
+++ b/CHANGELOG.md
@@ -19,6 +19,8 @@
 - **sd3** enable base LoRA support  
 - **sd3** simplified loading of model in single-file safetensors format  
   loading sd3 can now be performed fully offline  
+- extra networks: info display now contains link to source url if model if its known  
+  works for civitai and huggingface models  
 
 ### Fixes
 
diff --git a/modules/ui_extra_networks.py b/modules/ui_extra_networks.py
index 7a49ce660..bd5238cfb 100644
--- a/modules/ui_extra_networks.py
+++ b/modules/ui_extra_networks.py
@@ -812,15 +812,22 @@ def create_ui(container, button_parent, tabname, skip_indexing = False):
                     <tr><td>Preview Embedded</td><td>{item.preview.startswith('data:')}</td></tr>
                 '''
                 # desc = f'Name: {os.path.basename(item.name)}\nDescription: {item.description}\nPrompt: {item.prompt}\nNegative: {item.negative}\nExtra: {item.extra}\n'
+            if item.name.startswith('Diffusers'):
+                url = item.name.replace('Diffusers/', '')
+                url = f'<a href="https://huggingface.co/{url}" target="_blank">https://huggingface.co/models/{url}</a>' if url is not None else 'N/A'
+            else:
+                url = info.get('id', None) if info is not None else None
+                url = f'<a href="https://civitai.com/models/{url}" target="_blank">civitai.com/models/{url}</a>' if url is not None else 'N/A'
             text = f'''
                 <h2 style="border-bottom: 1px solid var(--button-primary-border-color); margin: 0em 0px 1em 0 !important">{item.name}</h2>
-                <table style="width: 100%; line-height: 1.3em;"><tbody>
+                <table style="width: 100%; line-height: 1.5em;"><tbody>
                     <tr><td>Type</td><td>{page.title}</td></tr>
                     <tr><td>Alias</td><td>{getattr(item, 'alias', 'N/A')}</td></tr>
                     <tr><td>Filename</td><td>{item.filename}</td></tr>
                     <tr><td>Hash</td><td>{getattr(item, 'hash', 'N/A')}</td></tr>
                     <tr><td>Size</td><td>{round(stat.st_size/1024/1024, 2) if stat is not None else 'N/A'} MB</td></tr>
                     <tr><td>Last modified</td><td>{datetime.fromtimestamp(stat.st_mtime) if stat is not None else 'N/A'}</td></tr>
+                    <tr><td>Source URL</td><td>{url}</td></tr>
                     <tr><td style="border-top: 1px solid var(--button-primary-border-color);"></td><td></td></tr>
                     {lora}
                     {model}