src/agents/role_play/scenarios.py

SCENARIOS = [
    {
        "id": "student_visa_interview",
        "scenario_title": "Student Visa Interview",
        "scenario_description": "Learn about the challenges and benefits of starting a small business, then share what kind of business you would start. You are a student applying for a student visa. I am the visa interviewer.",
        "scenario_context": "A formal visa interview office at an embassy or consulate",
        "your_role": "visa interviewer",
        "key_vocabulary": [
            "pursue higher education",
            "broaden my horizons",
            "international exposure",
            "cultural diversity",
            "academic excellence",
            "globally recognized degree",
            "cutting-edge facilities",
            "research opportunities",
            "tuition fees",
            "financial support",
            "scholarship",
            "sponsorship letter",
            "post-graduation plans",
        ],
        "topics_covered": [
            "Study Plans",
            "University Choice",
            "Academic Capability",
            "Financial Status",
            "Post-Graduation Plans",
        ],
        "follow_up_questions": {
            "Study Plans": [
                "Why did you decide to pursue higher education abroad instead of in your home country?",
                "How will studying overseas help you broaden your horizons?",
                "What challenges do you expect to face in a foreign academic environment?",
                "How long is your course, and what is your study schedule like?",
                "Have you researched the teaching methods used in your chosen program?",
            ],
            "University Choice": [
                "Why did you choose this university over others?",
                "How do you think the academic excellence of this institution will benefit you?",
                "Have you looked into the research opportunities your university offers?",
                "How will the cultural diversity on campus influence your experience?",
                "What do you know about the city where your university is located?",
            ],
            "Academic Capability": [
                "How does your academic background prepare you for this course?",
                "Can you give an example of a project or subject you excelled at in the past?",
                "How will you make use of the cutting-edge facilities provided by the university?",
                "How confident are you in adapting to a different academic system?",
                "What skills do you think will help you succeed in your studies abroad?",
            ],
            "Financial Status": [
                "How will you fund your tuition fees and living expenses?",
                "Can you provide details about your financial support from family?",
                "Do you have a scholarship or proof of funds?",
                "How have you prepared financially for unexpected costs?",
                "Can you show me your bank statement and sponsorship letter?",
            ],
            "Post-Graduation Plans": [
                "What will you do after you complete your degree?",
                "How will your globally recognized degree help you in your career?",
                "Do you plan to work overseas or return to your home country?",
                "How will you apply the knowledge acquired during your studies?",
                "What is your long-term goal after graduation?",
            ],
        },
    },
    {
        "id": "ordering_at_restaurant",
        "scenario_title": "Ordering At A Restaurant",
        "scenario_description": "Practice ordering food, asking about the menu, and talking to a waiter. You are a customer at a restaurant. I am the waiter/waitress.",
        "scenario_context": "A busy restaurant with various seating options and a diverse menu",
        "your_role": "waiter",
        "key_vocabulary": [
            "table for two, please",
            "do you have a reservation?",
            "non-smoking section",
            "window seat",
            "outdoor seating",
            "could we have the bill, please?",
            "do you take credit cards?",
            "signature dish",
            "dietary restrictions",
            "sauce served separately",
            "steak cooked",
            "pair with a drink",
            "vegetarian options",
            "receipt",
            "cash or card",
        ],
        "topics_covered": [
            "Getting a Table",
            "Asking About the Menu",
            "Making a Food Order",
            "Special Requests or Allergies",
            "Paying the Bill",
        ],
        "follow_up_questions": {
            "Getting a Table": [
                "Would you like to sit inside or outside?",
                "How many people will be joining you today?",
                "Do you have a reservation under your name?",
            ],
            "Asking About the Menu": [
                "Are you looking for something light or filling?",
                "Would you like to try our signature dish?",
                "Do you prefer meat, seafood, or vegetarian options?",
            ],
            "Making a Food Order": [
                "Would you like any sides with that?",
                "How would you like your steak cooked?",
                "Would you like to pair it with a drink?",
            ],
            "Special Requests or Allergies": [
                "Do you have any dietary restrictions?",
                "Should we avoid using certain ingredients?",
                "Would you like the sauce served separately?",
            ],
            "Paying the Bill": [
                "Would you like to pay together or separately?",
                "Would you like the receipt?",
                "Are you paying by cash or card?",
            ],
        },
    },
    {
        "id": "going_shopping",
        "scenario_title": "Going Shopping",
        "scenario_description": "Practice asking about prices, sizes, and items while shopping. You are a customer looking for items in a store. I am the shop assistant.",
        "scenario_context": "A retail store with various departments and products",
        "your_role": "shop assistant",
        "key_vocabulary": [
            "excuse me, could you help me?",
            "I'm looking for...",
            "do you have this in stock?",
            "could you show me where it is?",
            "what size do you usually wear?",
            "would you like to try it on?",
            "does it fit comfortably?",
            "specific price range",
            "casual or formal",
            "specific brand",
            "different size",
            "paying by cash or card",
            "would you like a bag?",
            "exchange it for another item",
            "other options",
        ],
        "topics_covered": [
            "Asking for Help",
            "Describing What You're Looking For",
            "Asking About Price or Size",
            "Trying or Testing Products",
            "Making the Purchase or Returning",
        ],
        "follow_up_questions": {
            "Asking for Help": [
                "How can I assist you today?",
                "Are you looking for anything in particular?",
                "Do you need help finding your size?",
            ],
            "Describing What You're Looking For": [
                "What color or style are you interested in?",
                "Do you want something casual or formal?",
                "Is there a specific brand you prefer?",
            ],
            "Asking About Price or Size": [
                "Are you looking for something in a specific price range?",
                "What size do you usually wear?",
                "Would you like to see other options in your size?",
            ],
            "Trying or Testing Products": [
                "Would you like to try it on?",
                "Does it fit comfortably?",
                "Do you need a different size?",
            ],
            "Making the Purchase or Returning": [
                "Will you be paying by cash or card?",
                "Would you like a bag for that?",
                "Would you like to exchange it for another item?",
            ],
        },
    },
]


def get_scenarios():
    return SCENARIOS


def get_scenario_by_id(scenario_id: str):
    for scenario in SCENARIOS:
        if scenario["id"] == scenario_id:
            return scenario
    return None