dearwolf/LocalAI
1
0
Fork 0
mirror of https://github.com/mudler/LocalAI.git synced 2025-05-20 10:35:01 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 21

feat(ui): do also filter tts and image models (#4871)

Browse source 
Signed-off-by: Ettore Di Giacinto <mudler@localai.io>
This commit is contained in:
Ettore Di Giacinto 2025-02-20 15:02:18 +01:00 • committed by GitHub
parent b993780a3b
commit 25bee71bb8
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
4 changed files with 94 additions and 61 deletions
Download patch file Download diff file Expand all files Collapse all files

105
core/http/routes/ui.go
View file

@ -327,23 +327,31 @@ func RegisterUIRoutes(app *fiber.App,
}) })
app.Get("/chat/", func(c *fiber.Ctx) error { app.Get("/chat/", func(c *fiber.Ctx) error {
allModels, _ := services.ListModels(cl, ml, config.NoFilterFn, services.SKIP_IF_CONFIGURED)
backendConfigs := cl.GetAllBackendConfigs() backendConfigs := cl.GetAllBackendConfigs()
modelsWithoutConfig, _ := services.ListModels(cl, ml, config.NoFilterFn, services.LOOSE_ONLY) modelsWithoutConfig, _ := services.ListModels(cl, ml, config.NoFilterFn, services.LOOSE_ONLY)
if len(allModels) == 0 { if len(backendConfigs)+len(modelsWithoutConfig) == 0 {
// If no model is available redirect to the index which suggests how to install models // If no model is available redirect to the index which suggests how to install models
return c.Redirect(utils.BaseURL(c)) return c.Redirect(utils.BaseURL(c))
} }
modelThatCanBeUsed := ""
title := "LocalAI - Chat"
for _, b := range backendConfigs {
if b.HasUsecases(config.FLAG_CHAT) {
modelThatCanBeUsed = b.Name
title = "LocalAI - Chat with " + modelThatCanBeUsed
break
}
}
summary := fiber.Map{ summary := fiber.Map{
"Title": "LocalAI - Chat with " + allModels[0], "Title": title,
"BaseURL": utils.BaseURL(c), "BaseURL": utils.BaseURL(c),
"AllModels": allModels,
"ModelsWithoutConfig": modelsWithoutConfig, "ModelsWithoutConfig": modelsWithoutConfig,
"ModelsConfig": backendConfigs, "ModelsConfig": backendConfigs,
"Model": allModels[0], "Model": modelThatCanBeUsed,
"Version": internal.PrintableVersion(), "Version": internal.PrintableVersion(),
"IsP2PEnabled": p2p.IsP2PEnabled(), "IsP2PEnabled": p2p.IsP2PEnabled(),
} }
@ -354,7 +362,6 @@ func RegisterUIRoutes(app *fiber.App,
// Show the Chat page // Show the Chat page
app.Get("/chat/:model", func(c *fiber.Ctx) error { app.Get("/chat/:model", func(c *fiber.Ctx) error {
allModels, _ := services.ListModels(cl, ml, config.NoFilterFn, services.SKIP_IF_CONFIGURED)
backendConfigs := cl.GetAllBackendConfigs() backendConfigs := cl.GetAllBackendConfigs()
modelsWithoutConfig, _ := services.ListModels(cl, ml, config.NoFilterFn, services.LOOSE_ONLY) modelsWithoutConfig, _ := services.ListModels(cl, ml, config.NoFilterFn, services.LOOSE_ONLY)
@ -363,7 +370,6 @@ func RegisterUIRoutes(app *fiber.App,
"BaseURL": utils.BaseURL(c), "BaseURL": utils.BaseURL(c),
"ModelsConfig": backendConfigs, "ModelsConfig": backendConfigs,
"ModelsWithoutConfig": modelsWithoutConfig, "ModelsWithoutConfig": modelsWithoutConfig,
"AllModels": allModels,
"Model": c.Params("model"), "Model": c.Params("model"),
"Version": internal.PrintableVersion(), "Version": internal.PrintableVersion(),
"IsP2PEnabled": p2p.IsP2PEnabled(), "IsP2PEnabled": p2p.IsP2PEnabled(),
@ -375,14 +381,16 @@ func RegisterUIRoutes(app *fiber.App,
app.Get("/text2image/:model", func(c *fiber.Ctx) error { app.Get("/text2image/:model", func(c *fiber.Ctx) error {
backendConfigs := cl.GetAllBackendConfigs() backendConfigs := cl.GetAllBackendConfigs()
modelsWithoutConfig, _ := services.ListModels(cl, ml, config.NoFilterFn, services.LOOSE_ONLY)
summary := fiber.Map{ summary := fiber.Map{
"Title": "LocalAI - Generate images with " + c.Params("model"), "Title": "LocalAI - Generate images with " + c.Params("model"),
"BaseURL": utils.BaseURL(c), "BaseURL": utils.BaseURL(c),
"ModelsConfig": backendConfigs, "ModelsConfig": backendConfigs,
"Model": c.Params("model"), "ModelsWithoutConfig": modelsWithoutConfig,
"Version": internal.PrintableVersion(), "Model": c.Params("model"),
"IsP2PEnabled": p2p.IsP2PEnabled(), "Version": internal.PrintableVersion(),
"IsP2PEnabled": p2p.IsP2PEnabled(),
} }
// Render index // Render index
@ -390,21 +398,33 @@ func RegisterUIRoutes(app *fiber.App,
}) })
app.Get("/text2image/", func(c *fiber.Ctx) error { app.Get("/text2image/", func(c *fiber.Ctx) error {
backendConfigs := cl.GetAllBackendConfigs() backendConfigs := cl.GetAllBackendConfigs()
modelsWithoutConfig, _ := services.ListModels(cl, ml, config.NoFilterFn, services.LOOSE_ONLY)
if len(backendConfigs) == 0 { if len(backendConfigs)+len(modelsWithoutConfig) == 0 {
// If no model is available redirect to the index which suggests how to install models // If no model is available redirect to the index which suggests how to install models
return c.Redirect(utils.BaseURL(c)) return c.Redirect(utils.BaseURL(c))
} }
modelThatCanBeUsed := ""
title := "LocalAI - Generate images"
for _, b := range backendConfigs {
if b.HasUsecases(config.FLAG_IMAGE) {
modelThatCanBeUsed = b.Name
title = "LocalAI - Generate images with " + modelThatCanBeUsed
break
}
}
summary := fiber.Map{ summary := fiber.Map{
"Title": "LocalAI - Generate images with " + backendConfigs[0].Name, "Title": title,
"BaseURL": utils.BaseURL(c), "BaseURL": utils.BaseURL(c),
"ModelsConfig": backendConfigs, "ModelsConfig": backendConfigs,
"Model": backendConfigs[0].Name, "ModelsWithoutConfig": modelsWithoutConfig,
"Version": internal.PrintableVersion(), "Model": modelThatCanBeUsed,
"IsP2PEnabled": p2p.IsP2PEnabled(), "Version": internal.PrintableVersion(),
"IsP2PEnabled": p2p.IsP2PEnabled(),
} }
// Render index // Render index
@ -413,14 +433,16 @@ func RegisterUIRoutes(app *fiber.App,
app.Get("/tts/:model", func(c *fiber.Ctx) error { app.Get("/tts/:model", func(c *fiber.Ctx) error {
backendConfigs := cl.GetAllBackendConfigs() backendConfigs := cl.GetAllBackendConfigs()
modelsWithoutConfig, _ := services.ListModels(cl, ml, config.NoFilterFn, services.LOOSE_ONLY)
summary := fiber.Map{ summary := fiber.Map{
"Title": "LocalAI - Generate images with " + c.Params("model"), "Title": "LocalAI - Generate images with " + c.Params("model"),
"BaseURL": utils.BaseURL(c), "BaseURL": utils.BaseURL(c),
"ModelsConfig": backendConfigs, "ModelsConfig": backendConfigs,
"Model": c.Params("model"), "ModelsWithoutConfig": modelsWithoutConfig,
"Version": internal.PrintableVersion(), "Model": c.Params("model"),
"IsP2PEnabled": p2p.IsP2PEnabled(), "Version": internal.PrintableVersion(),
"IsP2PEnabled": p2p.IsP2PEnabled(),
} }
// Render index // Render index
@ -428,21 +450,32 @@ func RegisterUIRoutes(app *fiber.App,
}) })
app.Get("/tts/", func(c *fiber.Ctx) error { app.Get("/tts/", func(c *fiber.Ctx) error {
backendConfigs := cl.GetAllBackendConfigs() backendConfigs := cl.GetAllBackendConfigs()
modelsWithoutConfig, _ := services.ListModels(cl, ml, config.NoFilterFn, services.LOOSE_ONLY)
if len(backendConfigs) == 0 { if len(backendConfigs)+len(modelsWithoutConfig) == 0 {
// If no model is available redirect to the index which suggests how to install models // If no model is available redirect to the index which suggests how to install models
return c.Redirect(utils.BaseURL(c)) return c.Redirect(utils.BaseURL(c))
} }
modelThatCanBeUsed := ""
title := "LocalAI - Generate audio"
for _, b := range backendConfigs {
if b.HasUsecases(config.FLAG_CHAT) {
modelThatCanBeUsed = b.Name
title = "LocalAI - Generate audio with " + modelThatCanBeUsed
break
}
}
summary := fiber.Map{ summary := fiber.Map{
"Title": "LocalAI - Generate audio with " + backendConfigs[0].Name, "Title": title,
"BaseURL": utils.BaseURL(c), "BaseURL": utils.BaseURL(c),
"ModelsConfig": backendConfigs, "ModelsConfig": backendConfigs,
"Model": backendConfigs[0].Name, "ModelsWithoutConfig": modelsWithoutConfig,
"IsP2PEnabled": p2p.IsP2PEnabled(), "Model": modelThatCanBeUsed,
"Version": internal.PrintableVersion(), "IsP2PEnabled": p2p.IsP2PEnabled(),
"Version": internal.PrintableVersion(),
} }
// Render index // Render index

18
core/http/views/chat.html
View file

@ -44,7 +44,7 @@ SOFTWARE.
<div class="flex items-center justify-between"> <div class="flex items-center justify-between">
<h1 class="text-lg font-semibold"> <i class="fa-solid fa-comments"></i> Chat with {{.Model}} <a href="browse?term={{.Model}}" ><i class="fas fa-brain pr-2"></i></a> <a href="https://localai.io/features/text-generation/" target="_blank" > <h1 class="text-lg font-semibold"> <i class="fa-solid fa-comments"></i> Chat {{ if .Model }} with {{.Model}} {{ end }} <a href="browse?term={{.Model}}" ><i class="fas fa-brain pr-2"></i></a> <a href="https://localai.io/features/text-generation/" target="_blank" >
<i class="fas fa-circle-info pr-2"></i> <i class="fas fa-circle-info pr-2"></i>
</a></h1> </a></h1>
<div x-show="component === 'menu'" id="menu"> <div x-show="component === 'menu'" id="menu">
@ -101,22 +101,14 @@ SOFTWARE.
{{ $model:=.Model}} {{ $model:=.Model}}
{{ range .ModelsConfig }} {{ range .ModelsConfig }}
{{ $cfg := . }} {{ $cfg := . }}
{{ if eq .Name $model }} {{ range .KnownUsecaseStrings }}
<option value="chat/{{.Name}}" selected class="bg-gray-700 text-white">{{.Name}}</option> {{ if eq . "FLAG_CHAT" }}
{{ else }} <option value="chat/{{$cfg.Name}}" {{ if eq $cfg.Name $model }} selected {{end}} class="bg-gray-700 text-white">{{$cfg.Name}}</option>
{{ range .KnownUsecaseStrings }}
{{ if eq . "FLAG_CHAT" }}
<option value="chat/{{$cfg.Name}}" class="bg-gray-700 text-white">{{$cfg.Name}}</option>
{{ end }}
{{ end }} {{ end }}
{{ end }} {{ end }}
{{ end }} {{ end }}
{{ range .ModelsWithoutConfig }} {{ range .ModelsWithoutConfig }}
{{ if eq . $model }} <option value="chat/{{.}}" {{ if eq . $model }} selected {{ end }} class="bg-gray-700 text-white">{{.}}</option>
<option value="chat/{{.}}" selected class="bg-gray-700 text-white">{{.}}</option>
{{ else }}
<option value="chat/{{.}}" class="bg-gray-700 text-white">{{.}}</option>
{{ end }}
{{end}} {{end}}
</select> </select>
</div> </div>

16
core/http/views/text2image.html
View file

@ -13,7 +13,7 @@
<div class="mt-12"> <div class="mt-12">
<div class="flex items-center justify-center text-center pb-2"> <div class="flex items-center justify-center text-center pb-2">
<span class="text-3xl font-semibold text-gray-100"> <span class="text-3xl font-semibold text-gray-100">
🖼️ Text to Image 🖼️ Text to Image {{ if .Model }} (using {{.Model}}) {{ end }}
<a href="https://localai.io/features/image-generation" target="_blank" > <a href="https://localai.io/features/image-generation" target="_blank" >
<i class="fas fa-circle-info pr-2"></i> <i class="fas fa-circle-info pr-2"></i>
</a> </a>
@ -49,12 +49,16 @@
<option value="" disabled class="text-gray-400" >Select a model</option> <option value="" disabled class="text-gray-400" >Select a model</option>
{{ $model:=.Model}} {{ $model:=.Model}}
{{ range .ModelsConfig }} {{ range .ModelsConfig }}
{{ if eq .Name $model }} {{ $cfg := . }}
<option value="text2image/{{.Name}}" selected class="bg-gray-700 text-white">{{.Name}}</option> {{ range .KnownUsecaseStrings }}
{{ else }} {{ if eq . "FLAG_IMAGE" }}
<option value="text2image/{{.Name}}" class="bg-gray-700 text-white">{{.Name}}</option> <option value="text2image/{{$cfg.Name}}" {{ if eq $cfg.Name $model }} selected {{end}} class="bg-gray-700 text-white">{{$cfg.Name}}</option>
{{ end }} {{ end }}
{{ end }}
{{ end }} {{ end }}
{{ range .ModelsWithoutConfig }}
<option value="text2image/{{.}}" {{ if eq . $model }} selected {{ end }} class="bg-gray-700 text-white">{{.}}</option>
{{end}}
</select> </select>
</div> </div>

16
core/http/views/tts.html
View file

@ -11,7 +11,7 @@
<div class="mt-12"> <div class="mt-12">
<div class="flex items-center justify-center text-center pb-2"> <div class="flex items-center justify-center text-center pb-2">
<span class="text-3xl font-semibold text-gray-100"> <span class="text-3xl font-semibold text-gray-100">
<i class="fa-solid fa-music"></i> Text to speech/audio <i class="fa-solid fa-music"></i> Text to speech/audio {{ if .Model }} (using {{.Model}}) {{ end }}
<a href="https://localai.io/features/text-to-audio/" target="_blank" > <a href="https://localai.io/features/text-to-audio/" target="_blank" >
<i class="fas fa-circle-info pr-2"></i> <i class="fas fa-circle-info pr-2"></i>
</a> </a>
@ -46,12 +46,16 @@
<option value="" disabled class="text-gray-400" >Select a model</option> <option value="" disabled class="text-gray-400" >Select a model</option>
{{ $model:=.Model}} {{ $model:=.Model}}
{{ range .ModelsConfig }} {{ range .ModelsConfig }}
{{ if eq .Name $model }} {{ $cfg := . }}
<option value="tts/{{.Name}}" selected class="bg-gray-700 text-white">{{.Name}}</option> {{ range .KnownUsecaseStrings }}
{{ else }} {{ if eq . "FLAG_TTS" }}
<option value="tts/{{.Name}}" class="bg-gray-700 text-white">{{.Name}}</option> <option value="tts/{{$cfg.Name}}" {{ if eq $cfg.Name $model }} selected {{end}} class="bg-gray-700 text-white">{{$cfg.Name}}</option>
{{ end }} {{ end }}
{{ end }}
{{ end }} {{ end }}
{{ range .ModelsWithoutConfig }}
<option value="tts/{{.}}" {{ if eq . $model }} selected {{ end }} class="bg-gray-700 text-white">{{.}}</option>
{{end}}
</select> </select>
</div> </div>