velmurugan1122 commited on
Commit
dea8fe5
Β·
verified Β·
1 Parent(s): 7ff3acf

Upload database.py

Browse files
Files changed (1) hide show
  1. back_end/application/database.py +16 -18
back_end/application/database.py CHANGED
@@ -1,25 +1,23 @@
 
1
  import sqlite3
2
  import pandas as pd
3
- import os
4
 
5
- DB_PATH = "/workspace/data/veludb.db"
 
 
 
6
 
7
- # Ensure database exists before connecting
8
- if not os.path.exists(DB_PATH):
9
- raise FileNotFoundError(f"❌ Database file not found at {DB_PATH}")
10
 
11
  def fetch_data():
12
- try:
13
- conn = sqlite3.connect(DB_PATH)
14
- query = "SELECT * FROM healthcare_data;"
15
- df = pd.read_sql(query, conn)
16
- conn.close()
17
- return df
18
- except sqlite3.Error as e:
19
- print(f"❌ Database error: {e}")
20
- return None
 
 
21
 
22
- df = fetch_data()
23
- if df is not None:
24
- df.drop_duplicates(inplace=True)
25
- print("βœ… Data loaded successfully!")
 
1
+ import os
2
  import sqlite3
3
  import pandas as pd
 
4
 
5
+ # DB_PATH = r"E:/Healthcare_chat_AI/data/veludb.db"
6
+ DB_PATH = f"{os.path.dirname(os.path.dirname(os.path.dirname(__file__)))}/data/veludb.db"
7
+ # print(DB_PATH)
8
+ # print("Database exists:", os.path.exists(DB_PATH))
9
 
 
 
 
10
 
11
  def fetch_data():
12
+ conn = sqlite3.connect(DB_PATH)
13
+ query = "SELECT * FROM healthcare_data;" # Fetch 5 rows
14
+ df = pd.read_sql(query, conn)
15
+ conn.close()
16
+ return df
17
+
18
+ df =fetch_data()
19
+ df.drop_duplicates(inplace=True)
20
+ # print(df) # Display the first 5 rows
21
+
22
+
23