Update app.py

app.py

@@ -5,9 +5,9 @@ import os
 import json
 import io
 import subprocess  # To call process_hf_dataset.py
-from database import init_chromadb, store_program, query_programs, load_chromadb_from_hf, DB_NAME, create_collection, save_chromadb_to_hf
-import logging
+from database import init_chromadb, store_program, query_programs, load_chromadb_from_hf, DB_NAME, create_collection, save_chromadb_to_hf, HF_DATASET_NAME
 from datasets import Dataset
+import logging
 
 # Set up logging
 logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
@@ -142,7 +142,7 @@ def index():
                         "program_vectors": []
                     }
                     empty_dataset = Dataset.from_dict(empty_data)
-                    empty_dataset.push_to_hub(HF_DATASET_NAME, token=os.getenv("HF_KEY"))
+                    empty_dataset.push_to_hub(HF_DATASET_NAME, token=os.getenv("HF_KEY"), exist_ok=True)
                     logger.info(f"Replaced Hugging Face dataset {HF_DATASET_NAME} with empty dataset")
                 except Exception as e:
                     logger.error(f"Error replacing Hugging Face dataset: {e}")
@@ -240,7 +240,7 @@ def index():
                         "program_vectors": []
                     }
                     empty_dataset = Dataset.from_dict(empty_data)
-                    empty_dataset.push_to_hub(HF_DATASET_NAME, token=os.getenv("HF_KEY"))
+                    empty_dataset.push_to_hub(HF_DATASET_NAME, token=os.getenv("HF_KEY"), exist_ok=True)
                     logger.info(f"Replaced Hugging Face dataset {HF_DATASET_NAME} with empty dataset")
                 except Exception as e:
                     logger.error(f"Error replacing Hugging Face dataset: {e}")