Spaces:
Running
Running
Commit
·
963bc31
1
Parent(s):
40be1e4
Update app.py
Browse files
app.py
CHANGED
|
@@ -53,8 +53,13 @@ async def search(data: dict):
|
|
| 53 |
query_tokens = board_to_tokens(board)
|
| 54 |
matches = query_positions(index, metadata, query_tokens)
|
| 55 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 56 |
results = []
|
| 57 |
-
for
|
| 58 |
row = dset[puzzle_row]
|
| 59 |
positions = get_puzzle_positions(row['FEN'], row['Moves'])
|
| 60 |
matched_board = positions[move_idx]
|
|
@@ -71,4 +76,4 @@ async def search(data: dict):
|
|
| 71 |
"MatchedMove": move_idx
|
| 72 |
})
|
| 73 |
|
| 74 |
-
return {"count": len(
|
|
|
|
| 53 |
query_tokens = board_to_tokens(board)
|
| 54 |
matches = query_positions(index, metadata, query_tokens)
|
| 55 |
|
| 56 |
+
seen_puzzles = {}
|
| 57 |
+
for pos_id, (puzzle_row, move_idx) in matches:
|
| 58 |
+
if puzzle_row not in seen_puzzles:
|
| 59 |
+
seen_puzzles[puzzle_row] = (pos_id, move_idx)
|
| 60 |
+
|
| 61 |
results = []
|
| 62 |
+
for puzzle_row, (pos_id, move_idx) in seen_puzzles.items():
|
| 63 |
row = dset[puzzle_row]
|
| 64 |
positions = get_puzzle_positions(row['FEN'], row['Moves'])
|
| 65 |
matched_board = positions[move_idx]
|
|
|
|
| 76 |
"MatchedMove": move_idx
|
| 77 |
})
|
| 78 |
|
| 79 |
+
return {"count": len(results), "results": results}
|