From 3bf2e9d065d5b155b0255453935ebafbc6cbf2e9 Mon Sep 17 00:00:00 2001
From: Ettore Di Giacinto <mudler@users.noreply.github.com>
Date: Thu, 27 Feb 2025 10:52:19 +0100
Subject: [PATCH] fix(ui): not all models have an Icon (#4913)

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

diff --git a/core/http/views/chat.html b/core/http/views/chat.html
index 59414fe4..b9a40687 100644
--- a/core/http/views/chat.html
+++ b/core/http/views/chat.html
@@ -94,7 +94,7 @@ SOFTWARE.
           <!-- Model info -->
           <div class="space-y-2">
             <div class="flex items-center">
-              <img src="{{$galleryConfig.Icon}}" class="rounded-lg w-8 h-8 mr-2">
+              {{ if $galleryConfig.Icon }}<img src="{{$galleryConfig.Icon}}" class="rounded-lg w-8 h-8 mr-2">{{end}}
               <h3 class="text-md font-medium">{{ $model }}</h3>
             </div>
             <button data-twe-ripple-init data-twe-ripple-color="light" class="w-full text-left flex items-center px-3 py-2 text-xs rounded text-white bg-gray-700 hover:bg-gray-600 transition-colors" data-modal-target="model-info-modal" data-modal-toggle="model-info-modal">
@@ -229,7 +229,7 @@ SOFTWARE.
             <i class="fa-solid fa-comments mr-2"></i>
             {{ if $model }}
             {{ $galleryConfig:= index $allGalleryConfigs $model}}
-            <img src="{{$galleryConfig.Icon}}" class="rounded-lg w-8 h-8 mr-2">
+            {{ if $galleryConfig.Icon }}<img src="{{$galleryConfig.Icon}}" class="rounded-lg w-8 h-8 mr-2">{{end}}
             {{ end }}
             <h1 class="text-lg font-semibold">
               Chat {{ if .Model }} with {{.Model}} {{ end }}
@@ -262,7 +262,7 @@ SOFTWARE.
                 </template>
                 <template x-if="message.role != 'user'">
                   <div class="flex items-center space-x-2">
-                    <img src="{{$galleryConfig.Icon}}" class="rounded-lg mt-2 max-w-8 max-h-8">
+                    {{ if $galleryConfig.Icon }}<img src="{{$galleryConfig.Icon}}" class="rounded-lg mt-2 max-w-8 max-h-8">{{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">
@@ -355,7 +355,7 @@ SOFTWARE.
           <!-- Body -->
           <div class="p-4 md:p-5 space-y-4">
             <div class="flex justify-center items-center">
-              <img class="lazy rounded-t-lg max-h-48 max-w-96 object-cover mt-3 entered loaded" src="{{$galleryConfig.Icon}}" loading="lazy"/>
+              {{ if $galleryConfig.Icon }}<img class="lazy rounded-t-lg max-h-48 max-w-96 object-cover mt-3 entered loaded" src="{{$galleryConfig.Icon}}" loading="lazy"/>{{end}}
             </div>
             <p class="text-base leading-relaxed text-gray-500 dark:text-gray-400">{{ $galleryConfig.Description }}</p>
             <hr>