Update app.py

app.py

@@ -67,7 +67,10 @@ def render_annotations(annotations, image_dir):
 # Gradio interface function
 def gradio_interface():
     example_dir = "examples"
-    annotations_file = random.choice([f for f in os.listdir(example_dir) if f.endswith('.json')])
+    random_index = random.choice([i for i in range(len(os.listdir(example_dir)))])
+    selected_example_dir_path = os.path.join(example_dir, str(random_index))
+    # annotations_file = random.choice([f for f in os.listdir(example_dir) if f.endswith('.json')])
+    annotations_file = os.path.join(selected_example_dir_path, "reading_annotation.json")
     annotations = load_annotations(os.path.join(example_dir, annotations_file))
     annotated_images = render_annotations(annotations, example_dir)
     return [gr.Image(value=image[0], label=image[1]) for image in annotated_images]