christopher commited on
Commit
963bc31
·
1 Parent(s): 40be1e4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -2
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 pos_id, (puzzle_row, move_idx) in matches[:100]:
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(matches), "results": results}
 
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}