Update app.py

app.py

@@ -18,6 +18,9 @@ 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()
 
+# List of sample images
+sample_images = [f"samples/{img}" for img in os.listdir("samples") if img.endswith((".png", ".jpg", ".jpeg"))]
+
 # 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)
@@ -35,7 +38,7 @@ 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"],
@@ -114,6 +117,10 @@ def generate_report(name, age, gender, weight, height, allergies, cause, xray):
 
     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
@@ -134,10 +141,16 @@ 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")
     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],