update pdf generation strings

services/pdf_service.py

@@ -9,6 +9,12 @@ import logging
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
+STRINGS = {
+    "meter_value": "Marca Métrica",
+    "last_valid_keypoint":"Último keypoint válido",
+    "cm_distance":"Distância água<->marca",
+}
+
 class PdfService:
     """
     A service to generate PDF reports.
@@ -36,8 +42,8 @@ class PdfService:
 
         # Metadata
         p.setFont("Helvetica", 12)
-        p.drawString(72, height - 108, f"Ship ID: {metadata.ship_id}")
-        p.drawString(72, height - 126, f"Timestamp: {metadata.timestamp.strftime('%Y-%m-%d %H:%M:%S')}")
+        p.drawString(72, height - 108, f"ID: {metadata.ship_id}")
+        p.drawString(72, height - 126, f"Data: {metadata.timestamp.strftime('%Y-%m-%d %H:%M:%S')}")
         p.drawString(72, height - 144, f"Latitude: {metadata.latitude}")
         p.drawString(72, height - 162, f"Longitude: {metadata.longitude}")
 
@@ -51,22 +57,28 @@ class PdfService:
         # ML Results
         p.setFont("Helvetica-Bold", 14)
         y_position = height - 270
-        p.drawString(72, y_position, "ML Results")
+        p.drawString(72, y_position, "Resultados de ML")
         y_position -= 18
         p.setFont("Helvetica", 12)
 
         # Highlight draft from ML results
         if 'draft' in ml_results:
             p.setFont("Helvetica-Bold", 12) # Highlight
-            p.drawString(90, y_position, f"ML Draft: {ml_results['draft']:.2f} meters")
+            p.drawString(90, y_position, f"Draft Predito: {ml_results['draft']:.2f} meters")
             p.setFont("Helvetica", 12) # Reset font
             y_position -= 18
 
         # Add other ML results (excluding arrays and images)
         for key, value in ml_results.items():
-            if key in ['pose_results', 'segment_results', 'original_image', 'pose_image_result', 'segment_image_result', 'final_image_result']:
+            if key in ['pose_results', 'segment_results', 'original_image',
+            'pose_image_result', 'segment_image_result', 'final_image_result', 
+            'extracted_mark_image', 'draft']:
                 continue
-            p.drawString(90, y_position, f"{key.replace('_', ' ').title()}: {value}")
+            try:
+                new_key = STRINGS[key]
+            except:
+                new_key = key.replace('_', ' ').title()
+            p.drawString(90, y_position, f"{new_key}: {value:.3f}")
             y_position -= 18
 
         # Images
@@ -104,7 +116,7 @@ class PdfService:
         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, "Extracted Mark Image:") # Label above image
+                p.drawString(current_x, y_image_row + image_height + 5, "Segmento Extraído p/ OCR") # 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
 
@@ -112,7 +124,7 @@ class PdfService:
         if 'pose_image_result' in ml_results and ml_results['pose_image_result'] is not None:
             pose_img_bytes = get_image_bytes(ml_results['pose_image_result'])
             if pose_img_bytes:
-                p.drawString(current_x, y_image_row + image_height + 5, "Pose Image Result:")
+                p.drawString(current_x, y_image_row + image_height + 5, "Estimação de Pose")
                 p.drawImage(ImageReader(BytesIO(pose_img_bytes)), current_x, y_image_row, width=image_width, height=image_height, preserveAspectRatio=True)
                 current_x += image_width + 20
 
@@ -120,13 +132,21 @@ class PdfService:
         if 'segment_image_result' in ml_results and ml_results['segment_image_result'] is not None:
             segment_img_bytes = get_image_bytes(ml_results['segment_image_result'])
             if segment_img_bytes:
-                p.drawString(current_x, y_image_row + image_height + 5, "Segment Image Result:")
+                p.drawString(current_x, y_image_row + image_height + 5, "Segmentação de Imagem")
                 p.drawImage(ImageReader(BytesIO(segment_img_bytes)), current_x, y_image_row, width=image_width, height=image_height, preserveAspectRatio=True)
                 # No need to update current_x as it's the last image in the row
 
         # Update y_position for content after images
         y_position = y_image_row - 20 # Adjust y_position to be below the images
 
+        # p.setFont("Helvetica-Bold", 14)
+        # y_position -= 18
+        # p.drawString(72, y_position, "Observações")
+        # y_position -= 18
+        # p.setFont("Helvetica", 12)
+        # p.drawString(72, y_position, "")
+
+
         p.showPage()
         p.save()