Update app.py

app.py

@@ -48,11 +48,20 @@ def generate_quiz(context):
 @app.route("/", methods=["GET", "POST"])
 def generate_quiz_page():
     if request.method == "POST":
-        context = request.form.get("context")
+        if request.content_type == 'application/json':
+            data = request.get_json()
+            context = data.get("context")
+            if context is None:
+                return jsonify({"error": "Missing 'context' parameter"}), 400
+        else:
+            context = request.form.get("context")
+            if context is None:
+                return jsonify({"error": "Missing 'context' parameter"}), 400
+        
         response = generate_quiz(context)
 
-        # if request.headers.get('Content-Type') == 'application/json':
-        #     return jsonify(response)
+        if request.content_type == 'application/json':
+            return jsonify(response)
 
 
         quiz_html = f"""