Update app.py

app.py

@@ -3,7 +3,8 @@ import smtplib
 import ssl
 from email.message import EmailMessage
 
-os.environ["CUDA_VISIBLE_DEVICES"] = "-1"  # Force TensorFlow to use CPU
+# Force TensorFlow to use CPU to prevent CUDA errors
+os.environ["CUDA_VISIBLE_DEVICES"] = "-1"
 
 import gradio as gr
 import tensorflow as tf
@@ -12,16 +13,14 @@ from tensorflow.keras.preprocessing import image
 from PIL import Image
 from reportlab.lib.pagesizes import letter
 from reportlab.pdfgen import canvas
-from reportlab.lib import colors
-from reportlab.platypus import Table, TableStyle
 
 # Load the trained model
 model = tf.keras.models.load_model("my_keras_model.h5")
 
-# Store generated report file path
+# Store generated report file paths
 report_paths = {}
 
-# Function to send email
+# Function to send email with the report
 def send_email(patient_email, patient_name):
     if patient_name not in report_paths:
         return "Error: Generate the report first before sending."
@@ -77,15 +76,7 @@ def generate_report(name, age, gender, weight, height, allergies, cause, xray):
 
     # Predict fracture
     prediction = predict_fracture(xray)
-    diagnosed_class = "normal" if prediction > 0.5 else "Fractured"
-
-    # Injury severity classification
-    severity = "Mild" if prediction < 0.3 else "Moderate" if prediction < 0.7 else "Severe"
-
-    # Save X-ray image for report
-    img = Image.open(xray).resize((300, 300))
-    img_path = f"{name}_xray.png"
-    img.save(img_path)
+    diagnosed_class = "Normal" if prediction > 0.5 else "Fractured"
 
     # Generate PDF report
     report_path = f"{name}_fracture_report.pdf"
@@ -96,11 +87,16 @@ def generate_report(name, age, gender, weight, height, allergies, cause, xray):
 
     c.drawString(120, 290, f"Fractured: {'Yes' if diagnosed_class == 'Fractured' else 'No'}")
 
+    # Save X-ray image for report
+    img = Image.open(xray).resize((300, 300))
+    img_path = f"{name}_xray.png"
+    img.save(img_path)
+
     c.drawInlineImage(img_path, 50, 320, width=250, height=250)
 
     c.save()
 
-    # Store the file path
+    # Store file path for later use
     report_paths[name] = report_path
 
     return report_path  # Return file path
@@ -118,24 +114,29 @@ with gr.Blocks() as app:
         weight = gr.Number(label="Weight (kg)")
         height = gr.Number(label="Height (cm)")
 
+    with gr.Row():
+        allergies = gr.Textbox(label="Allergies")
+        cause = gr.Textbox(label="Cause of Injury")
+
     with gr.Row():
         email = gr.Textbox(label="Patient Email", type="email")
 
+    xray = gr.Image(type="filepath", label="Upload X-ray Image")
+
     with gr.Row():
-        xray = gr.Image(type="filepath", label="Upload X-ray Image")
+        submit_button = gr.Button("Generate Report")
+        send_email_button = gr.Button("Send Report via Email")
 
-    submit_button = gr.Button("Generate Report")
-    send_email_button = gr.Button("Send Report via Email")
     output_file = gr.File(label="Download Report")
 
-    # When clicking "Generate Report", save the file path and allow downloading
+    # Generate Report Button
     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
     send_email_button.click(
         send_email,
         inputs=[email, name],