File size: 5,469 Bytes
dba26ff |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 |
import os
import pandas as pd
from reportlab.pdfgen import canvas
from reportlab.lib.pagesizes import letter
import tempfile
import matplotlib.pyplot as plt
from reportlab.lib import colors
from Damage_calculation import DamageCalculator
from reportlab.platypus import Table, TableStyle
class PDFReportGenerator:
def __init__(self, output_csv, output_folder, pdf_path):
self.output_csv = output_csv
self.damage_folder = os.path.join(output_folder, 'damage/labels')
self.parts_folder = os.path.join(output_folder, 'parts/labels')
self.pdf_path = pdf_path
self.calculator = DamageCalculator()
def create_pie_chart(self, part_class, damage_summary):
part_damage = damage_summary.get(part_class, {})
if not part_damage:
return None
damage_labels = list(part_damage['damage_types'].keys()) + ['Undamaged']
damage_values = list(part_damage['damage_types'].values()) + [part_damage['undamaged']]
fig, ax = plt.subplots(figsize=(6, 6)) # Adjust the figure size to ensure the pie is circular
ax.pie(damage_values, labels=damage_labels, autopct='%1.1f%%', startangle=140, textprops={'fontsize': 14}) # Increase font size
ax.axis('equal')
img_temp_path = tempfile.mktemp(suffix=".png")
plt.savefig(img_temp_path, bbox_inches='tight') # Ensure the pie chart is properly sized
plt.close(fig)
return img_temp_path
def add_table_to_pdf(self, pdf, dataframe, x_position, y_position):
# Replace NaN values with '-'
dataframe = dataframe.fillna('-')
data = [dataframe.columns.to_list()] + dataframe.values.tolist()
table = Table(data)
# Define table style
style = TableStyle([
('BACKGROUND', (0, 0), (-1, 0), colors.grey),
('TEXTCOLOR', (0, 0), (-1, 0), colors.whitesmoke),
('ALIGN', (0, 0), (-1, -1), 'CENTER'),
('FONTNAME', (0, 0), (-1, 0), 'Helvetica-Bold'),
('FONTSIZE', (0, 0), (-1, 0), 12),
('BOTTOMPADDING', (0, 0), (-1, 0), 12),
('BACKGROUND', (0, 1), (-1, -1), colors.beige),
('GRID', (0, 0), (-1, -1), 1, colors.black)
])
table.setStyle(style)
# Convert the table to a PDF flowable and draw it on the canvas
table.wrapOn(pdf, x_position, y_position)
table.drawOn(pdf, x_position, y_position)
def generate_report(self):
df = pd.read_csv(self.output_csv)
car_parts = df['Car Parts'].unique()
# Initialize damage summary dictionary
damage_summary = {}
# Populate damage and parts file paths
damage_files = sorted(os.listdir(self.damage_folder))
parts_files = sorted(os.listdir(self.parts_folder))
# Iterate over the damage and parts files
for damage_file, parts_file in zip(damage_files, parts_files):
damage_path = os.path.join(self.damage_folder, damage_file)
parts_path = os.path.join(self.parts_folder, parts_file)
damage_classes, damage_polygons = self.calculator.parse_coordinates(damage_path)
part_classes, part_polygons = self.calculator.parse_coordinates(parts_path)
damage_polygons = list(zip(damage_classes, damage_polygons))
part_polygons = list(zip(part_classes, part_polygons))
part_summary = self.calculator.summarize_damage_by_part(damage_polygons, part_polygons)
for part, summary in part_summary.items():
if part not in damage_summary:
damage_summary[part] = summary
else:
for damage_type, percentage in summary['damage_types'].items():
if damage_type not in damage_summary[part]['damage_types']:
damage_summary[part]['damage_types'][damage_type] = 0
damage_summary[part]['damage_types'][damage_type] += percentage
damage_summary[part]['undamaged'] += summary['undamaged']
pdf = canvas.Canvas(self.pdf_path, pagesize=letter)
pdf.setTitle("Car Damage Estimation Report")
width, height = letter
y_position = height - 40
for part_class in car_parts:
if part_class == 'Total':
continue
pdf.setFont("Helvetica-Bold", 12)
pdf.drawString(30, y_position, f'Damage Report for {part_class}')
y_position -= 20
pie_chart_path = self.create_pie_chart(part_class, damage_summary)
if pie_chart_path:
pdf.drawImage(pie_chart_path, 50, y_position - 300, width=275, height=275) # Adjust size to maintain aspect ratio
y_position -= 320
# Clean up the temporary file
os.remove(pie_chart_path)
if y_position < 100:
pdf.showPage()
y_position = height - 40
# Add title above the table
y_position -= 40
pdf.setFont("Helvetica-Bold", 14)
pdf.drawString(50, y_position, "Cost Breakdown Table")
y_position -= 200
# Add table from CSV to the PDF
self.add_table_to_pdf(pdf, df, 30, y_position-100)
pdf.save()
|