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: