Update app.py

app.py

@@ -35,14 +35,16 @@ def generate_report(name, age, gender, weight, height, allergies, cause, xray):
 
     # Injury severity classification
     severity = "Mild" if prediction < 0.3 else "Moderate" if prediction < 0.7 else "Severe"
-    treatment_details = {
-        "Mild": "Your injury is classified as **Mild**. It may heal with rest, pain relievers, and a follow-up X-ray. Avoid excessive movement of the affected area.",
-        "Moderate": "Your injury is classified as **Moderate**. You may require a plaster cast, splint, or minor surgery. Recovery takes **4-8 weeks**.",
-        "Severe": "Your injury is classified as **Severe**. Surgery with metal implants and extensive physiotherapy is required. Recovery takes **several months** with proper rehabilitation."
-    }
-    treatment = treatment_details[severity]
-
-    # Estimated cost & duration
+    
+    # Treatment details table
+    treatment_data = [
+        ["Severity Level", "Recommended Treatment", "Recovery Duration"],
+        ["Mild", "Rest, pain relievers, and follow-up X-ray", "4-6 weeks"],
+        ["Moderate", "Plaster cast, minor surgery if needed", "6-10 weeks"],
+        ["Severe", "Major surgery, metal implants, physiotherapy", "Several months"]
+    ]
+
+    # Estimated cost & duration table
     cost_duration_data = [
         ["Hospital Type", "Estimated Cost", "Recovery Time"],
         ["Government Hospital", f"₹{2000 if severity == 'Mild' else 8000 if severity == 'Moderate' else 20000} - ₹{5000 if severity == 'Mild' else 15000 if severity == 'Moderate' else 50000}", "4-12 weeks"],
@@ -57,12 +59,13 @@ def generate_report(name, age, gender, weight, height, allergies, cause, xray):
     # Generate PDF report
     report_path = f"{name}_fracture_report.pdf"
     c = canvas.Canvas(report_path, pagesize=letter)
-    c.setFont("Helvetica-Bold", 14)
+    
+    # Report title
+    c.setFont("Helvetica-Bold", 16)
     c.drawString(200, 770, "Bone Fracture Detection Report")
 
     # Patient details table
     patient_data = [
-        ["Attribute", "Details"],
         ["Patient Name", name],
         ["Age", age],
         ["Gender", gender],
@@ -74,42 +77,38 @@ def generate_report(name, age, gender, weight, height, allergies, cause, xray):
         ["Injury Severity", severity]
     ]
 
-    table = Table(patient_data)
-    table.setStyle(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'),
-        ('BOTTOMPADDING', (0, 0), (-1, 0), 12),
-        ('GRID', (0, 0), (-1, -1), 1, colors.black)
-    ]))
-    table.wrapOn(c, 400, 500)
-    table.drawOn(c, 50, 600)
+    # Format and align tables
+    def format_table(data):
+        table = Table(data, colWidths=[270, 270])  # Set 90% width
+        table.setStyle(TableStyle([
+            ('BACKGROUND', (0, 0), (-1, 0), colors.darkblue),
+            ('TEXTCOLOR', (0, 0), (-1, 0), colors.whitesmoke),
+            ('ALIGN', (0, 0), (-1, -1), 'CENTER'),
+            ('FONTNAME', (0, 0), (-1, 0), 'Helvetica-Bold'),
+            ('BOTTOMPADDING', (0, 0), (-1, 0), 12),
+            ('GRID', (0, 0), (-1, -1), 1, colors.black),
+            ('VALIGN', (0, 0), (-1, -1), 'MIDDLE')
+        ]))
+        return table
+
+    # Draw patient details table
+    patient_table = format_table(patient_data)
+    patient_table.wrapOn(c, 480, 500)
+    patient_table.drawOn(c, 50, 620)
 
     # Load and insert X-ray image
     c.drawInlineImage(img_path, 50, 320, width=250, height=250)
     c.setFont("Helvetica-Bold", 12)
     c.drawString(120, 290, f"Fractured: {'Yes' if diagnosed_class == 'Fractured' else 'No'}")
 
-    # Cost estimation table
-    cost_table = Table(cost_duration_data)
-    cost_table.setStyle(TableStyle([
-        ('BACKGROUND', (0, 0), (-1, 0), colors.darkblue),
-        ('TEXTCOLOR', (0, 0), (-1, 0), colors.whitesmoke),
-        ('ALIGN', (0, 0), (-1, -1), 'CENTER'),
-        ('FONTNAME', (0, 0), (-1, 0), 'Helvetica-Bold'),
-        ('BOTTOMPADDING', (0, 0), (-1, 0), 12),
-        ('GRID', (0, 0), (-1, -1), 1, colors.black)
-    ]))
-    cost_table.wrapOn(c, 400, 200)
-    cost_table.drawOn(c, 50, 120)
-
-    # Add Treatment Recommendations
-    c.setFont("Helvetica-Bold", 12)
-    c.drawString(50, 100, "Treatment & Recovery Recommendations:")
-    c.setFont("Helvetica", 10)
-    c.drawString(50, 80, treatment)
-    c.drawString(50, 60, "Follow proper medical care and consult your doctor regularly.")
+    # Draw treatment and cost tables
+    treatment_table = format_table(treatment_data)
+    treatment_table.wrapOn(c, 480, 200)
+    treatment_table.drawOn(c, 50, 200)
+
+    cost_table = format_table(cost_duration_data)
+    cost_table.wrapOn(c, 480, 150)
+    cost_table.drawOn(c, 50, 80)
 
     c.save()