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