sqlite error

draft_computation/draft_computation_app/app.py

@@ -159,6 +159,7 @@ def calculate_draft(pose_results, segment_data, original_image):
         'last_valid_keypoint_cm': last_valid_keypoint_cm,
         'cm_distance': cm_distance,
         'final_draft_cm': final_draft_cm,
+        'extracted_mark_image': meter_mark_image
     }
 
     return final_draft, mid_results

draft_computation/endpoint.py

@@ -36,6 +36,7 @@ def run(img_path_or_array):
         "original_image": results1[0].orig_img,
         "pose_image_result": results1[0].plot(),
         "segment_image_result": results2[0].plot(),
+        "extracted_mark_image": mid_results['extracted_mark_image']
     }
     output['final_image_result'] = combine_plots(output['original_image'],
                                                  output['pose_image_result'],

services/pdf_service.py

@@ -94,10 +94,17 @@ class PdfService:
         current_x = x_start
 
         # Draw original image
-        if 'original_image' in ml_results and ml_results['original_image'] is not None:
-            original_img_bytes = get_image_bytes(ml_results['original_image'])
+        # if 'original_image' in ml_results and ml_results['original_image'] is not None:
+        #     original_img_bytes = get_image_bytes(ml_results['original_image'])
+        #     if original_img_bytes:
+        #         p.drawString(current_x, y_image_row + image_height + 5, "Original Image:") # Label above image
+        #         p.drawImage(ImageReader(BytesIO(original_img_bytes)), current_x, y_image_row, width=image_width, height=image_height, preserveAspectRatio=True)
+        #         current_x += image_width + 20 # Move x for next image
+
+        if 'extracted_mark_image' in ml_results and ml_results['extracted_mark_image'] is not None:
+            original_img_bytes = get_image_bytes(ml_results['extracted_mark_image'])
             if original_img_bytes:
-                p.drawString(current_x, y_image_row + image_height + 5, "Original Image:") # Label above image
+                p.drawString(current_x, y_image_row + image_height + 5, "Extracted Mark Image:") # Label above image
                 p.drawImage(ImageReader(BytesIO(original_img_bytes)), current_x, y_image_row, width=image_width, height=image_height, preserveAspectRatio=True)
                 current_x += image_width + 20 # Move x for next image