import os
import django

os.environ.setdefault('DJANGO_SETTINGS_MODULE', 'core.settings')
django.setup()

from system.models import Announcement, AiModel
from reservations.models import ReservationResource, ResourceType
from users.models import User

def seed_data():
    # Delete old
    Announcement.objects.all().delete()
    AiModel.objects.all().delete()
    ReservationResource.objects.all().delete()

    admin_user = User.objects.filter(is_superuser=True).first()
    
    if not admin_user:
        print("No admin user found. Cannot seed announcements.")
        return

    # Seed Announcements
    Announcement.objects.create(
        title='CorpScale 2.0 品牌升级：助力一人公司跨越式增长',
        content='全新的 OPC 平台正式上线！为自由职业者提供更强大的背书与任务撮合能力。',
        publisher=admin_user,
        is_published=True
    )
    Announcement.objects.create(
        title='关于新增 4D 点云标注、多模态意图识别等高端任务类型的通知',
        content='针对自动驾驶和前沿 AI 研发，平台已引入高价值数据标注任务，欢迎拥有相关资质的专家承接。',
        publisher=admin_user,
        is_published=True
    )
    Announcement.objects.create(
        title='开发者生态支持计划：首批 OPC 认证专家将获得算力补贴',
        content='首批入驻的专家除了获得免排队特权外，还将每月获得 500,000 Tokens 的模型调用配额。',
        publisher=admin_user,
        is_published=True
    )

    # Seed AI Models
    AiModel.objects.create(
        name='CS-Llama-Instruct-7B',
        provider='CorpScale',
        description='基于Llama架构深度优化的行业指令微调模型，适用于客服机器人与知识库问答系统，支持超长上下文。',
        price_per_token=0.0001
    )
    AiModel.objects.create(
        name='VisionX-Segmentation-V3',
        provider='CorpScale',
        description='高精度图像语义分割模型，支持256个语义类别，专为自动驾驶避障和医疗辅助诊断分析设计。',
        price_per_token=0.0005
    )
    AiModel.objects.create(
        name='AutoTrans-Speech-Turbo',
        provider='CorpScale',
        description='低延迟实时语音转文字模型，支持极速多国语言翻译及50多种中国方言精准识别。',
        price_per_token=0.0002
    )

    # Seed Reservation Resources
    ReservationResource.objects.create(
        name='A栋101 大型全景会议室',
        type=ResourceType.MEETING_ROOM,
        description='支持50人规模的顶级会议，配备8K投影与哈曼卡顿音响系统。',
        capacity=50,
        location='创新园区 A栋 1层',
        price_per_unit=150.00,
        price_unit='小时'
    )
    ReservationResource.objects.create(
        name='园区通用智能门禁白名单',
        type=ResourceType.ACCESS_CONTROL,
        description='开通后刷脸即可进入园区各公共办公区及休息区。',
        capacity=1,
        location='全园区通用',
        price_per_unit=20.00,
        price_unit='次'
    )

    print("Phase 5 Seed data generated successfully!")

if __name__ == '__main__':
    seed_data()