Update app.py

app.py

@@ -6,31 +6,39 @@ from datetime import datetime
 # Database Setup
 db_file = "attendance_records.db"
 
-conn = sqlite3.connect(db_file)
-cursor = conn.cursor()
-
-# Create attendance table if not exists
-cursor.execute("""
-CREATE TABLE IF NOT EXISTS attendance (
-    id INTEGER PRIMARY KEY AUTOINCREMENT,
-    name TEXT,
-    day TEXT,
-    date TEXT,
-    status TEXT
-)
-""")
-conn.commit()
-
 # Helper Functions
+def get_db_connection():
+    """Create a new database connection."""
+    conn = sqlite3.connect(db_file)
+    return conn
+
 def log_attendance(name, day, date, status):
+    conn = get_db_connection()
+    cursor = conn.cursor()
+
+    cursor.execute("""
+    CREATE TABLE IF NOT EXISTS attendance (
+        id INTEGER PRIMARY KEY AUTOINCREMENT,
+        name TEXT,
+        day TEXT,
+        date TEXT,
+        status TEXT
+    )
+    """)
+    conn.commit()
+
     cursor.execute("""
     INSERT INTO attendance (name, day, date, status)
     VALUES (?, ?, ?, ?)
     """, (name, day, date, status))
     conn.commit()
+    conn.close()
     return "Attendance logged successfully!"
 
 def calculate_fees():
+    conn = get_db_connection()
+    cursor = conn.cursor()
+
     # Calculate attendance fees
     cursor.execute("""
     SELECT name, COUNT(*) * (1000 / 12) AS fees
@@ -40,6 +48,7 @@ def calculate_fees():
     """)
     fees_data = cursor.fetchall()
     fees_dict = {row[0]: row[1] for row in fees_data}
+    conn.close()
     return fees_dict
 
 def create_end_of_month_table():
@@ -47,6 +56,9 @@ def create_end_of_month_table():
     if today.day != pd.Period(today.strftime("%Y-%m")).days_in_month:
         return "It's not the end of the month yet."
 
+    conn = get_db_connection()
+    cursor = conn.cursor()
+
     # Create end-of-month table
     month = today.strftime("%Y-%m")
     table_name = f"fees_{month.replace('-', '_')}"
@@ -78,6 +90,7 @@ def create_end_of_month_table():
         """, (name, email, fees))
 
     conn.commit()
+    conn.close()
     return f"End-of-month table '{table_name}' created successfully!"
 
 def submit_attendance(name, day, date, status):