Update database.py

database.py

@@ -198,6 +198,7 @@ def insert_demo_data():
     print("📚 Demo data inserted!")
 
 def check_database_status():
+    """Simple database status check"""
     conn = get_connection()
     cursor = conn.cursor()
     
@@ -207,20 +208,35 @@ def check_database_status():
     # Show exact location
     db_path = os.path.abspath(DB_NAME)
     print(f"📍 FULL PATH: {db_path}")
-    print(f"📋 Tables: {tables}")
+    print(f"📋 Tables: {', '.join(tables)}")
     
     if os.path.exists(db_path):
-        print(f"✅ File exists! Size: {os.path.getsize(db_path)} bytes")
+        file_size = os.path.getsize(db_path)
+        print(f"✅ File exists! Size: {file_size:,} bytes ({file_size/1024:.2f} KB)")
     else:
         print(f"❌ File not found!")
+        conn.close()
+        return
     
     # Show counts
-    cursor.execute("SELECT COUNT(*) FROM Student")
-    student_count = cursor.fetchone()[0]
-    print(f"👥 Students registered: {student_count}")
-    
-    cursor.execute("SELECT COUNT(*) FROM Teacher")
-    teacher_count = cursor.fetchone()[0]
-    print(f"👨‍🏫 Teachers registered: {teacher_count}")
+    try:
+        cursor.execute("SELECT COUNT(*) FROM Student")
+        student_count = cursor.fetchone()[0]
+        print(f"👥 Students: {student_count}")
+        
+        cursor.execute("SELECT COUNT(*) FROM Teacher")
+        teacher_count = cursor.fetchone()[0]
+        print(f"👨‍🏫 Teachers: {teacher_count}")
+        
+        cursor.execute("SELECT COUNT(*) FROM ImageQuestion")
+        question_count = cursor.fetchone()[0]
+        print(f"🖼️ Questions: {question_count}")
+        
+        cursor.execute("SELECT COUNT(*) FROM Feedback")
+        feedback_count = cursor.fetchone()[0]
+        print(f"💬 Feedback: {feedback_count}")
+        
+    except Exception as e:
+        print(f"⚠️ Error counting records: {e}")
     
     conn.close()