christopher commited on
Commit
dde8525
·
verified ·
1 Parent(s): 39b62b5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -4
app.py CHANGED
@@ -11,11 +11,12 @@ def find_puzzles(pgn_text):
11
  return None
12
  engine = open_engine("/usr/games/stockfish", threads=2)
13
  try:
14
- puzzle = Generator(engine).analyze_game(game, tier=10)
15
- if not puzzle:
16
  return None
17
- puzzle.tags = cook(puzzle)
18
- return puzzle.to_dict()
 
19
  finally:
20
  engine.close()
21
 
 
11
  return None
12
  engine = open_engine("/usr/games/stockfish", threads=2)
13
  try:
14
+ puzzles = Generator(engine).analyze_game(game, tier=10, all_puzzles=True)
15
+ if not puzzles:
16
  return None
17
+ for puzzle in puzzles:
18
+ puzzle.tags = cook(puzzle, engine=engine)
19
+ return [puzzle.to_dict() for puzzle in puzzles]
20
  finally:
21
  engine.close()
22