Update app.py

app.py

@@ -25,6 +25,18 @@ def get_puzzle_positions(fen, moves_uci):
     
     return positions
 
+def load_index(path='chess_index.pkl'):
+    with open(path, 'rb') as f: data = pickle.load(f)
+    return data['index'], data['metadata']
+
+def query_positions(index, metadata, query_tokens):
+    result = index[query_tokens[0]].copy() if query_tokens[0] in index else BitMap()
+    for token in query_tokens[1:]:
+        if token in index: result &= index[token]
+        else: return BitMap()
+            
+    return [(pos_id, metadata[pos_id]) for pos_id in result]
+
 app = FastAPI()
 app.mount("/static", StaticFiles(directory="static"), name="static")
 templates = Jinja2Templates(directory="templates")