feat(models-ui): minor visual enhancements (#2109)

Show image if present, URL, tags, and better display buttons

Signed-off-by: Ettore Di Giacinto <mudler@localai.io>
This commit is contained in:
Ettore Di Giacinto 2024-04-23 18:43:25 +02:00 committed by GitHub
parent 3411e072ca
commit d344daf129
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
3 changed files with 96 additions and 28 deletions

View file

@ -26,8 +26,9 @@ func RegisterUIRoutes(app *fiber.App,
models, _ := gallery.AvailableGalleryModels(appConfig.Galleries, appConfig.ModelPath)
summary := fiber.Map{
"Title": "LocalAI API - Models",
"Models": template.HTML(elements.ListModels(models)),
"Title": "LocalAI - Models",
"Models": template.HTML(elements.ListModels(models)),
"Repositories": appConfig.Galleries,
// "ApplicationConfig": appConfig,
}
@ -49,7 +50,10 @@ func RegisterUIRoutes(app *fiber.App,
filteredModels := []*gallery.GalleryModel{}
for _, m := range models {
if strings.Contains(m.Name, form.Search) {
if strings.Contains(m.Name, form.Search) ||
strings.Contains(m.Description, form.Search) ||
strings.Contains(m.Gallery.Name, form.Search) ||
strings.Contains(strings.Join(m.Tags, ","), form.Search) {
filteredModels = append(filteredModels, m)
}
}