Update database.py

database.py

@@ -6,13 +6,20 @@ metadata_obj = MetaData()
 
 # Function to check existing tables
 def get_existing_tables():
-    """Returns a list of existing tables in the database."""
+    """
+    Returns a list of existing tables in the database.
+
+    Returns:
+        list: List of table names.
+    """
     inspector = inspect(engine)
     return inspector.get_table_names()
 
 # Function to initialize database if empty
 def initialize_database():
-    """Checks if database is empty and prompts user to upload an SQL file if needed."""
+    """
+    Checks if database is empty and prompts user to upload an SQL file if needed.
+    """
     tables = get_existing_tables()
     if not tables:
         print("No tables found in the database. Please upload an SQL file.")
@@ -20,3 +27,4 @@ def initialize_database():
         print(f"Database initialized with tables: {tables}")
 
 initialize_database()
+