Donr

image_routes.py

@@ -2,6 +2,7 @@ from flask import Blueprint, send_from_directory, current_app, request, jsonify
 from flask_login import login_required, current_user
 from utils import get_db_connection
 import os
+import base64
 from datetime import datetime
 
 image_bp = Blueprint('image_bp', __name__)
@@ -139,12 +140,13 @@ def delete_note():
 @image_bp.route('/save_note_json', methods=['POST'])
 @login_required
 def save_note_json():
-    """Save revision notes as JSON (no rasterization)."""
+    """Save revision notes as JSON and rasterized PNG for PDF/quiz display."""
     try:
         data = request.json
         image_id = data.get('image_id')
         session_id = data.get('session_id')
         json_data = data.get('json_data')
+        image_data = data.get('image_data')
 
         if not image_id or not session_id or not json_data:
             return jsonify({'error': 'Missing required fields'}), 400
@@ -162,6 +164,20 @@ def save_note_json():
 
         # Save JSON to database
         conn.execute("UPDATE images SET note_json = ? WHERE id = ?", (json_data, image_id))
+
+        # Save rasterized PNG if provided (needed for PDF generation and quiz display)
+        if image_data and image_data.startswith('data:image/'):
+            # Strip the data URL prefix (e.g. "data:image/png;base64,")
+            header, encoded = image_data.split(',', 1)
+            img_bytes = base64.b64decode(encoded)
+
+            filename = f"note_{session_id}_{image_id}_{int(datetime.now().timestamp())}.png"
+            save_path = os.path.join(current_app.config['PROCESSED_FOLDER'], filename)
+            with open(save_path, 'wb') as f:
+                f.write(img_bytes)
+
+            conn.execute("UPDATE images SET note_filename = ? WHERE id = ?", (filename, image_id))
+
         conn.commit()
         conn.close()
 

templates/_revision_notes.html

@@ -503,9 +503,12 @@
     }
 
     async function saveNotes() {
-        // Save as JSON instead of rasterized image
+        // Save as JSON AND rasterized PNG (needed for PDF/quiz display)
         const jsonData = JSON.stringify(canvas.toJSON());
 
+        // Rasterize canvas to PNG data URL for PDF/quiz rendering
+        const imageData = canvas.toDataURL({ format: 'png', multiplier: 2 });
+
         try {
             const response = await fetch('/save_note_json', {
                 method: 'POST',
@@ -513,7 +516,8 @@
                 body: JSON.stringify({
                     image_id: activeImageId,
                     session_id: '{{ session_id }}',
-                    json_data: jsonData
+                    json_data: jsonData,
+                    image_data: imageData
                 })
             });