Update app.py

app.py

@@ -58,6 +58,8 @@ def send_email(patient_email, report_path, patient_name):
         server.login(sender_email, sender_password)
         server.send_message(msg)
 
+    return f"Report sent successfully to {patient_email}!"
+
 # 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)
@@ -158,19 +160,15 @@ def generate_report(name, age, gender, weight, height, allergies, cause, xray):
 with gr.Blocks() as app:
     gr.HTML(html_content)  # Display `re.html` content in Gradio
     gr.Markdown("## Bone Fracture Detection System")
-    
+
     with gr.Row():
         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():
         email = gr.Textbox(label="Patient Email", type="email")
@@ -182,17 +180,8 @@ 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=[],
-    )
+    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")])
 
 # Launch the Gradio app
 if __name__ == "__main__":