Fixed UI issue

ui.py

@@ -88,40 +88,37 @@ def create_interface():
                 image_input = gr.Image(label="Upload Image")
                 submit_btn = gr.Button("Read Document", variant="primary", size="lg")
 
-                id_examples = gr.Examples(
-                    examples=ID_EXAMPLES,
-                    inputs=image_input,
-                    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(visible=True) as id_col:
+                    gr.Examples(
+                        examples=ID_EXAMPLES,
+                        inputs=image_input,
+                        label="ID Card Examples",
+                    )
+                with gr.Column(visible=False) as credit_col:
+                    gr.Examples(
+                        examples=CREDIT_EXAMPLES,
+                        inputs=image_input,
+                        label="Credit Card Examples",
+                    )
+                with gr.Column(visible=False) as mrz_col:
+                    gr.Examples(
+                        examples=MRZ_EXAMPLES,
+                        inputs=image_input,
+                        label="MRZ / Barcode Examples",
+                    )
 
             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 (
+        doc_type.change(
+            fn=lambda choice: (
                 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],
+            outputs=[id_col, credit_col, mrz_col],
         )
 
         submit_btn.click(