Condition add for images

ui.py

@@ -1,11 +1,9 @@
 import gradio as gr
 from api import check_id, check_credit, check_mrz
 
-ALL_EXAMPLES = (
-    [[f"images/id/{i}.jpg"] for i in range(1, 8)]
-    + [["images/bank/demo1.jpg"], ["images/bank/demo2.png"], ["images/bank/demo3.png"]]
-    + [["images/mrz_barcode/demo1.png"], ["images/mrz_barcode/demo2.png"]]
-)
+ID_EXAMPLES = [[f"images/id/{i}.jpg"] for i in range(1, 8)]
+CREDIT_EXAMPLES = [["images/bank/demo1.jpg"], ["images/bank/demo2.png"], ["images/bank/demo3.png"]]
+MRZ_EXAMPLES = [["images/mrz_barcode/demo1.png"], ["images/mrz_barcode/demo2.png"]]
 
 
 def _format_result_html(result: dict) -> str:
@@ -90,16 +88,42 @@ def create_interface():
                 image_input = gr.Image(label="Upload Image")
                 submit_btn = gr.Button("Read Document", variant="primary", size="lg")
 
-                all_examples = gr.Examples(
-                    examples=ALL_EXAMPLES,
+                id_examples = gr.Examples(
+                    examples=ID_EXAMPLES,
                     inputs=image_input,
-                    label="Example Images",
+                    label="ID Card Examples",
+                    visible=True,
+                )
+                credit_examples = gr.Examples(
+                    examples=CREDIT_EXAMPLES,
+                    inputs=image_input,
+                    label="Credit Card Examples",
+                    visible=False,
+                )
+                mrz_examples = gr.Examples(
+                    examples=MRZ_EXAMPLES,
+                    inputs=image_input,
+                    label="MRZ / Barcode Examples",
+                    visible=False,
                 )
 
             with gr.Column(scale=1, min_width=400):
                 result_html = gr.HTML(label="Extracted Data")
                 raw_json = gr.JSON(label="Raw Response")
 
+        def on_type_change(choice):
+            return (
+                gr.update(visible=choice == "ID"),
+                gr.update(visible=choice == "Credit Card"),
+                gr.update(visible=choice == "MRZ"),
+            )
+
+        doc_type.change(
+            fn=on_type_change,
+            inputs=doc_type,
+            outputs=[id_examples, credit_examples, mrz_examples],
+        )
+
         submit_btn.click(
             fn=process_image,
             inputs=[image_input, doc_type],