Update app.py

app.py

@@ -18,8 +18,8 @@ model = tf.keras.models.load_model("my_keras_model.h5")
 with open("templates/re.html", "r", encoding="utf-8") as file:
     html_content = file.read()
 
-# Function to process X-rays and generate a PDF report
-def generate_report(name, age, gender, allergies, cause, xray1, xray2):
+# Function to process X-ray and generate a PDF report
+def generate_report(name, age, gender, weight, height, allergies, cause, xray):
     image_size = (224, 224)
 
     def predict_fracture(xray_path):
@@ -29,18 +29,16 @@ def generate_report(name, age, gender, allergies, cause, xray1, xray2):
         prediction = model.predict(img_array)[0][0]
         return prediction
 
-    # Predict on both X-rays
-    prediction1 = predict_fracture(xray1)
-    prediction2 = predict_fracture(xray2)
-    avg_prediction = (prediction1 + prediction2) / 2
-    diagnosed_class = "Fractured" if avg_prediction > 0.5 else "Normal"
+    # Predict fracture
+    prediction = predict_fracture(xray)
+    diagnosed_class = "Fractured" if prediction > 0.5 else "Normal"
 
     # Injury severity classification
-    severity = "Mild" if avg_prediction < 0.3 else "Moderate" if avg_prediction < 0.7 else "Severe"
+    severity = "Mild" if prediction < 0.3 else "Moderate" if prediction < 0.7 else "Severe"
     treatment_details = {
-        "Mild": "Your fracture 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 fracture is classified as **Moderate**. You may require a plaster cast, splint, or minor surgery. Recovery takes **4-8 weeks**.",
-        "Severe": "Your fracture is classified as **Severe**. Surgery with metal implants and extensive physiotherapy is required. Recovery takes **several months** with proper rehabilitation."
+        "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]
 
@@ -51,13 +49,10 @@ def generate_report(name, age, gender, allergies, cause, xray1, xray2):
         ["Private Hospital", f"₹{10000 if severity == 'Mild' else 30000 if severity == 'Moderate' else 100000}+", "6 weeks - Several months"]
     ]
 
-    # Save X-ray images for report
-    img1 = Image.open(xray1).resize((300, 300))
-    img2 = Image.open(xray2).resize((300, 300))
-    img1_path = f"{name}_xray1.png"
-    img2_path = f"{name}_xray2.png"
-    img1.save(img1_path)
-    img2.save(img2_path)
+    # Save X-ray image for report
+    img = Image.open(xray).resize((300, 300))
+    img_path = f"{name}_xray.png"
+    img.save(img_path)
 
     # Generate PDF report
     report_path = f"{name}_fracture_report.pdf"
@@ -65,39 +60,56 @@ def generate_report(name, age, gender, allergies, cause, xray1, xray2):
     c.setFont("Helvetica-Bold", 14)
     c.drawString(200, 770, "Bone Fracture Detection Report")
 
-    # Patient details
-    c.setFont("Helvetica", 12)
-    c.drawString(100, 740, f"Patient Name: {name}")
-    c.drawString(100, 720, f"Age: {age}")
-    c.drawString(100, 700, f"Gender: {gender}")
-    c.drawString(100, 680, f"Allergies: {allergies if allergies else 'None'}")
-    c.drawString(100, 660, f"Cause of Injury: {cause if cause else 'Not Provided'}")
+    # Patient details table
+    patient_data = [
+        ["Attribute", "Details"],
+        ["Patient Name", name],
+        ["Age", age],
+        ["Gender", gender],
+        ["Weight", f"{weight} kg"],
+        ["Height", f"{height} cm"],
+        ["Allergies", allergies if allergies else "None"],
+        ["Cause of Injury", cause if cause else "Not Provided"],
+        ["Diagnosis", diagnosed_class],
+        ["Injury Severity", severity]
+    ]
 
-    # Diagnosis
-    c.setFont("Helvetica-Bold", 12)
-    c.drawString(100, 630, "Diagnosis & Treatment Plan:")
-    c.setFont("Helvetica", 11)
-    c.drawString(100, 610, f"Fracture Detected: {diagnosed_class}")
-    c.drawString(100, 590, f"Injury Severity: {severity}")
-    c.setFont("Helvetica", 10)
-    c.drawString(100, 570, f"{treatment}")
+    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)
 
-    # Load and insert X-ray images
-    c.drawInlineImage(img1_path, 50, 250, width=250, height=250)
-    c.drawInlineImage(img2_path, 320, 250, width=250, height=250)
+    # 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
-    table = Table(cost_duration_data)
-    table.setStyle(TableStyle([
-        ('BACKGROUND', (0, 0), (-1, 0), colors.grey),
+    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)
     ]))
-    table.wrapOn(c, 400, 300)
-    table.drawOn(c, 100, 150)
+    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.")
 
     c.save()
 
@@ -112,21 +124,24 @@ with gr.Blocks() as app:
         name = gr.Textbox(label="Patient Name")
         age = gr.Number(label="Age")
         gender = gr.Radio(["Male", "Female", "Other"], label="Gender")
-
+    
+    with gr.Row():
+        weight = gr.Number(label="Weight (kg)")
+        height = gr.Number(label="Height (cm)")
+    
     with gr.Row():
         allergies = gr.Textbox(label="Allergies (if any)")
         cause = gr.Textbox(label="Cause of Injury")
 
     with gr.Row():
-        xray1 = gr.Image(type="filepath", label="Upload X-ray Image 1")
-        xray2 = gr.Image(type="filepath", label="Upload X-ray Image 2")
+        xray = gr.Image(type="filepath", label="Upload X-ray Image")
 
     submit_button = gr.Button("Generate Report")
     output_file = gr.File(label="Download Report")
 
     submit_button.click(
         generate_report,
-        inputs=[name, age, gender, allergies, cause, xray1, xray2],
+        inputs=[name, age, gender, weight, height, allergies, cause, xray],
         outputs=[output_file],
     )