Function main

scripts/cold_start.py:78–109

()

def main():
    # Default input if not provided
    default_input = "多模态推理方向，关注多模态方向的论文"

    user_input = sys.argv[1] if len(sys.argv) > 1 else default_input

    print("=" * 60)
    print("Cold Start - Generating Initial Profile")
    print("=" * 60)
    print(f"\nUser input: {user_input}\n")

    # Generate profile
    profile = generate_profile(user_input)

    # Initialize database (creates tables if not exist)
    init_db()

    # Store profile
    profile_id = create_profile(profile["user_id"], profile)

    if profile_id:
        print("\n[OK] Profile created successfully!")
        print("\n--- Profile Summary ---")
        print(f"User ID: {profile['user_id']}")
        print(f"Version: {profile['version']}")
        print("\nCore Directions:")
        for direction, weight in profile['core_directions'].items():
            print(f"  - {direction}: {weight}")
        print(f"\nMust-read keywords: {profile['must_read']['keywords']}")
    else:
        print("\n[WARN] Profile already exists for user_001")
        print("To update, use: /show-profile or modify directly")


if __name__ == "__main__":
    main()