Update database.py

database.py

@@ -19,6 +19,24 @@ def initialize_database():
     conn.commit()
     conn.close()
 
+from huggingface_hub import HfApi
+
+def commit_to_huggingface():
+    """Commit the dataset.db file to the Hugging Face Space repository."""
+    api = HfApi()
+
+    # Replace with your Space's repository name
+    repo_id = "/Danielrahmai1991/dataset_interface"
+
+    # Upload and commit the dataset.db file
+    api.upload_file(
+        path_or_fileobj="dataset.db",
+        path_in_repo="dataset.db",
+        repo_id=repo_id,
+        repo_type="space"
+    )
+
+
 def save_to_db(chunks, topics=None):
     """Save chunks to the SQLite database."""
     # Ensure the database and table are initialized
@@ -34,4 +52,8 @@ def save_to_db(chunks, topics=None):
 
     # Commit changes and close the connection
     conn.commit()
-    conn.close()
+    conn.close()
+    commit_to_huggingface()
+
+
+