Update src/report_generator.py

src/report_generator.py

@@ -12,40 +12,52 @@ import streamlit as st
 
 # download from Hugging Face dataset
 def ensure_db():
-    import tempfile
-    db_dir = "database"
-    if os.path.exists(db_dir) and not os.path.isdir(db_dir):
-        raise RuntimeError(f"Đã tồn tại file tên '{db_dir}', không thể tạo thư mục. Hãy xóa hoặc đổi tên file này.")
+    repo_path = os.path.join(os.getcwd(), "database", "stock_insights.db")
+    if os.path.exists(repo_path):
+        return repo_path
+
+    candidates = [
+        os.path.join("/app", "database", "stock_insights.db"),
+        os.path.join("/tmp", "database", "stock_insights.db"),
+        os.path.join("/tmp", "stock_insights.db"),
+    ]
+    for p in candidates:
+        if os.path.exists(p):
+            return p
+
     try:
-        os.makedirs(db_dir, exist_ok=True)
-        db_path = os.path.join(db_dir, "stock_insights.db")
-        local_dir = db_dir
-    except Exception:
-        temp_dir = os.path.join(tempfile.gettempdir(), "stockinsights_db")
-        os.makedirs(temp_dir, exist_ok=True)
-        db_path = os.path.join(temp_dir, "stock_insights.db")
-        local_dir = temp_dir
-    if not os.path.exists(db_path):
         from huggingface_hub import hf_hub_download
-        hf_hub_download(
+        tmp_dir = os.path.join("/tmp", "database")
+        os.makedirs(tmp_dir, exist_ok=True)
+        local_file = hf_hub_download(
             repo_id="PuppetLover/stock_insights",
             filename="stock_insights.db",
             repo_type="dataset",
-            local_dir=local_dir,
+            local_dir=tmp_dir,
             local_dir_use_symlinks=False,
         )
-    return db_path
+        return local_file
+    except Exception as e:
+        local_rel = os.path.join("database", "stock_insights.db")
+        if os.path.exists(local_rel):
+            return local_rel
+        raise RuntimeError(f"Cannot access or download database file: {e}")
 
-DB_PATH = ensure_db()
+# gọi và gán hằng DB_PATH dùng trong module
+DB_PATH = ensure_db()   
 
 def generate_stock_report(stock_code, time_period):
+
     start_date, end_date = time_period
     today = datetime.now().date()
+    # db_path = os.path.join("database", "stock_insights.db")
     db_path = DB_PATH
+
     report = {
         "stock_code": stock_code,
         "report_period": f"{start_date} to {end_date}"
     }
+
     with sqlite3.connect(db_path) as conn:
         # Tạo bảng tạm relevant_articles
         conn.execute("DROP TABLE IF EXISTS relevant_articles;")