update database

database.py

@@ -3,19 +3,27 @@ import json
 import pandas as pd
 import re
 import streamlit as st
+import logging
 
+# Configure logging
+logging.basicConfig(level=logging.INFO)  # Set logging level to INFO
 
 # Database connection
 def initialize_database():
-    # Database Connection
-    db_params = {"host": st.secrets["host"],
-                "user": st.secrets["username"],
-                "password": st.secrets["password"],
-                "port": int(st.secrets["port"]),
-                "database": st.secrets["database"]
-                }
-    db = pymysql.connect(**db_params)
-    return db
+    try:
+        # Database Connection
+        db_params = {"host": st.secrets["host"],
+                     "user": st.secrets["username"],
+                     "password": st.secrets["password"],
+                     "port": int(st.secrets["port"]),
+                     "database": st.secrets["database"]
+                     }
+        db = pymysql.connect(**db_params)
+        logging.info("Connected to the database successfully!")
+        return db
+    except pymysql.MySQLError as e:
+        logging.error("Error connecting to the database: %s", e)
+        raise  # Re-raise the exception to propagate it up the call stack
 
 def execute_query(query):
     db = initialize_database()
@@ -25,9 +33,10 @@ def execute_query(query):
         description = cursor.description
         result = cursor.fetchall()  # Fetch all rows from the result set
         db.commit()
+        logging.info("Query executed successfully: %s", query)
         return description, result
     except Exception as e:
-        print("Error executing query:", e)
+        logging.error("Error executing query: %s", e)
         db.rollback()
         return None  # Return None if an error occurs
     finally: