better dummy data

self_hosted_env/voice_agent/apply_test_schema.py

@@ -30,11 +30,23 @@ CREATE INDEX IF NOT EXISTS idx_students_grade ON students(grade);
 CREATE INDEX IF NOT EXISTS idx_students_grade_language ON students(grade, language);
 
 -- Insert dummy data for testing
-INSERT INTO students (student_id, student_name, grade, language, nationality) VALUES 
-    ('student_001', 'Ahmed Ali', 4, TRUE, 'EGYPTIAN'),
-    ('student_002', 'Sara Hassan', 6, FALSE, 'SAUDI'),
-    ('student_003', 'Mona Adel', 5, TRUE, 'EGYPTIAN'),
-    ('student_004', 'Omar Youssef', 6, FALSE, 'SAUDI')
+INSERT INTO students (student_id, student_name, grade, language, nationality) VALUES
+ -- Arabic Grade 4: One Egyptian, One Saudi
+ ('student_001', 'Ahmed Ali', 4, TRUE, 'EGYPTIAN'),
+ ('student_002', 'Khalid Al-Rashid', 4, TRUE, 'SAUDI'),
+
+-- Arabic Grade 6: One Egyptian, One Saudi
+ ('student_003', 'Fatima Hassan', 6, TRUE, 'EGYPTIAN'),
+ ('student_004', 'Nora Al-Zahrani', 6, TRUE, 'SAUDI'),
+
+-- English Grade 5: One Egyptian, One Saudi
+ ('student_005', 'Mona Adel', 5, FALSE, 'EGYPTIAN'),
+ ('student_006', 'Sara Al-Mutairi', 5, FALSE, 'SAUDI'),
+
+-- English Grade 6: One Egyptian, One Saudi
+ ('student_007', 'Omar Youssef', 6, FALSE, 'EGYPTIAN'),
+ ('student_008', 'Abdullah Al-Harbi', 6, FALSE, 'SAUDI')
+
 ON CONFLICT (student_id) DO NOTHING;
 """
 
@@ -102,7 +114,7 @@ def setup_database(drop_existing_tables: bool = False):
 
 if __name__ == "__main__":
     # To run with a clean slate, pass True
-    # setup_database(drop_existing_tables=True)
+    setup_database(drop_existing_tables=True)
     
     # To run without dropping tables (default)
-    setup_database()
+    #setup_database()
\ No newline at end of file