diff --git a/serve.py b/serve.py index 6aecb8e..32e7894 100644 --- a/serve.py +++ b/serve.py @@ -10,6 +10,7 @@ ideas: import os import re import time +import math from random import shuffle import numpy as np @@ -262,6 +263,7 @@ def main(): pids, scores = [pids[i] for i in keep], [scores[i] for i in keep] # crop the number of results to RET_NUM, and paginate + total_pages = math.ceil(len(pids) / RET_NUM) try: page_number = max(1, int(opt_page_number)) except ValueError: @@ -297,6 +299,7 @@ def main(): context['gvars']['search_query'] = opt_q context['gvars']['svm_c'] = str(C) context['gvars']['page_number'] = str(page_number) + context['gvars']['total_pages'] = str(total_pages) return render_template('index.html', **context) @app.route('/inspect', methods=['GET']) diff --git a/templates/index.html b/templates/index.html index 0c814b7..0282b28 100644 --- a/templates/index.html +++ b/templates/index.html @@ -102,9 +102,14 @@ var move_page = function(int_offset) {