Db Deployment handles

backend/app/db/db_connector.py

@@ -1,6 +1,7 @@
 from sqlalchemy import create_engine
 from sqlalchemy.engine import Engine
 from sqlalchemy.orm import sessionmaker
+from app.services.utility import UtilityClass
 import sqlite3
 from pathlib import Path
 import os
@@ -9,10 +10,11 @@ load_dotenv()
 
 # Path to SQLite DB file, configurable via environment variable or .env
 print("DB_PATH from env:", os.getenv("DB_PATH"))
-DB_PATH = os.getenv("DB_PATH")
+# DB_PATH = os.getenv("DB_PATH")
+DB_PATH = UtilityClass.download_sqlite_db_from_dropbox(os.getenv("DROPBOX_TOKEN"), os.getenv("DROPBOX_DB_PATH"))
 if not DB_PATH:
     BASE_DIR = os.path.dirname(os.path.abspath(__file__))
-    DB_PATH = os.path.join(BASE_DIR, "BankingDb.db")
+    DB_PATH = os.path.join(BASE_DIR, "Banking.db")
 
 DATABASE_URL = f"sqlite:///{DB_PATH}"
 

backend/app/services/utility.py

@@ -1,8 +1,10 @@
+import dropbox
 import openai
 from config.settings import settings
 from langchain_openai import ChatOpenAI
 import sqlparse
 import re
+import os
 class UtilityClass:
     """
     Classifies user input as 'sql' (SQL/data context) or 'chat' (normal conversation) using an LLM.
@@ -102,4 +104,19 @@ class UtilityClass:
 
         return False
 
-
+    @staticmethod
+    def download_sqlite_db_from_dropbox(dropbox_token: str, dropbox_db_path: str, local_db_path: str = None) -> str:
+        """
+        Downloads a SQLite DB file from Dropbox and returns the local file path.
+        :param dropbox_token: Dropbox API access token.
+        :param dropbox_db_path: Path to the DB file in Dropbox (e.g., '/folder/mydb.db').
+        :param local_db_path: Local path to save the file. If None, saves to './downloaded_db.db'.
+        :return: Local file path of the downloaded DB.
+        """
+        if local_db_path is None:
+            local_db_path = os.path.join(os.getcwd(), "downloaded_db.db")
+        dbx = dropbox.Dropbox(dropbox_token)
+        with open(local_db_path, 'wb') as f:
+            metadata, res = dbx.files_download(path=dropbox_db_path)
+            f.write(res.content)
+        return local_db_path

backend/requirements.txt

@@ -12,4 +12,4 @@ faiss-cpu          # optional for embeddings
 pandas
 matplotlib
 python-dotenv
-
+dropbox