updates

app.py

@@ -76,7 +76,7 @@ def get_next_image(session_data):
 # Function to save the annotation to Hugging Face dataset and fetch the next image
 def save_annotation(caption, session_data):
     if session_data["current_image"] is None:
-        return gr.update(visible=False), gr.update(value="All images have been annotated!")
+        return gr.update(visible=False), gr.update(value="All images have been annotated!"), gr.update(value="")
 
     with lock:
         image_id = session_data["current_image"]
@@ -100,14 +100,14 @@ def save_annotation(caption, session_data):
     # Fetch the next image
     next_image = get_next_image(session_data)
     if next_image:
-        next_caption = get_caption_for_image_id(image_id)  # Retrieve the caption for the new image
-        return gr.update(value=next_image), gr.update(value=next_caption or "")
+        next_caption = get_caption_for_image_id(os.path.basename(next_image))  # Retrieve the caption for the new image
+        return gr.update(value=next_image), gr.update(value=""), gr.update(value=next_caption or "")
     else:
-        return gr.update(visible=False), gr.update(value="All images have been annotated!")
+        return gr.update(visible=False), gr.update(value="All images have been annotated!"), gr.update(value="")
 
 # Function to skip the current image
-def skip_image(session_data):
-    return save_annotation("skip", session_data)
+# def skip_image(session_data):
+#     return save_annotation("skip", session_data)
 
 def initialize_interface(session_data):
     next_image = get_next_image(session_data)
@@ -121,7 +121,7 @@ def initialize_interface(session_data):
 # Build the Gradio interface
 with gr.Blocks() as demo:
     gr.Markdown("# Image Captioning Tool")
-    gr.Markdown("Please provide a caption for each image displayed. Click 'Submit' after writing your caption, or type 'skip' if you don’t want to annotate this image.")
+    gr.Markdown("Please provide your caption in Egyptian Arabic 'Masri'")
 
     session_data = gr.State({"current_image": None})  # Session-specific state
 
@@ -130,11 +130,11 @@ with gr.Blocks() as demo:
         caption = gr.Textbox(placeholder="Enter caption here...")
         existing_caption = gr.Textbox(label="Existing Caption", interactive=False)  # Display existing caption
         submit = gr.Button("Submit")
-        skip = gr.Button("Skip")  # Skip button
+        # skip = gr.Button("Skip")  # Skip button
 
     # Define actions for buttons
-    submit.click(fn=save_annotation, inputs=[caption, session_data], outputs=[image, caption])
-    skip.click(fn=skip_image, inputs=session_data, outputs=[image, caption])
+    submit.click(fn=save_annotation, inputs=[caption, session_data], outputs=[image, caption, existing_caption])
+    # skip.click(fn=skip_image, inputs=session_data, outputs=[image, caption])
 
     # Load initial image
     demo.load(fn=initialize_interface, inputs=session_data, outputs=[image, existing_caption])