load_sample_data.py

"""
Sample Data Loader for Academic Evaluation System

This script populates the database with sample data for testing and demonstration.
Run after migrations and ML model training.

Usage:
    python manage.py shell < load_sample_data.py
    
Or run in Django shell:
    python manage.py shell
    >>> exec(open('load_sample_data.py').read())
"""

import os
import django

# Setup Django environment
os.environ.setdefault('DJANGO_SETTINGS_MODULE', 'academic_evaluation.settings')
django.setup()

from django.contrib.auth.models import User
from django.utils import timezone
from datetime import datetime, timedelta
import random

from evaluation.models import (
    Student, Project, Team, RubricCategory, Evaluation,
    AttendanceRecord, Assignment, AssignmentSubmission
)
from evaluation.ml_service import update_student_risk_status

print("=" * 70)
print("Loading Sample Data for Academic Evaluation System")
print("=" * 70)
print()

# ============================================================================
# 1. Create Rubric Categories
# ============================================================================
print("1. Creating Rubric Categories...")

rubric_categories = [
    {
        'name': 'Technical Skills',
        'description': 'Programming, coding standards, and technical implementation',
        'max_score': 100,
        'weight': 1.5
    },
    {
        'name': 'Communication',
        'description': 'Presentation skills, documentation, and articulation',
        'max_score': 100,
        'weight': 1.0
    },
    {
        'name': 'Teamwork',
        'description': 'Collaboration, team dynamics, and contribution',
        'max_score': 100,
        'weight': 1.0
    },
    {
        'name': 'Problem Solving',
        'description': 'Analytical thinking and problem-solving approach',
        'max_score': 100,
        'weight': 1.2
    },
    {
        'name': 'Innovation',
        'description': 'Creative thinking and innovative solutions',
        'max_score': 100,
        'weight': 1.3
    }
]

for cat_data in rubric_categories:
    category, created = RubricCategory.objects.get_or_create(
        name=cat_data['name'],
        defaults={
            'description': cat_data['description'],
            'max_score': cat_data['max_score'],
            'weight': cat_data['weight']
        }
    )
    if created:
        print(f"   ✓ Created: {category.name}")
    else:
        print(f"   - Exists: {category.name}")

print()

# ============================================================================
# 2. Create Projects
# ============================================================================
print("2. Creating Sample Projects...")

projects_data = [
    {
        'project_id': 'PROJ001',
        'title': 'E-Commerce Web Application',
        'description': 'Full-stack e-commerce platform with payment integration',
        'department': 'Computer Science',
        'semester': 6
    },
    {
        'project_id': 'PROJ002',
        'title': 'IoT Smart Home System',
        'description': 'IoT-based home automation using sensors and actuators',
        'department': 'Electronics',
        'semester': 7
    },
    {
        'project_id': 'PROJ003',
        'title': 'Machine Learning Disease Predictor',
        'description': 'ML model for early disease detection using patient data',
        'department': 'Computer Science',
        'semester': 8
    },
    {
        'project_id': 'PROJ004',
        'title': 'Mobile Fitness Tracker',
        'description': 'Cross-platform mobile app for fitness tracking',
        'department': 'Information Technology',
        'semester': 6
    }
]

projects = []
for proj_data in projects_data:
    start_date = timezone.now().date() - timedelta(days=90)
    end_date = timezone.now().date() + timedelta(days=30)
    
    project, created = Project.objects.get_or_create(
        project_id=proj_data['project_id'],
        defaults={
            'title': proj_data['title'],
            'description': proj_data['description'],
            'department': proj_data['department'],
            'semester': proj_data['semester'],
            'start_date': start_date,
            'end_date': end_date,
            'max_team_size': 4
        }
    )
    projects.append(project)
    if created:
        print(f"   ✓ Created: {project.project_id} - {project.title}")
    else:
        print(f"   - Exists: {project.project_id}")

print()

# ============================================================================
# 3. Create Sample Students
# ============================================================================
print("3. Creating Sample Students...")

