Update app.py

app.py

@@ -5,7 +5,8 @@ import time
 from datetime import datetime
 import json
 import shutil
-from huggingface_hub import HfApi, hf_hub_download
+
+from huggingface_hub import HfApi, hf_hub_download, create_repo
 
 # -----------------------------
 # CONFIG & HF CONNECTION
@@ -21,10 +22,22 @@ def save_theme_to_dataset(xml_file):
         return "❌ Please upload an XML file first."
         
     try:
-        # Safely get the file path whether Gradio returned a string or an object
+        # Safely get the file path
         file_path = xml_file if isinstance(xml_file, str) else xml_file.name
         
-        # Upload directly using HfApi without copying to a local temp file first
+        # NEW: Automatically create the dataset repository if it doesn't exist
+        try:
+            create_repo(
+                repo_id=DATASET_REPO, 
+                repo_type="dataset", 
+                private=True, 
+                exist_ok=True, 
+                token=HF_TOKEN
+            )
+        except Exception as repo_err:
+            return f"❌ Failed to access or create repository: {repo_err}. Check your HF_TOKEN permissions."
+
+        # Upload directly using HfApi
         api = HfApi()
         api.upload_file(
             path_or_fileobj=file_path,