Update app.py

app.py

@@ -150,75 +150,165 @@ def preprocess_image(image_bytes):
 def generate_pdf(report, filepath):
     c = canvas.Canvas(filepath, pagesize=A4)
     width, height = A4
-    y = height - 80
+    y = height - 60
 
-    # Logo
+ 
+    c.setFillColor(colors.Color(0.97, 0.98, 1, alpha=1))  # Very light blue
+    c.rect(0, 0, width, height, fill=1, stroke=0)
+    c.setFillColor(colors.Color(0.85, 0.93, 0.98, alpha=1))
+    c.rect(0, height-120, width, 120, fill=1, stroke=0)
     try:
         if os.path.exists(LOGO_PATH):
-            c.drawImage(LOGO_PATH, 50, y, width=80, preserveAspectRatio=True, mask='auto')
+            # Add circular background for logo
+            c.setFillColor(colors.white)
+            c.circle(90, y, 35, fill=1, stroke=1)
+            c.setStrokeColor(colors.Color(0.2, 0.4, 0.8, alpha=1))
+            c.setLineWidth(2)
+            c.circle(90, y, 35, fill=0, stroke=1)
+            c.drawImage(LOGO_PATH, 60, y-25, width=60, preserveAspectRatio=True, mask='auto')
     except Exception as e:
         logger.warning("Logo error: %s", str(e))
+    c.setFont("Helvetica-Bold", 24)
+    c.setFillColor(colors.Color(0.1, 0.3, 0.6, alpha=1))  # Deep blue
+    c.drawCentredString(width / 2, y + 10, "Skin Lesion Diagnosis Report")
+    
+    c.setFont("Helvetica", 12)
+    c.setFillColor(colors.Color(0.4, 0.5, 0.7, alpha=1))
+    c.drawCentredString(width / 2, y - 10, "AI-Powered Medical Analysis Report")
+    
+    c.setStrokeColor(colors.Color(0.2, 0.7, 0.8, alpha=1))
+    c.setLineWidth(3)
+    c.line(80, y - 30, width - 80, y - 30)
+    c.setStrokeColor(colors.Color(0.4, 0.8, 0.9, alpha=0.5))
+    c.setLineWidth(1)
+    c.line(80, y - 32, width - 80, y - 32)
+    
+    y -= 80
 
-    # Title
-    c.setFont("Helvetica-Bold", 20)
-    c.setFillColor(colors.HexColor("#005e8a"))
-    c.drawCentredString(width / 2, y, "Skin Lesion Diagnosis Report")
-    c.setStrokeColor(colors.HexColor("#005e8a"))
-    c.setLineWidth(1.5)
-    c.line(60, y - 10, width - 60, y - 10)
-    y -= 40
-
-    def section_box(title, fields, extra_gap=20):
+    def modern_section_box(title, fields, color_scheme="blue", extra_gap=25):
         nonlocal y
-        c.setFillColor(colors.lightgrey)
-        box_height = len(fields) * 18 + 30
-        c.rect(50, y - box_height, width - 100, box_height, fill=1, stroke=0)
-        c.setFont("Helvetica-Bold", 13)
-        c.setFillColor(colors.HexColor("#003f63"))
-        c.drawString(60, y - 20, title)
-        y -= 40
+        
+        color_schemes = {
+            "blue": {
+                "bg": colors.Color(0.94, 0.97, 1, alpha=1),
+                "border": colors.Color(0.3, 0.6, 0.9, alpha=1),
+                "title": colors.Color(0.1, 0.3, 0.7, alpha=1),
+                "accent": colors.Color(0.2, 0.7, 0.8, alpha=1)
+            },
+            "green": {
+                "bg": colors.Color(0.94, 0.98, 0.94, alpha=1),
+                "border": colors.Color(0.3, 0.7, 0.4, alpha=1),
+                "title": colors.Color(0.1, 0.5, 0.2, alpha=1),
+                "accent": colors.Color(0.4, 0.8, 0.5, alpha=1)
+            },
+            "purple": {
+                "bg": colors.Color(0.97, 0.94, 0.98, alpha=1),
+                "border": colors.Color(0.6, 0.3, 0.8, alpha=1),
+                "title": colors.Color(0.4, 0.1, 0.6, alpha=1),
+                "accent": colors.Color(0.7, 0.4, 0.8, alpha=1)
+            },
+            "orange": {
+                "bg": colors.Color(0.99, 0.96, 0.92, alpha=1),
+                "border": colors.Color(0.9, 0.5, 0.2, alpha=1),
+                "title": colors.Color(0.7, 0.3, 0.1, alpha=1),
+                "accent": colors.Color(0.9, 0.6, 0.3, alpha=1)
+            }
+        }
+        
+        scheme = color_schemes.get(color_scheme, color_schemes["blue"])
+        box_height = len(fields) * 22 + 45
+        
+        c.setFillColor(scheme["bg"])
+        c.roundRect(40, y - box_height, width - 80, box_height, 8, fill=1, stroke=0)
+        
+        c.setStrokeColor(scheme["border"])
+        c.setLineWidth(2)
+        c.roundRect(40, y - box_height, width - 80, box_height, 8, fill=0, stroke=1)
+        
+        c.setFillColor(colors.Color(0.8, 0.8, 0.8, alpha=0.3))
+        c.roundRect(43, y - box_height - 3, width - 80, box_height, 8, fill=1, stroke=0)
+        
+        c.setFillColor(scheme["accent"])
+        c.roundRect(40, y - 35, width - 80, 35, 8, fill=1, stroke=0)
+        
+        c.setFont("Helvetica-Bold", 14)
+        c.setFillColor(colors.white)
+        icon_map = {
+            "Patient Details": "👤",
+            "Diagnosis": "🔬", 
+            "Recommended Treatment": "💊",
+            "Suggested Medications": "🏥"
+        }
+        icon = icon_map.get(title, "📋")
+        
+        c.drawString(55, y - 25, f"{icon} {title}")
+        
+        y -= 50
         c.setFont("Helvetica", 11)
