Update certificate_upload_module.py

certificate_upload_module.py

@@ -2,7 +2,7 @@ import tempfile
 import time
 import logging
 from PIL import Image as PILImage
-from datasets import Dataset, Image
+from datasets import Dataset, Image, concatenate_datasets
 from huggingface_hub import HfApi, login
 import pandas as pd
 import os
@@ -12,7 +12,6 @@ from datetime import datetime
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
-# Replace with your actual certificate dataset name
 CERTIFICATE_DATASET_NAME = "ysharma/gradio-agents-mcp-hackathon-certificates"
 
 def safe_add_certificate_to_dataset(certificate_image, hf_username, max_retries=5, retry_delay=3):
@@ -118,7 +117,8 @@ def safe_add_certificate_to_dataset(certificate_image, hf_username, max_retries=
             # Combine with existing dataset or use new dataset if empty
             if existing_dataset is not None and not is_empty_dataset:
                 try:
-                    combined_dataset = existing_dataset.concatenate_datasets([new_dataset])
+                    #combined_dataset = existing_dataset.concatenate_datasets([new_dataset])
+                    combined_dataset = concatenate_datasets([existing_dataset, new_dataset])
                     logger.info(f"Combined dataset now has {len(combined_dataset)} certificates (was {len(existing_dataset)})")
                 except Exception as concat_error:
                     logger.error(f"Failed to combine datasets: {concat_error}")