Minor improvement to pagination

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'])

templates/index.html

@@ -102,9 +102,13 @@ var move_page = function(int_offset) {
 
 <!-- links to previous and next pages -->
 <div id="pagination">
+    {% if gvars.page_number|int > 1 %}
     <span id="link-prev-page" onclick='move_page(-1);'>prev</span>
-    <span>current page: {{ gvars.page_number }} </span>
+    {% endif %}
+    <span>page {{ gvars.page_number }} of {{ gvars.total_pages }}</span>
+    {% if gvars.page_number|int < gvars.total_pages|int %}
     <span id="link-next-page" onclick='move_page(1);'>next</span>
+    {% endif %}
 </div>
 {% endblock %}