-        c.setFillColor(colors.black)
-        for label, val in fields.items():
-            c.drawString(70, y, f"{label}: {val}")
-            y -= 18
+        c.setFillColor(colors.Color(0.2, 0.2, 0.2, alpha=1))
+        
+        for i, (label, val) in enumerate(fields.items()):
+            if i % 2 == 0:
+                c.setFillColor(colors.Color(1, 1, 1, alpha=0.5))
+                c.rect(50, y - 3, width - 100, 18, fill=1, stroke=0)
+            
+            c.setFillColor(colors.Color(0.3, 0.3, 0.3, alpha=1))
+            c.setFont("Helvetica-Bold", 10)
+            c.drawString(60, y, f"{label}:")
+            c.setFont("Helvetica", 10)
+            c.drawString(180, y, str(val))
+            y -= 22
+        
         y -= extra_gap
-
-    # Sections
-    section_box("Patient Details", {
-        "Name": report["name"],
-        "Email": report["email"],
+    modern_section_box("Patient Details", {
+        "Full Name": report["name"],
+        "Email Address": report["email"],
         "Gender": report["gender"],
-        "Age": report["age"]
-    })
+        "Age": f"{report['age']} years old"
+    }, "blue")
 
-    section_box("Diagnosis", {
-        "Prediction": report["prediction"],
-        "Confidence": report["confidence"],
-        "Note": report["message"] if report["message"] else "N/A"
-    })
+    confidence_val = float(report["confidence"].replace('%', ''))
+    confidence_color = "green" if confidence_val > 80 else "orange" if confidence_val > 60 else "red"
+    
+    modern_section_box("Diagnosis Results", {
+        "Medical Prediction": report["prediction"],
+        "Confidence Level": f"{report['confidence']} {'🟢' if confidence_val > 80 else '🟡' if confidence_val > 60 else '🔴'}",
+        "Clinical Notes": report["message"] if report["message"] else "No additional notes"
+    }, "green")
 
     disease = report["prediction"]
     treatment = recommendations.get(disease, recommendations["Unknown"])
 
-    section_box("Recommended Treatment", {
-        f"• {line}": "" for line in treatment["solutions"]
-    })
+    modern_section_box("Recommended Treatment Plan", {
+        f"Step {i+1}": line for i, line in enumerate(treatment["solutions"])
+    }, "purple")
 
-    section_box("Suggested Medications", {
-        f"• {line}": "" for line in treatment["medications"]
-    })
+    modern_section_box("Suggested Medications", {
+        f"Medication {i+1}": line for i, line in enumerate(treatment["medications"])
+    }, "orange")
 
-    # Disclaimer
-    disclaimer_text1 = "DISCLAIMER:"
-    disclaimer_text2 = "This report is AI-generated using a machine learning model and is intended only to provide an approximate overview."
-    disclaimer_text3 = "It is not a substitute for professional medical advice. Please consult a certified doctor for accurate diagnosis and treatment."
-    c.setFont("Helvetica-Bold", 9)
-    c.setFillColor(colors.red)
-    c.drawString(50, 60, disclaimer_text1)
-    c.setFont("Helvetica-Oblique", 9)
-    c.drawString(110, 60, disclaimer_text2)
-    c.drawString(50, 48, disclaimer_text3)
+    y = 120
+    c.setFillColor(colors.Color(1, 0.95, 0.9, alpha=1))  # Light orange background
+    c.roundRect(40, 30, width - 80, 80, 8, fill=1, stroke=0)
+    c.setStrokeColor(colors.Color(0.9, 0.4, 0.1, alpha=1))
+    c.setLineWidth(2)
+    c.roundRect(40, 30, width - 80, 80, 8, fill=0, stroke=1)
+    
+    c.setFont("Helvetica-Bold", 12)
+    c.setFillColor(colors.Color(0.8, 0.3, 0.1, alpha=1))
+    c.drawString(55, 95, "⚠️ IMPORTANT MEDICAL DISCLAIMER")
+    
+    c.setFont("Helvetica", 9)
+    c.setFillColor(colors.Color(0.5, 0.2, 0.1, alpha=1))
+    disclaimer_lines = [
+        "This report is generated using artificial intelligence and machine learning algorithms.",
+        "The results provided are for informational purposes only and should not be considered",
+        "as a substitute for professional medical advice, diagnosis, or treatment.",
+        "Always consult with a qualified healthcare provider for proper medical evaluation."
+    ]
+    
+    for i, line in enumerate(disclaimer_lines):
+        c.drawString(55, 80 - (i * 12), line)
 
     c.save()