departments = ['Computer Science', 'Electronics', 'Information Technology', 'Mechanical']
first_names = ['Alice', 'Bob', 'Charlie', 'Diana', 'Eve', 'Frank', 'Grace', 'Henry', 'Ivy', 'Jack']
last_names = ['Anderson', 'Brown', 'Chen', 'Davis', 'Evans', 'Garcia', 'Hall', 'Ibrahim', 'Johnson', 'Kumar']

students = []
for i in range(20):
    student_id = f'STU{2024001 + i}'
    
    # Create varied student profiles
    if i < 12:  # Safe students
        cgpa = round(random.uniform(6.5, 9.5), 2)
        attendance = round(random.uniform(75, 98), 2)
        assignment_comp = round(random.uniform(80, 100), 2)
        participation = round(random.uniform(70, 95), 2)
    else:  # At-risk students
        cgpa = round(random.uniform(3.5, 6.0), 2)
        attendance = round(random.uniform(45, 70), 2)
        assignment_comp = round(random.uniform(40, 75), 2)
        participation = round(random.uniform(35, 65), 2)
    
    first_name = random.choice(first_names)
    last_name = random.choice(last_names)
    department = random.choice(departments)
    semester = random.randint(5, 8)
    
    student, created = Student.objects.get_or_create(
        student_id=student_id,
        defaults={
            'first_name': first_name,
            'last_name': last_name,
            'email': f'{first_name.lower()}.{last_name.lower()}@university.edu',
            'phone': f'+91-98{random.randint(10000000, 99999999)}',
            'department': department,
            'semester': semester,
            'enrollment_year': 2024 - (semester // 2),
            'cgpa': cgpa,
            'attendance_percentage': attendance,
            'assignment_completion': assignment_comp,
            'participation_score': participation
        }
    )
    
    if created:
        students.append(student)
        print(f"   ✓ Created: {student.student_id} - {student.full_name}")
        
        # Update ML risk status
        try:
            update_student_risk_status(student)
            print(f"      Risk: {student.risk_status} ({student.risk_confidence:.2%})")
        except Exception as e:
            print(f"      ⚠ Risk prediction failed: {e}")
    else:
        students.append(student)
        print(f"   - Exists: {student.student_id}")

print()

# ============================================================================
# 4. Create Teams
# ============================================================================
print("4. Creating Teams...")

# Create 5 teams
for i in range(5):
    team_id = f'TEAM{i+1:03d}'
    team_name = f'Team {chr(65+i)}'  # Team A, Team B, etc.
    
    # Assign 3-4 students per team
    team_students = students[i*4:(i+1)*4]
    if not team_students:
        continue
    
    project = projects[i % len(projects)]
    
    team, created = Team.objects.get_or_create(
        team_id=team_id,
        defaults={
            'team_name': team_name,
            'project': project,
            'team_leader': team_students[0] if team_students else None
        }
    )
    
    if created:
        team.students.set(team_students)
        print(f"   ✓ Created: {team.team_id} - {team.team_name} ({len(team_students)} members)")
    else:
        print(f"   - Exists: {team.team_id}")

print()

# ============================================================================
# 5. Create Evaluations
# ============================================================================
print("5. Creating Sample Evaluations...")

# Create admin user if doesn't exist for evaluator
admin_user, _ = User.objects.get_or_create(
    username='admin',
    defaults={
        'email': 'admin@university.edu',
        'is_staff': True,
        'is_superuser': True
    }
)

evaluation_types = ['Mid-Term', 'Final', 'Weekly', 'Presentation']
categories = list(RubricCategory.objects.all())

eval_count = 0
for student in students[:10]:  # Create evaluations for first 10 students
    project = random.choice(projects)
    
    # Create 2-3 evaluations per student
    for _ in range(random.randint(2, 3)):
        eval_id = f'EVAL{2024001 + eval_count}'
        eval_type = random.choice(evaluation_types)
        category = random.choice(categories)
        
        # Score correlates with student performance
        base_score = (student.cgpa / 10) * 100
        score = max(0, min(100, base_score + random.uniform(-15, 15)))
        
        evaluation, created = Evaluation.objects.get_or_create(
            evaluation_id=eval_id,
            defaults={
                'student': student,
                'project': project,
                'evaluator': admin_user,
                'evaluation_type': eval_type,
                'evaluation_date': timezone.now() - timedelta(days=random.randint(1, 60)),
                'rubric_category': category,
                'score': round(score, 2),
                'max_score': 100.0,
                'comments': f'Good work on {category.name.lower()}. Keep it up!',
                'strengths': 'Strong technical understanding',
                'improvements': 'Can improve on time management'
            }
        )
        
        if created:
            eval_count += 1
            if eval_count <= 5:  # Show first 5
                print(f"   ✓ {eval_id}: {student.student_id} - {eval_type} - {score:.1f}/100")

if eval_count > 5:
    print(f"   ... and {eval_count - 5} more evaluations")

print()

# ============================================================================
# 6. Create Attendance Records
# ============================================================================
print("6. Creating Attendance Records...")

session_types = ['Lecture', 'Lab', 'Tutorial', 'Project']
attendance_count = 0

for student in students[:10]:
    # Create 10 attendance records per student
    for day_offset in range(10):
        date = timezone.now().date() - timedelta(days=day_offset)
        session = random.choice(session_types)
        
        # Status based on student's attendance percentage
        rand_val = random.uniform(0, 100)
        if rand_val < student.attendance_percentage:
            status = 'Present'
        elif rand_val < student.attendance_percentage + 5:
            status = 'Late'
        else:
            status = 'Absent'
        
        record, created = AttendanceRecord.objects.get_or_create(
            student=student,
            date=date,
            session_type=session,
            defaults={'status': status}
        )
        
        if created:
            attendance_count += 1

print(f"   ✓ Created {attendance_count} attendance records")
print()

# ============================================================================
# 7. Create Assignments
# ============================================================================
print("7. Creating Assignments...")

for project in projects[:2]:
    for i in range(3):
        assignment_id = f'ASSGN{project.project_id[-3:]}0{i+1}'
        
        assignment, created = Assignment.objects.get_or_create(
            assignment_id=assignment_id,
            defaults={
                'title': f'Assignment {i+1} - {project.title}',
                'description': f'Complete part {i+1} of the project',
                'project': project,
                'due_date': timezone.now() + timedelta(days=random.randint(5, 30)),
                'max_marks': 100.0
            }
        )
        
        if created:
            print(f"   ✓ Created: {assignment.assignment_id}")

print()

# ============================================================================
# Summary
# ============================================================================
print("=" * 70)
print("Sample Data Loading Complete!")
print("=" * 70)
print()
print(f"✓ Rubric Categories: {RubricCategory.objects.count()}")
print(f"✓ Projects: {Project.objects.count()}")
print(f"✓ Students: {Student.objects.count()}")
print(f"✓ Teams: {Team.objects.count()}")
print(f"✓ Evaluations: {Evaluation.objects.count()}")
print(f"✓ Attendance Records: {AttendanceRecord.objects.count()}")
print(f"✓ Assignments: {Assignment.objects.count()}")
print()

# Risk Status Summary
safe_count = Student.objects.filter(risk_status='Safe').count()
at_risk_count = Student.objects.filter(risk_status='At Risk').count()
unknown_count = Student.objects.filter(risk_status='Unknown').count()

print("Risk Status Distribution:")
print(f"   Safe: {safe_count}")
print(f"   At Risk: {at_risk_count}")
print(f"   Unknown: {unknown_count}")
print()

print("🎉 You can now:")
print("   1. Access dashboards at http://localhost:8000/")
print("   2. Login to admin at http://localhost:8000/admin/")
print("   3. View students at http://localhost:8000/students/")
print("   4. Test API endpoints")
print()
print("Sample login credentials (if created):")
print("   Username: admin")
print("   Password: (what you set during createsuperuser)")
print()
print("=" * 70)