updates

app.py

@@ -7,7 +7,11 @@ from huggingface_hub import login
 import json
 import re
 
-reset = True
+# reset = False
+
+def check_word_count(caption):
+    # Check if the caption has 3 or more words
+    return gr.update(interactive=len(caption.split()) >= 3)
 
 # Authenticate with Hugging Face
 token = os.getenv("HUGGINGFACE_TOKEN")
@@ -67,28 +71,28 @@ except Exception as e:
 
 
 # Initialize or reset data as needed based on the `reset` flag
-if reset:
-    # Clear the annotation counts
-    annotation_counts = {}
-    shown_counts = {}  # If you are tracking shown counts separately for images
-
-    # Optionally, clear or reinitialize the dataset
-    features = Features({
-        'image_id': Value(dtype='string'),
-        'caption': Value(dtype='string'),
-        'annotation_count': Value(dtype='int32'),
-        'split': Value(dtype='string')
-    })
-    dataset = Dataset.from_dict({
-        'image_id': [],
-        'caption': [],
-        'annotation_count': [],
-        'split': []
-    }, features=features)
+# if reset:
+#     # Clear the annotation counts
+#     annotation_counts = {}
+#     shown_counts = {}  # If you are tracking shown counts separately for images
+
+#     # Optionally, clear or reinitialize the dataset
+#     features = Features({
+    #     'image_id': Value(dtype='string'),
+    #     'caption': Value(dtype='string'),
+    #     'annotation_count': Value(dtype='int32'),
+    #     'split': Value(dtype='string')
+    # })
+    # dataset = Dataset.from_dict({
+    #     'image_id': [],
+    #     'caption': [],
+    #     'annotation_count': [],
+    #     'split': []
+    # }, features=features)
     
-    # Push the reset dataset to Hugging Face or perform other necessary actions
-    dataset.push_to_hub(dataset_name)
-    print("Data has been reset.")
+    # # Push the reset dataset to Hugging Face or perform other necessary actions
+    # dataset.push_to_hub(dataset_name)
+    # print("Data has been reset.")
 
 image_folder = "images"
 image_files = [f for f in os.listdir(image_folder) if f.endswith(('.png', '.jpg', '.jpeg'))]
@@ -232,7 +236,10 @@ with gr.Blocks() as demo:
         image = gr.Image()
         caption = gr.Textbox(placeholder="Enter caption here...")
         existing_caption = gr.Textbox(label="Existing Caption", interactive=False)  # Display existing caption
-        submit = gr.Button("Submit")
+        submit = gr.Button("Submit", interactive=False)  # Initially disabled
+
+    # Enable/disable the submit button based on word count
+    caption.change(fn=check_word_count, inputs=caption, outputs=submit)
 
     # Define actions for buttons
     submit.click(fn=save_annotation, inputs=[caption, session_data], outputs=[image, caption, existing_caption])
@@ -240,4 +247,4 @@ with gr.Blocks() as demo:
     # Load initial image
     demo.load(fn=initialize_interface, inputs=session_data, outputs=[image, existing_caption])
 
-demo.launch(share=True)
+demo.launch(share=True)