Update app.py

app.py

@@ -1,6 +1,8 @@
 import json
 import skia
 import gradio as gr
+import os
+import random
 
 # Define the category dictionary
 category_dict = {
@@ -63,21 +65,20 @@ def render_annotations(annotations, image_dir):
     return annotated_images
 
 # Gradio interface function
-def gradio_interface(annotations_file, image_dir):
-    annotations = load_annotations(annotations_file.name)
-    annotated_images = render_annotations(annotations, image_dir)
+def gradio_interface():
+    example_dir = "examples/1"
+    annotations_file = random.choice([f for f in os.listdir(example_dir) if f.endswith('.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]
 
 # Create Gradio interface
 iface = gr.Interface(
     fn=gradio_interface,
-    inputs=[
-        gr.File(label="Upload Annotations JSON File"),
-        gr.Textbox(label="Image Directory")
-    ],
+    inputs=[],
     outputs=gr.Gallery(label="Annotated Images"),
     title="Paper Annotation Renderer",
-    description="Upload an annotations JSON file and specify the image directory to render annotations onto the images."
+    description="Click to randomly choose an example and render annotations onto the images."
 )
 
 # Launch the Gradio interface