mirror of
https://github.com/mudler/LocalAI.git
synced 2024-12-27 00:01:07 +00:00
a1634b219a
* initial pass: roll out bluemonday sanitization more widely Signed-off-by: Dave Lee <dave@gray101.com> * add one additional sanitize - the overall modelslist used by the docs site Signed-off-by: Dave Lee <dave@gray101.com> --------- Signed-off-by: Dave Lee <dave@gray101.com>
305 lines
12 KiB
Go
305 lines
12 KiB
Go
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"html/template"
|
|
"io/ioutil"
|
|
"os"
|
|
|
|
"github.com/microcosm-cc/bluemonday"
|
|
"gopkg.in/yaml.v3"
|
|
)
|
|
|
|
var modelPageTemplate string = `
|
|
<!DOCTYPE html>
|
|
<html>
|
|
<head>
|
|
<meta charset="UTF-8">
|
|
<meta name="viewport" content="width=device-width, initial-scale=1.0">
|
|
<title>LocalAI models</title>
|
|
<link href="https://cdnjs.cloudflare.com/ajax/libs/flowbite/2.3.0/flowbite.min.css" rel="stylesheet" />
|
|
<script src="https://cdn.jsdelivr.net/npm/vanilla-lazyload@19.1.3/dist/lazyload.min.js"></script>
|
|
|
|
<link
|
|
rel="stylesheet"
|
|
href="https://cdn.jsdelivr.net/gh/highlightjs/cdn-release@11.8.0/build/styles/default.min.css"
|
|
/>
|
|
<script
|
|
defer
|
|
src="https://cdn.jsdelivr.net/gh/highlightjs/cdn-release@11.8.0/build/highlight.min.js"
|
|
></script>
|
|
<script
|
|
defer
|
|
src="https://cdn.jsdelivr.net/npm/alpinejs@3.x.x/dist/cdn.min.js"
|
|
></script>
|
|
<script
|
|
defer
|
|
src="https://cdn.jsdelivr.net/npm/marked/marked.min.js"
|
|
></script>
|
|
<script
|
|
defer
|
|
src="https://cdn.jsdelivr.net/npm/dompurify@3.0.6/dist/purify.min.js"
|
|
></script>
|
|
|
|
<link href="/static/general.css" rel="stylesheet" />
|
|
<link href="https://fonts.googleapis.com/css2?family=Inter:wght@400;600;700&family=Roboto:wght@400;500&display=swap" rel="stylesheet">
|
|
<link
|
|
href="https://fonts.googleapis.com/css?family=Roboto:300,400,500,700,900&display=swap"
|
|
rel="stylesheet" />
|
|
<link
|
|
rel="stylesheet"
|
|
href="https://cdn.jsdelivr.net/npm/tw-elements/css/tw-elements.min.css" />
|
|
<script src="https://cdn.tailwindcss.com/3.3.0"></script>
|
|
<script>
|
|
tailwind.config = {
|
|
darkMode: "class",
|
|
theme: {
|
|
fontFamily: {
|
|
sans: ["Roboto", "sans-serif"],
|
|
body: ["Roboto", "sans-serif"],
|
|
mono: ["ui-monospace", "monospace"],
|
|
},
|
|
},
|
|
corePlugins: {
|
|
preflight: false,
|
|
},
|
|
};
|
|
</script>
|
|
<link rel="stylesheet" href="https://cdnjs.cloudflare.com/ajax/libs/font-awesome/6.1.1/css/all.min.css">
|
|
<script src="https://unpkg.com/htmx.org@1.9.12" integrity="sha384-ujb1lZYygJmzgSwoxRggbCHcjc0rB2XoQrxeTUQyRjrOnlCoYta87iKBWq3EsdM2" crossorigin="anonymous"></script>
|
|
</head>
|
|
|
|
<body class="bg-gray-900 text-gray-200">
|
|
<div class="flex flex-col min-h-screen">
|
|
|
|
<nav class="bg-gray-800 shadow-lg">
|
|
<div class="container mx-auto px-4 py-4">
|
|
<div class="flex items-center justify-between">
|
|
<div class="flex items-center">
|
|
<a href="/" class="text-white text-xl font-bold"><img src="https://github.com/mudler/LocalAI/assets/2420543/0966aa2a-166e-4f99-a3e5-6c915fc997dd" alt="LocalAI Logo" class="h-10 mr-3 border-2 border-gray-300 shadow rounded"></a>
|
|
<a href="/" class="text-white text-xl font-bold">LocalAI</a>
|
|
</div>
|
|
<!-- Menu button for small screens -->
|
|
<div class="lg:hidden">
|
|
<button id="menu-toggle" class="text-gray-400 hover:text-white focus:outline-none">
|
|
<i class="fas fa-bars fa-lg"></i>
|
|
</button>
|
|
</div>
|
|
<!-- Navigation links -->
|
|
<div class="hidden lg:flex lg:items-center lg:justify-end lg:flex-1 lg:w-0">
|
|
<a href="https://localai.io" class="text-gray-400 hover:text-white px-3 py-2 rounded" target="_blank" ><i class="fas fa-book-reader pr-2"></i> Documentation</a>
|
|
</div>
|
|
</div>
|
|
<!-- Collapsible menu for small screens -->
|
|
<div class="hidden lg:hidden" id="mobile-menu">
|
|
<div class="pt-4 pb-3 border-t border-gray-700">
|
|
|
|
<a href="https://localai.io" class="block text-gray-400 hover:text-white px-3 py-2 rounded mt-1" target="_blank" ><i class="fas fa-book-reader pr-2"></i> Documentation</a>
|
|
|
|
</div>
|
|
</div>
|
|
</div>
|
|
</nav>
|
|
|
|
<style>
|
|
.is-hidden {
|
|
display: none;
|
|
}
|
|
</style>
|
|
|
|
<div class="container mx-auto px-4 flex-grow">
|
|
|
|
<div class="models mt-12">
|
|
<h2 class="text-center text-3xl font-semibold text-gray-100">
|
|
LocalAI model gallery list </h2><br>
|
|
|
|
<h2 class="text-center text-3xl font-semibold text-gray-100">
|
|
|
|
🖼️ Available {{.AvailableModels}} models</i> <a href="https://localai.io/models/" target="_blank" >
|
|
<i class="fas fa-circle-info pr-2"></i>
|
|
</a></h2>
|
|
|
|
<h3>
|
|
Refer to the Model gallery <a href="https://localai.io/models/" target="_blank" ><i class="fas fa-circle-info pr-2"></i></a> for more information on how to use the models with LocalAI.<br>
|
|
|
|
You can install models with the CLI command <code>local-ai models install <model-name></code>. or by using the WebUI.
|
|
</h3>
|
|
|
|
<input class="form-control appearance-none block w-full mt-5 px-3 py-2 text-base font-normal text-gray-300 pb-2 mb-5 bg-gray-800 bg-clip-padding border border-solid border-gray-600 rounded transition ease-in-out m-0 focus:text-gray-300 focus:bg-gray-900 focus:border-blue-500 focus:outline-none" type="search"
|
|
id="searchbox" placeholder="Live search keyword..">
|
|
<div class="dark grid grid-cols-1 grid-rows-1 md:grid-cols-3 block rounded-lg shadow-secondary-1 dark:bg-surface-dark">
|
|
{{ range $_, $model := .Models }}
|
|
<div class="box me-4 mb-2 block rounded-lg bg-white shadow-secondary-1 dark:bg-gray-800 dark:bg-surface-dark dark:text-white text-surface pb-2">
|
|
<div>
|
|
{{ $icon := "https://upload.wikimedia.org/wikipedia/commons/6/65/No-Image-Placeholder.svg" }}
|
|
{{ if $model.Icon }}
|
|
{{ $icon = $model.Icon }}
|
|
{{ end }}
|
|
<div class="flex justify-center items-center">
|
|
<img data-src="{{ $icon }}" alt="{{$model.Name}}" class="rounded-t-lg max-h-48 max-w-96 object-cover mt-3 lazy">
|
|
</div>
|
|
<div class="p-6 text-surface dark:text-white">
|
|
<h5 class="mb-2 text-xl font-medium leading-tight">{{$model.Name}}</h5>
|
|
|
|
|
|
<p class="mb-4 text-base truncate">{{ $model.Description }}</p>
|
|
|
|
</div>
|
|
<div class="px-6 pt-4 pb-2">
|
|
|
|
<!-- Modal toggle -->
|
|
<button data-modal-target="{{ $model.Name}}-modal" data-modal-toggle="{{ $model.Name }}-modal" class="block text-white bg-blue-700 hover:bg-blue-800 focus:ring-4 focus:outline-none focus:ring-blue-300 font-medium rounded-lg text-sm px-5 py-2.5 text-center dark:bg-blue-600 dark:hover:bg-blue-700 dark:focus:ring-blue-800" type="button">
|
|
More info
|
|
</button>
|
|
|
|
<!-- Main modal -->
|
|
<div id="{{ $model.Name}}-modal" tabindex="-1" aria-hidden="true" class="hidden overflow-y-auto overflow-x-hidden fixed top-0 right-0 left-0 z-50 justify-center items-center w-full md:inset-0 h-[calc(100%-1rem)] max-h-full">
|
|
<div class="relative p-4 w-full max-w-2xl max-h-full">
|
|
<!-- Modal content -->
|
|
<div class="relative bg-white rounded-lg shadow dark:bg-gray-700">
|
|
<!-- Modal header -->
|
|
<div class="flex items-center justify-between p-4 md:p-5 border-b rounded-t dark:border-gray-600">
|
|
<h3 class="text-xl font-semibold text-gray-900 dark:text-white">
|
|
{{ $model.Name}}
|
|
</h3>
|
|
<button type="button" class="text-gray-400 bg-transparent hover:bg-gray-200 hover:text-gray-900 rounded-lg text-sm w-8 h-8 ms-auto inline-flex justify-center items-center dark:hover:bg-gray-600 dark:hover:text-white" data-modal-hide="{{$model.Name}}-modal">
|
|
<svg class="w-3 h-3" aria-hidden="true" xmlns="http://www.w3.org/2000/svg" fill="none" viewBox="0 0 14 14">
|
|
<path stroke="currentColor" stroke-linecap="round" stroke-linejoin="round" stroke-width="2" d="m1 1 6 6m0 0 6 6M7 7l6-6M7 7l-6 6"/>
|
|
</svg>
|
|
<span class="sr-only">Close modal</span>
|
|
</button>
|
|
</div>
|
|
<!-- Modal body -->
|
|
<div class="p-4 md:p-5 space-y-4">
|
|
<div class="flex justify-center items-center">
|
|
<img data-src="{{ $icon }}" alt="{{$model.Name}}" class="lazy rounded-t-lg max-h-48 max-w-96 object-cover mt-3">
|
|
</div>
|
|
|
|
<p class="text-base leading-relaxed text-gray-500 dark:text-gray-400">
|
|
{{ $model.Description }}
|
|
|
|
</p>
|
|
|
|
<p class="text-base leading-relaxed text-gray-500 dark:text-gray-400">
|
|
To install the model with the CLI, run: <br>
|
|
<code> local-ai models install {{$model.Name}} </code> <br>
|
|
|
|
<hr>
|
|
See also <a href="https://localai.io/models/" target="_blank" >
|
|
Installation <i class="fas fa-circle-info pr-2"></i>
|
|
</a> to see how to install models with the REST API.
|
|
</p>
|
|
|
|
<p class="text-base leading-relaxed text-gray-500 dark:text-gray-400">
|
|
<ul>
|
|
{{ range $_, $u := $model.URLs }}
|
|
<li><a href="{{ $u }}" target=_blank><i class="fa-solid fa-link"></i> {{ $u }}</a></li>
|
|
{{ end }}
|
|
</ul>
|
|
</p>
|
|
</div>
|
|
<!-- Modal footer -->
|
|
<div class="flex items-center p-4 md:p-5 border-t border-gray-200 rounded-b dark:border-gray-600">
|
|
<button data-modal-hide="{{ $model.Name}}-modal" type="button" class="py-2.5 px-5 ms-3 text-sm font-medium text-gray-900 focus:outline-none bg-white rounded-lg border border-gray-200 hover:bg-gray-100 hover:text-blue-700 focus:z-10 focus:ring-4 focus:ring-gray-100 dark:focus:ring-gray-700 dark:bg-gray-800 dark:text-gray-400 dark:border-gray-600 dark:hover:text-white dark:hover:bg-gray-700">Close</button>
|
|
</div>
|
|
</div>
|
|
</div>
|
|
</div>
|
|
|
|
|
|
</div>
|
|
</div>
|
|
</div>
|
|
{{ end }}
|
|
|
|
</div>
|
|
</div>
|
|
</div>
|
|
|
|
<script>
|
|
var lazyLoadInstance = new LazyLoad({
|
|
// Your custom settings go here
|
|
});
|
|
|
|
let cards = document.querySelectorAll('.box')
|
|
|
|
function liveSearch() {
|
|
let search_query = document.getElementById("searchbox").value;
|
|
|
|
//Use innerText if all contents are visible
|
|
//Use textContent for including hidden elements
|
|
for (var i = 0; i < cards.length; i++) {
|
|
if(cards[i].textContent.toLowerCase()
|
|
.includes(search_query.toLowerCase())) {
|
|
cards[i].classList.remove("is-hidden");
|
|
} else {
|
|
cards[i].classList.add("is-hidden");
|
|
}
|
|
}
|
|
}
|
|
|
|
//A little delay
|
|
let typingTimer;
|
|
let typeInterval = 500;
|
|
let searchInput = document.getElementById('searchbox');
|
|
|
|
searchInput.addEventListener('keyup', () => {
|
|
clearTimeout(typingTimer);
|
|
typingTimer = setTimeout(liveSearch, typeInterval);
|
|
});
|
|
</script>
|
|
|
|
</div>
|
|
|
|
<script src="https://cdnjs.cloudflare.com/ajax/libs/flowbite/2.3.0/flowbite.min.js"></script>
|
|
</body>
|
|
</html>
|
|
`
|
|
|
|
type GalleryModel struct {
|
|
Name string `json:"name" yaml:"name"`
|
|
URLs []string `json:"urls" yaml:"urls"`
|
|
Icon string `json:"icon" yaml:"icon"`
|
|
Description string `json:"description" yaml:"description"`
|
|
}
|
|
|
|
func main() {
|
|
// read the YAML file which contains the models
|
|
|
|
f, err := ioutil.ReadFile(os.Args[1])
|
|
if err != nil {
|
|
fmt.Println("Error reading file:", err)
|
|
return
|
|
}
|
|
|
|
models := []*GalleryModel{}
|
|
err = yaml.Unmarshal(f, &models)
|
|
if err != nil {
|
|
// write to stderr
|
|
os.Stderr.WriteString("Error unmarshaling YAML: " + err.Error() + "\n")
|
|
return
|
|
}
|
|
|
|
// Ensure that all arbitrary text content is sanitized before display
|
|
for i, m := range models {
|
|
models[i].Name = bluemonday.StrictPolicy().Sanitize(m.Name)
|
|
models[i].Description = bluemonday.StrictPolicy().Sanitize(m.Description)
|
|
}
|
|
|
|
// render the template
|
|
data := struct {
|
|
Models []*GalleryModel
|
|
AvailableModels int
|
|
}{
|
|
Models: models,
|
|
AvailableModels: len(models),
|
|
}
|
|
tmpl := template.Must(template.New("modelPage").Parse(modelPageTemplate))
|
|
|
|
err = tmpl.Execute(os.Stdout, data)
|
|
if err != nil {
|
|
fmt.Println("Error executing template:", err)
|
|
return
|
|
}
|
|
}
|