updates

app.py

@@ -98,11 +98,11 @@ def save_annotation(caption, session_data):
 def skip_image(session_data):
     return save_annotation("skip", session_data)
 
-# Function to initialize the interface
 def initialize_interface(session_data):
     next_image = get_next_image(session_data)
     if next_image:
-        next_caption = get_caption_for_image_id(session_data["current_image"])  # Retrieve caption for initial image
+        next_caption = get_caption_for_image_id(os.path.basename(next_image))  # Retrieve caption for initial image
+        print(next_caption)
         return gr.update(value=next_image), gr.update(value=next_caption or "")
     else:
         return gr.update(visible=False), gr.update(value="All images have been annotated!")