change default num

app.py

@@ -2,6 +2,7 @@ from flask_bootstrap import Bootstrap5
 from flask import Flask, render_template, request
 import utils
 import random
+import os
 
 app = Flask(__name__)
 bootstrap = Bootstrap5(app)
@@ -35,18 +36,21 @@ def get_questions(categories, num):
     utils.save_weights()
     return questions
 
+def get_num():
+    default_num = 5
+    return int(os.getenv("NUM_QUESTIONS", default_num))
 
 @app.route("/", methods=["POST", "GET"])
 def index():
     categories = utils.get_categories()
     selected = categories
-    n = 2
+    n = get_num()
     questions = {}
     if request.method == "POST":
         selected = request.form.getlist("categories[]")
         n = int(request.form.get("num"))
         questions = get_questions(selected, n)
-    print(selected, n)
+        
     return render_template("index.html", categories=categories, selected=selected, n=n, questions=questions)