Update app.py

app.py

@@ -59,7 +59,7 @@ def send_email(patient_email, report_path, patient_name):
         server.send_message(msg)
 
 # Function to process X-ray and generate a PDF report
-def generate_report(name, age, gender, weight, height, allergies, cause, xray, email):
+def generate_report(name, age, gender, weight, height, allergies, cause, xray):
     image_size = (224, 224)
 
     def predict_fracture(xray_path):
@@ -152,15 +152,8 @@ def generate_report(name, age, gender, weight, height, allergies, cause, xray, e
 
     c.save()
 
-    # Send the report via email
-    send_email(email, report_path, name)
-
     return report_path  # Return path for auto-download
 
-# Function to select a sample image
-def use_sample_image(sample_image_path):
-    return sample_image_path  # Returns selected sample image filepath
-
 # Define Gradio Interface
 with gr.Blocks() as app:
     gr.HTML(html_content)  # Display `re.html` content in Gradio
@@ -184,22 +177,23 @@ with gr.Blocks() as app:
 
     with gr.Row():
         xray = gr.Image(type="filepath", label="Upload X-ray Image")
-    
-    with gr.Row():
-        sample_selector = gr.Dropdown(choices=sample_images, label="Use Sample Image")
-        select_button = gr.Button("Load Sample Image")
 
     submit_button = gr.Button("Generate Report")
+    send_email_button = gr.Button("Send Report via Email")
     output_file = gr.File(label="Download Report")
 
-    select_button.click(use_sample_image, inputs=[sample_selector], outputs=[xray])
-
     submit_button.click(
         generate_report,
-        inputs=[name, age, gender, weight, height, allergies, cause, xray, email],
+        inputs=[name, age, gender, weight, height, allergies, cause, xray],
         outputs=[output_file],
     )
 
+    send_email_button.click(
+        send_email,
+        inputs=[email, output_file, name],
+        outputs=[],
+    )
+
 # Launch the Gradio app
 if __name__ == "__main__":
     app.launch()