huylaughmad commited on
Commit
13a6bc6
·
verified ·
1 Parent(s): 3847f4d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -9
app.py CHANGED
@@ -1,11 +1,13 @@
1
  import os
2
  import sqlite3
3
  import json
4
- import logging
5
- from huggingface_hub import hf_hub_download, upload_file
6
  from flask import Flask, render_template, request, redirect, url_for, abort, flash, jsonify
7
  from flask_httpauth import HTTPBasicAuth
8
  from werkzeug.security import generate_password_hash, check_password_hash
 
 
 
 
9
 
10
  app = Flask(__name__)
11
  app.config['SECRET_KEY'] = os.getenv("FLASK_SECRET_KEY", "your-secret-key-here")
@@ -36,13 +38,9 @@ def get_db_connection():
36
  if not hf_token:
37
  raise RuntimeError("HF_TOKEN không được thiết lập trong biến môi trường")
38
 
39
- # Khởi tạo Hugging Face API
40
- api = HfApi()
41
-
42
  # Khóa file để tránh xung đột giữa các worker
43
  lock_file = '/tmp/database.lock'
44
  with open(lock_file, 'a'):
45
- import fcntl
46
  fcntl.flock(lock_file, fcntl.LOCK_EX)
47
  try:
48
  # Kiểm tra xem file database.db đã tồn tại cục bộ chưa
@@ -457,9 +455,6 @@ def init_db():
457
  conn.commit()
458
  conn.close()
459
 
460
- # Gọi hàm init_db khi ứng dụng khởi động
461
- init_db()
462
-
463
  # Bảo mật cơ bản cho CMS
464
  ADMIN_USERNAME = os.getenv("ADMIN_USERNAME", "admin")
465
  ADMIN_PASSWORD_HASH = generate_password_hash(os.getenv("ADMIN_PASSWORD", "password"))
 
1
  import os
2
  import sqlite3
3
  import json
 
 
4
  from flask import Flask, render_template, request, redirect, url_for, abort, flash, jsonify
5
  from flask_httpauth import HTTPBasicAuth
6
  from werkzeug.security import generate_password_hash, check_password_hash
7
+ from huggingface_hub import hf_hub_download, upload_file, HfApi
8
+ import logging
9
+ import tempfile
10
+ import fcntl
11
 
12
  app = Flask(__name__)
13
  app.config['SECRET_KEY'] = os.getenv("FLASK_SECRET_KEY", "your-secret-key-here")
 
38
  if not hf_token:
39
  raise RuntimeError("HF_TOKEN không được thiết lập trong biến môi trường")
40
 
 
 
 
41
  # Khóa file để tránh xung đột giữa các worker
42
  lock_file = '/tmp/database.lock'
43
  with open(lock_file, 'a'):
 
44
  fcntl.flock(lock_file, fcntl.LOCK_EX)
45
  try:
46
  # Kiểm tra xem file database.db đã tồn tại cục bộ chưa
 
455
  conn.commit()
456
  conn.close()
457
 
 
 
 
458
  # Bảo mật cơ bản cho CMS
459
  ADMIN_USERNAME = os.getenv("ADMIN_USERNAME", "admin")
460
  ADMIN_PASSWORD_HASH = generate_password_hash(os.getenv("ADMIN_PASSWORD", "password"))