Upload 2 files

app/database.py

@@ -1,12 +1,21 @@
 import sqlite3
 from typing import List, Dict
-from pathlib import Path
-
-# Chemin vers la base SQLite
-DB_PATH = Path("data/articles.db")
-
-def get_connection():
-    conn = sqlite3.connect(DB_PATH)
+import os
+from huggingface_hub import hf_hub_download
+
+# Télécharger le fichier SQLite depuis le dataset
+REPO_ID = "Loren/articles_db"  # dataset HF
+DB_NAME = 'articles.db'
+hf_token = os.environ["API_HF_TOKEN"]
+sqlite_path = hf_hub_download(
+    repo_id=REPO_ID,
+    filename=DB_NAME,
+    repo_type="dataset",
+    token=hf_token
+)
+
+def get_connection(sqlite_path):
+    conn = sqlite3.connect(sqlite_path)
     conn.row_factory = sqlite3.Row
     return conn