Pseudo-color-4

app.py

@@ -286,6 +286,14 @@ def update_gallery_color(raw_list_state, color_name):
     bar_img = generate_colorbar_preview(color_name)
     return new_gallery, bar_img
 
+# --- Event Handler Helper (Fix for Lambda Issue) ---
+def get_gallery_selection(evt: gr.SelectData):
+    """
+    Standard event handler for gallery selection.
+    Returns the caption (which stores the filepath/prompt) from the selected item.
+    """
+    return evt.value['caption']
+
 # --- Generation Functions ---
 @spaces.GPU(duration=120)
 def generate_t2i(prompt, num_inference_steps, current_color):
@@ -514,7 +522,8 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
                 inputs=[t2i_raw_state, t2i_color],
                 outputs=[t2i_out, t2i_colorbar]
             )
-            t2i_gal.select(lambda e: e.value['caption'], None, t2i_prompt)
+            # FIXED: Use defined function instead of lambda
+            t2i_gal.select(fn=get_gallery_selection, inputs=None, outputs=t2i_prompt)
 
         # --- TAB 2: Super-Resolution ---
         with gr.Tab("Super-Resolution", id="super_res"):
@@ -544,7 +553,8 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
             sr_model.change(update_sr_prompt, sr_model, sr_prompt)
             sr_btn.click(run_super_resolution, [sr_file, sr_model, sr_prompt, sr_steps, sr_seed, sr_color], [sr_in_disp, sr_out_disp, sr_dl, sr_raw_state, sr_colorbar])
             sr_color.change(update_single_image_color, [sr_raw_state, sr_color], [sr_out_disp, sr_colorbar])
-            sr_gal.select(lambda e: e.value['caption'], None, sr_file)
+            # FIXED: Use defined function instead of lambda
+            sr_gal.select(fn=get_gallery_selection, inputs=None, outputs=sr_file)
 
         # --- TAB 3: Denoising ---
         with gr.Tab("Denoising", id="denoising"):
@@ -572,7 +582,8 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
             
             dn_btn.click(run_denoising, [dn_img, dn_type, dn_steps, dn_seed, dn_color], [dn_orig, dn_out, dn_dl, dn_raw_state, dn_colorbar])
             dn_color.change(update_single_image_color, [dn_raw_state, dn_color], [dn_out, dn_colorbar])
-            dn_gal.select(lambda e: e.value['caption'], None, dn_img)
+            # FIXED: Use defined function instead of lambda
+            dn_gal.select(fn=get_gallery_selection, inputs=None, outputs=dn_img)
 
         # --- TAB 4: Mask-to-Image ---
         with gr.Tab("Mask-to-Image", id="mask2img"):
@@ -600,7 +611,8 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
             
             m2i_btn.click(run_mask_to_image_generation, [m2i_file, m2i_type, m2i_num, m2i_steps, m2i_seed, m2i_color], [m2i_in_disp, m2i_gal_out, m2i_dl, m2i_raw_state, m2i_colorbar])
             m2i_color.change(update_gallery_color, [m2i_raw_state, m2i_color], [m2i_gal_out, m2i_colorbar])
-            m2i_gal.select(lambda e: e.value['caption'], None, m2i_file)
+            # FIXED: Use defined function instead of lambda
+            m2i_gal.select(fn=get_gallery_selection, inputs=None, outputs=m2i_file)
 
         # --- TAB 5: Cell Segmentation ---
         with gr.Tab("Cell Segmentation", id="segmentation"):
@@ -618,7 +630,8 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
                         seg_out = gr.Image(label="Overlay", type="pil", interactive=False, show_download_button=False)
             seg_gal = gr.Gallery(value=seg_examples, label="Examples", columns=6, height="auto")
             seg_btn.click(run_segmentation, [seg_img, seg_model, seg_diam, seg_flow, seg_prob], [seg_orig, seg_out])
-            seg_gal.select(lambda e: e.value['caption'], None, seg_img)
+            # FIXED: Use defined function instead of lambda
+            seg_gal.select(fn=get_gallery_selection, inputs=None, outputs=seg_img)
 
         # --- TAB 6: Classification ---
         with gr.Tab("Classification", id="classification"):
@@ -632,7 +645,8 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
                     cls_res = gr.Label(label="Results", num_top_classes=10)
             cls_gal = gr.Gallery(value=cls_examples, label="Examples", columns=6, height="auto")
             cls_btn.click(run_classification, [cls_img, cls_model], [cls_orig, cls_res])
-            cls_gal.select(lambda e: e.value['caption'], None, cls_img)
+            # FIXED: Use defined function instead of lambda
+            cls_gal.select(fn=get_gallery_selection, inputs=None, outputs=cls_img)
 
 if __name__ == "__main__":
     demo.launch()