Update app.py

app.py

@@ -11,32 +11,49 @@ auth = HTTPBasicAuth()
 
 # Hàm kết nối với SQLite
 def get_db_connection():
-    db_path = '/app/database.db'
-    db_dir = os.path.dirname(db_path)
+    # Đường dẫn tạm thời để lưu database.db
+    temp_db_path = '/tmp/database.db'
 
-    # Tạo thư mục /app/ nếu chưa tồn tại
-    if not os.path.exists(db_dir):
-        try:
-            os.makedirs(db_dir, exist_ok=True)
-        except Exception as e:
-            raise RuntimeError(f"Không thể tạo thư mục {db_dir}: {str(e)}")
-
-    # Kiểm tra quyền ghi
-    if not os.access(db_dir, os.W_OK):
-        # Nếu không có quyền ghi, thử sử dụng /tmp/
-        db_path = '/tmp/database.db'
-        db_dir = os.path.dirname(db_path)
-        if not os.path.exists(db_dir):
-            os.makedirs(db_dir, exist_ok=True)
-        if not os.access(db_dir, os.W_OK):
-            raise RuntimeError(f"Không có quyền ghi tại {db_dir} và /tmp/")
+    # Thông tin dataset trên Hugging Face
+    repo_id = "huylaughmad/web"
+    filename = "database.db"
 
     try:
-        conn = sqlite3.connect(db_path)
-        conn.row_factory = sqlite3.Row  # Để trả về dữ liệu dưới dạng dictionary
+        # Tải file database.db từ Hugging Face Dataset
+        downloaded_path = hf_hub_download(
+            repo_id=repo_id,
+            filename=filename,
+            repo_type="dataset",
+            token=os.getenv("HF_TOKEN", None)  # Token nếu dataset yêu cầu xác thực
+        )
+
+        # Sao chép file từ đường dẫn tải về sang /tmp/database.db
+        with open(downloaded_path, 'rb') as src, open(temp_db_path, 'wb') as dst:
+            dst.write(src.read())
+
+        # Kiểm tra quyền ghi vào /tmp/
+        if not os.access(os.path.dirname(temp_db_path), os.W_OK):
+            raise RuntimeError(f"Không có quyền ghi tại {os.path.dirname(temp_db_path)}")
+
+        # Kết nối với SQLite
+        conn = sqlite3.connect(temp_db_path)
+        conn.row_factory = sqlite3.Row  # Trả về dữ liệu dưới dạng dictionary
         return conn
-    except sqlite3.OperationalError as e:
-        raise RuntimeError(f"Không thể mở file cơ sở dữ liệu tại {db_path}: {str(e)}")
+
+    except Exception as e:
+        # Nếu không tìm thấy file hoặc có lỗi, tạo mới database.db trong /tmp/
+        if not os.path.exists(temp_db_path):
+            try:
+                open(temp_db_path, 'a').close()  # Tạo file rỗng
+            except Exception as create_error:
+                raise RuntimeError(f"Không thể tạo file cơ sở dữ liệu tại {temp_db_path}: {str(create_error)}")
+
+        try:
+            conn = sqlite3.connect(temp_db_path)
+            conn.row_factory = sqlite3.Row
+            return conn
+        except sqlite3.OperationalError as sql_error:
+            raise RuntimeError(f"Không thể mở file cơ sở dữ liệu tại {temp_db_path}: {str(sql_error)}")
 
 # Tạo các bảng nếu chưa tồn tại
 def init_db():