Update app.py

app.py

@@ -6,25 +6,24 @@ import pickle
 import io
 import os
 import matplotlib.pyplot as plt
+from reportlab.lib.pagesizes import A4
+from reportlab.lib import colors
 from reportlab.pdfgen import canvas
+from reportlab.lib.units import inch
 from datetime import datetime
 import logging
 
 app = Flask(__name__)
-app.secret_key = "4af58d07324a1f0226471fe2d526751a"  # Required for session handling
-
-# Logging setup
-logging.basicConfig(level=logging.INFO)
-logger = logging.getLogger(__name__)
+app.secret_key = "e3f6f40bb8b2471b9f07c4025d845be9"  # Replace with secure key if needed
 
 # Paths
 MODEL_PATH = "skin_lesion_model.h5"
 HISTORY_PATH = "training_history.pkl"
 PLOT_PATH = "/tmp/static/training_plot.png"
+LOGO_PATH = "static/logo.jpg"  # Logo in static folder
 IMG_SIZE = (224, 224)
 CONFIDENCE_THRESHOLD = 0.30
 
-# Label map
 label_map = {
     0: "Melanoma",
     1: "Melanocytic nevus",
@@ -36,6 +35,10 @@ label_map = {
     7: "Squamous cell carcinoma"
 }
 
+# Logging setup
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger(__name__)
+
 # Load model
 try:
     logger.info("Loading model from %s", MODEL_PATH)
@@ -44,7 +47,7 @@ except Exception as e:
     logger.error("Failed to load model: %s", str(e))
     raise
 
-# Load and plot training history
+# Load training history and generate plot
 history_dict = {}
 if os.path.exists(HISTORY_PATH):
     try:
@@ -63,9 +66,8 @@ if os.path.exists(HISTORY_PATH):
             plt.close()
             logger.info("Training plot saved at %s", PLOT_PATH)
     except Exception as e:
-        logger.error("Failed to load training history or plot: %s", str(e))
+        logger.error("Failed to process training history: %s", str(e))
 
-# Preprocess uploaded image
 def preprocess_image(image_bytes):
     try:
         image = Image.open(io.BytesIO(image_bytes)).convert("RGB")
@@ -77,17 +79,72 @@ def preprocess_image(image_bytes):
         logger.error("Image preprocessing failed: %s", str(e))
         raise
 
-# Generate PDF report
 def generate_pdf(report_data, filepath):
-    c = canvas.Canvas(filepath)
-    c.setFont("Helvetica", 14)
-    c.drawString(50, 800, "Skin Lesion Diagnosis Report")
-    c.setFont("Helvetica", 12)
-    y = 770
-    for key, value in report_data.items():
-        c.drawString(50, y, f"{key.capitalize()}: {value}")
-        y -= 20
-    c.drawString(50, y - 20, f"Generated on: {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}")
+    c = canvas.Canvas(filepath, pagesize=A4)
+    width, height = A4
+
+    # Add logo if exists
+    try:
+        if os.path.exists(LOGO_PATH):
+            c.drawImage(LOGO_PATH, 50, height - 100, width=80, preserveAspectRatio=True, mask='auto')
+    except Exception as e:
+        logger.warning("Could not load logo: %s", str(e))
+
+    # Title
+    c.setFillColor(colors.HexColor("#007ACC"))
+    c.setFont("Helvetica-Bold", 20)
+    c.drawCentredString(width / 2, height - 80, "Skin Lesion Diagnosis Report")
+    c.setStrokeColor(colors.HexColor("#007ACC"))
+    c.setLineWidth(2)
+    c.line(60, height - 90, width - 60, height - 90)
+
+    # Info box background
+    c.setFillColor(colors.lightgrey)
+    c.rect(50, height - 250, width - 100, 140, fill=1, stroke=0)
+
+    # Patient Info
+    c.setFillColor(colors.black)
+    c.setFont("Helvetica-Bold", 12)
+    y = height - 120
+    spacing = 20
+
+    def draw_field(label, value):
+        nonlocal y
+        c.setFont("Helvetica-Bold", 12)
+        c.drawString(70, y, f"{label}:")
+        c.setFont("Helvetica", 12)
+        c.drawString(180, y, value)
+        y -= spacing
+
+    draw_field("Full Name", report_data.get("name", "N/A"))
+    draw_field("Email", report_data.get("email", "N/A"))
+    draw_field("Gender", report_data.get("gender", "N/A"))
+    draw_field("Age", str(report_data.get("age", "N/A")))
+
+    # Prediction
+    y -= 20
+    c.setFont("Helvetica-Bold", 14)
+    c.setFillColor(colors.HexColor("#007ACC"))
+    c.drawString(50, y, "AI Diagnosis Result")
+    c.setFillColor(colors.black)
+    y -= spacing
+    draw_field("Prediction", report_data.get("prediction", "N/A"))
+    draw_field("Confidence", report_data.get("confidence", "N/A"))
+
+    # Optional message
+    message = report_data.get("message", "")
+    if message:
+        y -= 10
+        c.setFont("Helvetica-Oblique", 11)
+        c.setFillColor(colors.red)
+        c.drawString(70, y, message)
+
+    # Timestamp
+    y -= 40
+    c.setFont("Helvetica", 10)
+    c.setFillColor(colors.grey)
+    c.drawString(50, y, f"Generated on: {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}")
+
     c.save()
 
 @app.route("/form", methods=["GET"])