Update app.py

app.py

@@ -518,12 +518,27 @@ with gr.Blocks(title="Microscopy Analysis Suite", theme=gr.themes.Soft()) as dem
                         label="🖼️ 上传图像",
                         type="filepath"
                     )
+                    
+                    # 示例图片Gallery (与Segmentation相同)
+                    count_example_gallery = gr.Gallery(
+                        label="📁 示例图片",
+                        columns=3,
+                        object_fit="cover",
+                        height=150
+                    )
+                    
+                    # 上传示例图片
+                    count_image_uploader = gr.Image(
+                        label="➕ 上传新示例到Gallery",
+                        type="filepath"
+                    )
+                    
                     count_btn = gr.Button("▶️ 运行计数", variant="primary", size="lg")
                     
                     gr.Markdown(
                         """
                         **使用说明:**
-                        1. 上传细胞图像
+                        1. 上传图像或从Gallery选择示例
                         2. 点击 "运行计数"
                         3. 查看密度图和计数结果
                         """
@@ -533,18 +548,60 @@ with gr.Blocks(title="Microscopy Analysis Suite", theme=gr.themes.Soft()) as dem
                     count_output = gr.Image(
                         label="📸 密度图",
                         type="filepath",
-                        height=500
+                        height=400
                     )
                     count_status = gr.Textbox(
                         label="📊 统计信息",
                         lines=2
                     )
             
+            # 绑定事件
             count_btn.click(
                 fn=count_cells_handler,
                 inputs=count_input,
                 outputs=[count_output, count_status]
             )
+            
+            # 初始化Gallery显示
+            demo.load(
+                fn=lambda: example_images.copy(),
+                outputs=count_example_gallery
+            )
+            
+            # 绑定事件: 上传示例图片到Counting Gallery
+            count_user_examples = gr.State(example_images.copy())
+            
+            def add_to_count_gallery(img_path, current_imgs):
+                if not img_path:
+                    return current_imgs
+                try:
+                    if img_path not in current_imgs:
+                        current_imgs.append(img_path)
+                    return current_imgs
+                except:
+                    return current_imgs
+            
+            count_image_uploader.change(
+                fn=add_to_count_gallery,
+                inputs=[count_image_uploader, count_user_examples],
+                outputs=count_user_examples
+            ).then(
+                fn=lambda imgs: imgs,
+                inputs=count_user_examples,
+                outputs=count_example_gallery
+            )
+            
+            # 绑定事件: 点击Gallery加载到count_input
+            def load_from_count_gallery(evt: gr.SelectData, all_imgs):
+                if evt.index is not None and evt.index < len(all_imgs):
+                    return all_imgs[evt.index]
+                return None
+            
+            count_example_gallery.select(
+                fn=load_from_count_gallery,
+                inputs=count_user_examples,
+                outputs=count_input
+            )
         
         # ===== Tab 3: Tracking =====
         with gr.Tab("🎬 跟踪 (Tracking)"):