updates

app.py

@@ -3,12 +3,6 @@ import os
 import threading
 import random
 from datasets import load_dataset, Dataset, Features, Value
-from huggingface_hub import login
-token = os.getenv("HUGGINGFACE_TOKEN")
-if token:
-    login(token=token)
-else:
-    print("HUGGINGFACE_TOKEN environment variable not set.")
 
 # Set up your Hugging Face dataset name
 dataset_name = "GeorgeIbrahim/EGYCOCO"  # Replace with your dataset name
@@ -28,6 +22,7 @@ except Exception as e:
 image_folder = "test"
 image_files = [f for f in os.listdir(image_folder) if f.endswith(('.png', '.jpg', '.jpeg'))]
 lock = threading.Lock()
+current_image = None
 
 # Function to get a random image that hasn’t been annotated or skipped
 def get_next_image():
@@ -52,7 +47,7 @@ def save_annotation(caption):
 
         # Add the new annotation to the dataset
         new_data = {"image_id": image_id, "caption": caption}
-        dataset = dataset.add_item(new_data)
+        dataset.add_item(new_data)  # Update the dataset variable
 
         # Save updated dataset to Hugging Face
         dataset.push_to_hub(dataset_name)
@@ -74,7 +69,7 @@ def skip_image():
         image_id = os.path.basename(current_image)
         # Record "skipped" caption directly here
         new_data = {"image_id": image_id, "caption": "skipped"}
-        dataset = dataset.add_item(new_data)
+        dataset.add_item(new_data)  # Update the dataset variable
 
         # Save updated dataset to Hugging Face
         dataset.push_to_hub(dataset_name)