Update app.py

app.py

@@ -1,69 +1,93 @@
 import gradio as gr
 import sqlite3
 import datetime
+import os
 
 # Create or connect to the database
-conn = sqlite3.connect("attendance.db")
-c = conn.cursor()
-
-# Create necessary tables
-c.execute('''CREATE TABLE IF NOT EXISTS users (
-                id INTEGER PRIMARY KEY AUTOINCREMENT,
-                username TEXT UNIQUE,
-                password TEXT,
-                role TEXT,
-                device_id TEXT
-            )''')
-
-c.execute('''CREATE TABLE IF NOT EXISTS attendance (
-                id INTEGER PRIMARY KEY AUTOINCREMENT,
-                username TEXT,
-                date TEXT,
-                time TEXT,
-                session TEXT
-            )''')
-conn.commit()
+def init_db():
+    conn = sqlite3.connect("attendance.db")
+    c = conn.cursor()
+
+    # Create tables if they don't exist
+    c.execute('''CREATE TABLE IF NOT EXISTS users (
+                    id INTEGER PRIMARY KEY AUTOINCREMENT,
+                    username TEXT UNIQUE,
+                    password TEXT,
+                    role TEXT,
+                    device_id TEXT
+                )''')
+
+    c.execute('''CREATE TABLE IF NOT EXISTS attendance (
+                    id INTEGER PRIMARY KEY AUTOINCREMENT,
+                    username TEXT,
+                    date TEXT,
+                    time TEXT,
+                    session TEXT
+                )''')
+    conn.commit()
+    return conn, c
+
+conn, c = init_db()
 
 # Auth functions
 def register_user(username, password, role):
     try:
         c.execute("INSERT INTO users (username, password, role) VALUES (?, ?, ?)", (username, password, role))
         conn.commit()
-        return "User registered successfully."
+        return "✅ User registered successfully."
     except sqlite3.IntegrityError:
-        return "Username already exists."
+        return "❌ Username already exists."
+    except Exception as e:
+        return f"❌ Registration error: {str(e)}"
 
 def login_user(username, password):
-    c.execute("SELECT * FROM users WHERE username=? AND password=?", (username, password))
-    return c.fetchone()
+    try:
+        c.execute("SELECT * FROM users WHERE username=? AND password=?", (username, password))
+        return c.fetchone()
+    except Exception as e:
+        return None
 
 def set_device(username, device_id):
-    c.execute("UPDATE users SET device_id=? WHERE username=?", (device_id, username))
-    conn.commit()
-    return "Device registered."
+    try:
+        c.execute("UPDATE users SET device_id=? WHERE username=?", (device_id, username))
+        conn.commit()
+        return "Device registered."
+    except Exception as e:
+        return f"❌ Device registration error: {str(e)}"
 
 def check_device(username, device_id):
-    c.execute("SELECT device_id FROM users WHERE username=?", (username,))
-    result = c.fetchone()
-    return result and result[0] == device_id
+    try:
+        c.execute("SELECT device_id FROM users WHERE username=?", (username,))
+        result = c.fetchone()
+        return result and result[0] == device_id
+    except:
+        return False
 
 # Attendance marking
 def mark_attendance(username, session):
-    date = datetime.date.today().isoformat()
-    time = datetime.datetime.now().strftime("%H:%M:%S")
-    c.execute("INSERT INTO attendance (username, date, time, session) VALUES (?, ?, ?, ?)", (username, date, time, session))
-    conn.commit()
-    return f"Attendance marked at {time} for session: {session}"
+    try:
+        date = datetime.date.today().isoformat()
+        time = datetime.datetime.now().strftime("%H:%M:%S")
+        c.execute("INSERT INTO attendance (username, date, time, session) VALUES (?, ?, ?, ?)", (username, date, time, session))
+        conn.commit()
+        return f"✅ Attendance marked at {time} for session: {session}"
+    except Exception as e:
+        return f"❌ Attendance error: {str(e)}"
 
 # View attendance dashboard
 def view_attendance():
-    c.execute("SELECT * FROM attendance ORDER BY date DESC, time DESC")
-    records = c.fetchall()
-    return "\n".join([f"{r[1]} | {r[2]} | {r[3]} | {r[4]}" for r in records])
+    try:
+        c.execute("SELECT * FROM attendance ORDER BY date DESC, time DESC")
+        records = c.fetchall()
+        if not records:
+            return "No attendance records yet."
+        return "\n".join([f"{r[1]} | {r[2]} | {r[3]} | {r[4]}" for r in records])
+    except Exception as e:
+        return f"❌ Error fetching attendance: {str(e)}"
 
 # Gradio UI
 with gr.Blocks() as app:
-    gr.Markdown("# Attendance System (No Firebase)")
+    gr.Markdown("# 📘 Attendance System (No Firebase, Local DB)")
 
     with gr.Tab("Register"):
         reg_username = gr.Text(label="Username")
@@ -78,20 +102,20 @@ with gr.Blocks() as app:
         log_password = gr.Text(label="Password", type="password")
         log_device = gr.Text(label="Device ID")
         session_input = gr.Text(label="Session ID")
-        login_button = gr.Button("Login & Mark")
+        login_button = gr.Button("Login & Mark Attendance")
         login_output = gr.Textbox(label="Result")
 
         def login_and_mark(username, password, device_id, session):
             user = login_user(username, password)
             if not user:
-                return "Login failed."
+                return "❌ Login failed. Invalid credentials."
             if not user[4]:
                 set_device(username, device_id)
                 return mark_attendance(username, session)
             elif check_device(username, device_id):
                 return mark_attendance(username, session)
             else:
-                return "Unauthorized device."
+                return "❌ Unauthorized device."
 
         login_button.click(fn=login_and_mark, inputs=[log_username, log_password, log_device, session_input], outputs=login_output)
 
@@ -105,7 +129,7 @@ with gr.Blocks() as app:
             user = login_user(username, password)
             if user and user[3] == "teacher":
                 return view_attendance()
-            return "Access denied."
+            return "❌ Access denied. Only teachers can view this."
 
         dash_button.click(fn=dashboard, inputs=[dash_username, dash_password], outputs=dash_output)