Update app.py

app.py

@@ -18,15 +18,16 @@ from reportlab.platypus import Table, TableStyle
 # Load the trained model
 model = tf.keras.models.load_model("my_keras_model.h5")
 
-# Read HTML content from `re.html`
-with open("templates/re.html", "r", encoding="utf-8") as file:
-    html_content = file.read()
-
-# List of sample images
-sample_images = [f"samples/{img}" for img in os.listdir("samples") if img.endswith((".png", ".jpg", ".jpeg"))]
+# Store generated report file path
+report_paths = {}
 
 # Function to send email
-def send_email(patient_email, report_path, patient_name):
+def send_email(patient_email, patient_name):
+    if patient_name not in report_paths:
+        return "Error: Generate the report first before sending."
+
+    report_path = report_paths[patient_name]
+
     sender_email = "[email protected]"
     sender_password = "your_email_password"
 
@@ -60,8 +61,11 @@ def send_email(patient_email, report_path, patient_name):
 
     return f"Report sent successfully to {patient_email}!"
 
-# Function to process X-ray and generate a PDF report
+# Function to generate report
 def generate_report(name, age, gender, weight, height, allergies, cause, xray):
+    if not name:
+        return "Error: Please enter a patient name."
+
     image_size = (224, 224)
 
     def predict_fracture(xray_path):
@@ -78,21 +82,6 @@ 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 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"],
-        ["Private Hospital", f"₹{10000 if severity == 'Mild' else 30000 if severity == 'Moderate' else 100000}+", "6 weeks - Several months"]
-    ]
-
     # Save X-ray image for report
     img = Image.open(xray).resize((300, 300))
     img_path = f"{name}_xray.png"
@@ -102,63 +91,22 @@ def generate_report(name, age, gender, weight, height, allergies, cause, xray):
     report_path = f"{name}_fracture_report.pdf"
     c = canvas.Canvas(report_path, pagesize=letter)
     
-    # Report title
     c.setFont("Helvetica-Bold", 16)
     c.drawString(200, 770, "Bone Fracture Detection Report")
 
-    # Patient details table
-    patient_data = [
-        ["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]
-    ]
-
-    # 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'}")
 
-    # 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.drawInlineImage(img_path, 50, 320, width=250, height=250)
 
     c.save()
 
-    return report_path  # Return path for auto-download
+    # Store the file path
+    report_paths[name] = report_path
+
+    return report_path  # Return file path
 
 # Define Gradio Interface
 with gr.Blocks() as app:
-    gr.HTML(html_content)  # Display `re.html` content in Gradio
     gr.Markdown("## Bone Fracture Detection System")
 
     with gr.Row():
@@ -180,8 +128,19 @@ with gr.Blocks() as app:
     send_email_button = gr.Button("Send Report via Email")
     output_file = gr.File(label="Download Report")
 
-    submit_button.click(generate_report, inputs=[name, age, gender, weight, height, allergies, cause, xray], outputs=[output_file])
-    send_email_button.click(send_email, inputs=[email, output_file, name], outputs=[gr.Textbox(label="Status")])
+    # When clicking "Generate Report", save the file path and allow downloading
+    submit_button.click(
+        generate_report,
+        inputs=[name, age, gender, weight, height, allergies, cause, xray],
+        outputs=[output_file]
+    )
+
+    # When clicking "Send Report via Email", send the stored report file
+    send_email_button.click(
+        send_email,
+        inputs=[email, name],
+        outputs=[gr.Textbox(label="Status")]
+    )
 
 # Launch the Gradio app
 if __name__ == "__main__":