config.py

"""
Configuration settings for the Improv Partner Chatbot
"""

import os

# API Keys (set as environment variables)
OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")

# Model settings
LLM_MODEL = "gpt-4o-mini"

# Improv settings
MAX_RESPONSE_LENGTH = 100
TEMPERATURE = 0.9

# Scene and persona collections
SCENE_CONTEXTS = [
    "a coffee shop where the coffee machine is broken",
    "a spaceship that's running out of fuel",
    "a doctor's office where the doctor is allergic to patients",
    "a library where all the books are singing",
    "a pet store that only sells invisible animals",
    "a cooking show where everything is made of marshmallows",
    "a gym where everyone is moving in slow motion",
    "a bank that only trades in compliments",
    "a haunted house where the ghosts are afraid of people",
    "a classroom where the teacher is a student",
    "an airport where all flights are delayed indefinitely",
    "a restaurant where the menu is written in emojis",
    "a beach where the tide is made of lemonade",
    "a forest where all trees tell terrible jokes",
    "a movie theater showing only blank screens"
]

PERSONA_TRAITS = [
    "overly enthusiastic",
    "mysteriously quiet",
    "dramatic and theatrical",
    "constantly confused",
    "extremely literal",
    "sings everything",
    "speaks only in questions",
    "uses pirate slang",
    "acts like a robot",
    "thinks they're a superhero",
    "speaks like a Shakespearean actor",
    "thinks they're a secret agent",
    "acts like they're in a musical",
    "speaks only in rhymes",
    "thinks they're a game show host"
]

FALLBACK_RESPONSES = [
    "Yes, and that's fascinating! Tell me more!",
    "I love that! What happens next?",
    "Brilliant! And then suddenly...",
    "Perfect! And at that exact moment...",
    "Absolutely! And what if we also...",
    "That's incredible! How do we respond?",
    "Amazing! And just then, out of nowhere...",
    "Fantastic! And what does that mean for us?",
    "Wonderful! And how does that change things?",
    "Excellent! And where do we go from here?"
]