example code update

app.py

@@ -17,6 +17,14 @@ from draco.model import (
     load_pretrained
 )
 
+example_files = {
+    "EMPIAR-10078": "example/empiar-10078-00-000093-full_patch_aligned.h5",
+    "EMPIAR-10154": "example/empiar-10154-00-000130-full_patch_aligned.h5",
+    "EMPIAR-10185": "example/empiar-10185-00-000032-full_patch_aligned.h5",
+    "EMPIAR-10200": "example/empiar-10200-00-000139-full_patch_aligned.h5",
+    "EMPIAR-10216": "example/empiar-10216-00-000036-full_patch_aligned.h5"
+}
+
 class DRACODenoiser(object):
     def __init__(self,
         cfg: DictConfig,
@@ -135,7 +143,7 @@ def process_and_denoise(data, x_offset, y_offset, auto_contrast) -> Image:
     return denoised_image
 
 def clear_images() -> tuple:
-    return None, None, None, gr.update(maximum=512), gr.update(maximum=512)  
+    return None, None, None, gr.update(maximum=1024), gr.update(maximum=1024)  
 
 with gr.Blocks(css="""
     .gradio-container {
@@ -166,12 +174,13 @@ with gr.Blocks(css="""
             </div>
             """
         )
-        
+
+        example_selector = gr.Radio(label="Choose an example Raw Micrograph File", choices=list(example_files.keys()))        
         file_input = gr.File(label="Or upload a Micrograph File in .h5 or .mrc format")
         auto_contrast = gr.Checkbox(label="Enable Auto Contrast", value=False)
 
-        x_slider = gr.Slider(0, 512, step=10, label="X Offset")
-        y_slider = gr.Slider(0, 512, step=10, label="Y Offset")
+        x_slider = gr.Slider(0, 1024, step=100, label="X Offset")
+        y_slider = gr.Slider(0, 1024, step=100, label="Y Offset")
 
         with gr.Row():
             denoise_button = gr.Button("Denoise")
@@ -189,7 +198,13 @@ with gr.Blocks(css="""
         data = load_data(file_path)
         h, w = data.shape[:2]
         return data, gr.update(maximum=w-1024), gr.update(maximum=h-1024)
-     
+
+    
+    example_selector.change(
+        lambda choice:load_image_and_update_sliders(example_files[choice]),  
+        inputs=example_selector, 
+        outputs=[active_data, x_slider, y_slider]
+    )     
     
     file_input.clear(
         clear_images,
@@ -198,7 +213,7 @@ with gr.Blocks(css="""
     )    
 
     file_input.change(
-        lambda file: load_image_and_update_sliders(file.name) if file else (None, None, None, gr.update(maximum=512), gr.update(maximum=512)),
+        lambda file: load_image_and_update_sliders(file.name) if file else (None, None, None, gr.update(maximum=1024), gr.update(maximum=1024)),
         inputs=file_input,
         outputs=[active_data, x_slider, y_slider]
     )