Upload database.py

back_end/application/database.py

@@ -1,23 +1,25 @@
-import os
-import sqlite3
-import pandas as pd
-
-# DB_PATH = r"E:/Healthcare_chat_AI/data/veludb.db"
-DB_PATH = f"{os.path.dirname(os.path.dirname(os.path.dirname(__file__)))}/data/veludb.db"
-print(DB_PATH)
-# print("Database exists:", os.path.exists(DB_PATH))
-
-
-def fetch_data():
-    conn = sqlite3.connect(DB_PATH)
-    query = "SELECT * FROM healthcare_data;"  # Fetch 5 rows
-    df = pd.read_sql(query, conn)
-    conn.close()
-    return df
-
-df =fetch_data()
-df.drop_duplicates(inplace=True)
-# print(df)  # Display the first 5 rows
-
-
-
+import sqlite3
+import pandas as pd
+import os
+
+DB_PATH = "/workspace/data/veludb.db"
+
+# Ensure database exists before connecting
+if not os.path.exists(DB_PATH):
+    raise FileNotFoundError(f"❌ Database file not found at {DB_PATH}")
+
+def fetch_data():
+    try:
+        conn = sqlite3.connect(DB_PATH)
+        query = "SELECT * FROM healthcare_data;"
+        df = pd.read_sql(query, conn)
+        conn.close()
+        return df
+    except sqlite3.Error as e:
+        print(f"❌ Database error: {e}")
+        return None
+
+df = fetch_data()
+if df is not None:
+    df.drop_duplicates(inplace=True)
+    print("✅ Data loaded successfully!")