diff --git a/serve.py b/serve.py index 921fe49..6c3103f 100644 --- a/serve.py +++ b/serve.py @@ -38,6 +38,103 @@ _use_rerank = False _stats = {"start_time": 0, "queries": 0, "avg_latency_ms": 0, "_latency_sum": 0} +HTML = """ + + + + + + Knowledge Search + + + +

Knowledge Search

+
+ + +
+
+ + + + +""" + def _reload(): """Load/reload embeddings from DB.""" @@ -66,9 +163,19 @@ def do_GET(self): elif path == "/reload": _reload() self._json_response({"status": "reloaded", "items": len(_rows)}) + elif path == "/": + self._html_response() else: self._json_response({"error": "not found"}, status=404) + def _html_response(self): + body = HTML.encode() + self.send_response(200) + self.send_header("Content-Type", "text/html; charset=utf-8") + self.send_header("Content-Length", str(len(body))) + self.end_headers() + self.wfile.write(body) + def _handle_search(self, params): q = params.get("q", [""])[0] if not q: