Update app.py

app.py

@@ -5,6 +5,7 @@ from fastapi.templating import Jinja2Templates
 import chess
 from datasets import load_dataset
 import pickle
+import time
 
 
 def board_to_tokens(board):
@@ -47,8 +48,10 @@ def read_root(request: Request):
     return templates.TemplateResponse("index.html", {"request": request})
 
 
+
 @app.post("/search")
 async def search(data: dict):
+    start = time.time()
     board = chess.Board(data['fen'])
     query_tokens = board_to_tokens(board)
     matches = query_positions(index, metadata, query_tokens)
@@ -76,4 +79,5 @@ async def search(data: dict):
             "MatchedMove": move_idx
         })
     
-    return {"count": len(results), "results": results}
+    elapsed_ms = (time.time() - start) * 1000
+    return {"count": len(results), "results": results, "time_ms": elapsed_ms}