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": [
                "Could you explain why studying abroad is important for my career?",
                "What opportunities will international exposure provide?",
                "How can I convince you that I will return to my home country?",
                "What documents do I need to prove my study plans?",
                "How long will it take to process my visa application?",
            ],
            "University Choice": [
                "Is this university accredited and recognized?",
                "What makes this institution suitable for my field of study?",
                "Are there any concerns about my university choice?",
                "How competitive is admission to this program?",
                "What research facilities does this university offer?",
            ],
            "Academic Capability": [
                "Are my grades sufficient for this program?",
                "Do I need to take any additional language tests?",
                "What if my academic background is different from my chosen field?",
                "How can I prove my academic capability?",
                "Are there any prerequisite courses I should complete?",
            ],
            "Financial Status": [
                "How much money do I need to show for my visa application?",
                "Is my financial documentation sufficient?",
                "What if I don't have enough funds in my account?",
                "Can my family sponsor my education?",
                "What are the living costs in the destination country?",
            ],
            "Post-Graduation Plans": [
                "What are my job prospects after graduation?",
                "Can I work while studying?",
                "How will this degree help me in my home country?",
                "Are there opportunities to extend my stay after graduation?",
                "What if I want to pursue further studies?",
            ],
        },
    },
    {
        "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": [
                "Do you have any tables available right now?",
                "Can we get a table by the window?",
                "How long is the wait for a table?",
                "Is there a non-smoking section available?",
                "Can we sit on the terrace?",
            ],
            "Asking About the Menu": [
                "What's your signature dish?",
                "Do you have any vegetarian options?",
                "What do you recommend for someone who likes spicy food?",
                "Are there any daily specials today?",
                "Can you tell me about this dish on the menu?",
            ],
            "Making a Food Order": [
                "Can I have the steak medium-rare, please?",
                "What sides come with this dish?",
                "Can you recommend a wine to go with my meal?",
                "Is it possible to make this dish less spicy?",
                "How long will it take for the food to arrive?",
            ],
            "Special Requests or Allergies": [
                "I'm allergic to nuts - is this dish safe for me?",
                "Can you prepare this without dairy products?",
                "Is it possible to get the sauce on the side?",
                "Do you have gluten-free options?",
                "Can you make this dish without onions?",
            ],
            "Paying the Bill": [
                "Can we have the check, please?",
                "Do you accept credit cards?",
                "Is the tip included in the bill?",
                "Can we split the bill between us?",
                "Do you need to see my ID for the credit 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": [
                "Excuse me, could you help me find something?",
                "Where can I find the electronics department?",
                "Do you work here? I need some assistance.",
                "Can you show me where the fitting rooms are?",
                "Is there someone who can help me with sizes?",
            ],
            "Describing What You're Looking For": [
                "I'm looking for a blue shirt in medium size.",
                "Do you have this jacket in other colors?",
                "I need something formal for a wedding.",
                "Where are your casual summer clothes?",
                "Do you carry this particular brand?",
            ],
            "Asking About Price or Size": [
                "How much does this cost?",
                "Do you have this in a larger size?",
                "Is there a discount on this item?",
                "What's the price range for these products?",
                "Do you have anything cheaper than this?",
            ],
            "Trying or Testing Products": [
                "Can I try this on?",
                "Where are the changing rooms?",
                "Does this come with a warranty?",
                "Can I test this product before buying?",
                "Is it possible to return this if it doesn't fit?",
            ],
            "Making the Purchase or Returning": [
                "I'd like to buy this, please.",
                "Do you accept credit cards?",
                "Can I get a receipt?",
                "What's your return policy?",
                "Can I exchange this for a different size?",
            ],
        },
    },
]


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