From 345f84f1354f359512d20c35b83e09d475512e7b Mon Sep 17 00:00:00 2001
From: Ettore Di Giacinto <mudler@localai.io>
Date: Thu, 27 Feb 2025 19:11:47 +0100
Subject: [PATCH] fix(ui): not all models comes from gallery

Signed-off-by: Ettore Di Giacinto <mudler@localai.io>
---
 core/http/views/chat.html | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/core/http/views/chat.html b/core/http/views/chat.html
index b9a40687..3486e101 100644
--- a/core/http/views/chat.html
+++ b/core/http/views/chat.html
@@ -229,8 +229,10 @@ SOFTWARE.
             <i class="fa-solid fa-comments mr-2"></i>
             {{ if $model }}
             {{ $galleryConfig:= index $allGalleryConfigs $model}}
+            {{ if $galleryConfig }}
             {{ if $galleryConfig.Icon }}<img src="{{$galleryConfig.Icon}}" class="rounded-lg w-8 h-8 mr-2">{{end}}
             {{ end }}
+            {{ end }}
             <h1 class="text-lg font-semibold">
               Chat {{ if .Model }} with {{.Model}} {{ end }}
             </h1>
@@ -262,7 +264,9 @@ SOFTWARE.
                 </template>
                 <template x-if="message.role != 'user'">
                   <div class="flex items-center space-x-2">
+                    {{ if $galleryConfig }}
                     {{ if $galleryConfig.Icon }}<img src="{{$galleryConfig.Icon}}" class="rounded-lg mt-2 max-w-8 max-h-8">{{end}}
+                    {{ end }}
                     <div class="flex flex-col flex-1">
                       <span class="text-xs font-semibold text-gray-400">{{if .Model}}{{.Model}}{{else}}Assistant{{end}}</span>
                       <div class="flex-1 text-white flex items-center space-x-2">