Update app.py

app.py

@@ -8,9 +8,9 @@ import tempfile
 from huggingface_hub import HfApi, HfFolder
 
 DB_FAISS_PATH = 'vectorstore/db_faiss'
-DATASET_REPO = "GovindRaj/faiss-vectorstore"  # Your Hugging Face Dataset ID
+SPACE_REPO = "GovindRaj/ebiz-chatbot"  # Your Hugging Face Space ID
 
-# Function to create or update FAISS vector DB and upload to Hugging Face
+# Function to create or update FAISS vector DB and upload to Hugging Face Space
 def create_vector_db(uploaded_files):
     # Create a temporary directory
     with tempfile.TemporaryDirectory() as temp_dir:
@@ -61,14 +61,14 @@ def create_vector_db(uploaded_files):
         if not hf_token:
             raise ValueError("Hugging Face token not found. Please set the token in Hugging Face secrets.")
 
-        # Push the updated vector database to Hugging Face Dataset
+        # Push the updated vector database to Hugging Face Space
         HfFolder.save_token(hf_token)
         api = HfApi()
         api.upload_folder(
             folder_path=DB_FAISS_PATH,  # Local path to the FAISS folder
-            path_in_repo="faiss_data",  # Save at the root of the dataset
-            repo_id=DATASET_REPO,  # Hugging Face Dataset ID
-            repo_type="dataset",  # Specify this is a dataset
+            path_in_repo="faiss_data",  # Path in the Space repo
+            repo_id=SPACE_REPO,  # Hugging Face Space ID
+            repo_type="space",  # Specify that this is a Space
             token=hf_token  # Use the token from secrets
         )
 
@@ -89,9 +89,9 @@ def main():
             try:
                 success = create_vector_db(uploaded_files)
                 if success:
-                    st.success("Vector database created and uploaded successfully!")
+                    st.success("Vector database created and uploaded to your Hugging Face Space successfully!")
             except Exception as e:
                 st.error(f"An error occurred: {str(e)}")
 
 if __name__ == "__main__":
-    main()
+    main()