diff --git a/serve.py b/serve.py
index 9a32879..2e4d937 100644
--- a/serve.py
+++ b/serve.py
@@ -38,6 +38,72 @@
_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 +132,17 @@ 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):
+ self.send_response(200)
+ self.send_header("Content-Type", "text/html")
+ self.end_headers()
+ self.wfile.write(HTML.encode())
+
def _handle_search(self, params):
q = params.get("q", [""])[0]
if not q:
@@ -162,4 +236,4 @@ def main():
if __name__ == "__main__":
- main()
+ main()
\ No newline at end of file