cf21ba4516
Phase 1 (Foundation): - Project restructure (presenton-main → backend/ + frontend/) - Database schema (8 new models, Alembic config, seed script) - Auth (Azure AD SSO + dev bypass, JWT sessions, AuthMiddleware) - RBAC (access_service, rbac_middleware, admin routers) - Audit logging (fire-and-forget, AuditMiddleware, admin router) - i18n (react-i18next with 5 namespace files) Phase 2 (Admin Panel & Client Management): - Admin panel shell (sidebar layout, role guard, 12 pages) - Redux admin slice with 18 async thunks - User management (role changes, deactivation) - Client management (CRUD, brand config, team management) - Brand config editor (colors, fonts, logos, voice rules) - Master deck upload & parser (PPTX → HTML → React pipeline) - Audit log viewer with filters and CSV/JSON export Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
56 lines
2 KiB
Python
56 lines
2 KiB
Python
import asyncio
|
|
import json
|
|
import chromadb
|
|
from chromadb.config import Settings
|
|
from chromadb.utils.embedding_functions import ONNXMiniLM_L6_V2
|
|
|
|
|
|
class IconFinderService:
|
|
def __init__(self):
|
|
self.collection_name = "icons"
|
|
self.client = chromadb.PersistentClient(
|
|
path="chroma", settings=Settings(anonymized_telemetry=False)
|
|
)
|
|
print("Initializing icons collection...")
|
|
self._initialize_icons_collection()
|
|
print("Icons collection initialized.")
|
|
|
|
def _initialize_icons_collection(self):
|
|
self.embedding_function = ONNXMiniLM_L6_V2()
|
|
self.embedding_function.DOWNLOAD_PATH = "chroma/models"
|
|
self.embedding_function._download_model_if_not_exists()
|
|
try:
|
|
self.collection = self.client.get_collection(
|
|
self.collection_name, embedding_function=self.embedding_function
|
|
)
|
|
except Exception:
|
|
with open("assets/icons.json", "r") as f:
|
|
icons = json.load(f)
|
|
|
|
documents = []
|
|
ids = []
|
|
|
|
for i, each in enumerate(icons["icons"]):
|
|
if each["name"].split("-")[-1] == "bold":
|
|
doc_text = f"{each['name']} {each['tags']}"
|
|
documents.append(doc_text)
|
|
ids.append(each["name"])
|
|
|
|
if documents:
|
|
self.collection = self.client.create_collection(
|
|
name=self.collection_name,
|
|
embedding_function=self.embedding_function,
|
|
metadata={"hnsw:space": "cosine"},
|
|
)
|
|
self.collection.add(documents=documents, ids=ids)
|
|
|
|
async def search_icons(self, query: str, k: int = 1):
|
|
result = await asyncio.to_thread(
|
|
self.collection.query,
|
|
query_texts=[query],
|
|
n_results=k,
|
|
)
|
|
return [f"/static/icons/bold/{each}.svg" for each in result["ids"][0]]
|
|
|
|
|
|
ICON_FINDER_SERVICE = IconFinderService()
|