Upload Home.py

Home.py

@@ -342,7 +342,7 @@ def check_valid_name():
 
 def query_excecuter_postgres(
     query,
-    db_path=r"db\imp_db.db",
+    db_path=None,
     params=None,
     insert=True,
     insert_retrieve=False,
@@ -360,8 +360,13 @@ def query_excecuter_postgres(
 
     """
     try:
+        # Construct a cross-platform path to the database
+        db_dir = os.path.join("db")
+        os.makedirs(db_dir, exist_ok=True)  # Make sure the directory exists
+        db_path = os.path.join(db_dir, "imp_db.db")
+
         # Establish connection to the SQLite database
-        conn = sqlite3.connect(r"db\imp_db.db")
+        conn = sqlite3.connect(db_path)
     except sqlite3.Error as e:
         st.warning(f"Unable to connect to the SQLite database: {e}")
         st.stop()