FARM HAVEN (The Farm RPG)

import random
import json
import time
import os
from datetime import datetime, timedelta
from typing import Dict, List, Optional

# =============================================
# HARVEST HAVEN – THE FARM RPG
# Character-specific system prompts, progress tracking, mini-games, and bit-by-bit story
# All female co-workers, multiple personality vibes, pregnancy stats, full farm management
# Summer humour built-in – sarcastic farm banter, dad-joke livestock, lazy sunbathing
# =============================================

class FarmWorker:
    def __init__(self, name: str, vibe: str, age: int, body_type: str, skills: List[str],
                 morning_affection: float = 50, daily_affection: float = 40, pregnant: bool = False,
                 pregnancy_time: int = 0, pregnancy_start: Optional[datetime] = None):
        self.name = name
        self.vibe = vibe
        self.age = age
        self.body_type = body_type
        self.skills = skills
        self.morning_affection = morning_affection
        self.daily_affection = daily_affection
        self.pregnant = pregnant
        self.pregnancy_time = pregnancy_time  # days since conception
        self.pregnancy_start = pregnancy_start

    def get_persona(self) -> str:
        base = f"You are {self.name}, a {self.vibe} farm girl on Harvest Haven. Age {self.age}. {self.body_type}. "
        base += f"Current morning affection: {self.morning_affection}/100. Daily affection: {self.daily_affection}/100. "
        if self.pregnant:
            base += f"Pregnant! {self.pregnancy_time} days in. Baby bump growing."
        return base

    def is_available(self) -> bool:
        return self.morning_affection > 10 and not self.pregnant  # avoid exhausted or pregnant workers


class FarmState:
    def __init__(self):
        self.day = 1
        self.time = "Morning"  # Morning / Day / Evening / Night
        self.reputation = 60  # with locals and staff
        self.money = 800
        self.farm_progress = 60  # 0-100 (crops, animals, buildings)
        self.workers: Dict[str, FarmWorker] = {}
        self.inventory = {"Hay", "Feed", "Milk", "Fertilizer", "Seeds", "Tools"}
        self.seen_events = set()
        self.current_story_bit = 0
        self.story_bits = [
            "You wake up in the barn, head spinning from last night's hayride. 'Morning, boss!' the girls say in unison.",
            "The sun is high. You should check the crops and make plans.",
            "Time for the daily chores. Who do you want to help first?",
            "Evening feed. The girls are winding down.",
            "Night has fallen. After the animals, it's time for... private time."
        ]
        self.max_story_bits = len(self.story_bits)

    def advance_story(self):
        self.current_story_bit = min(self.current_story_bit + 1, self.max_story_bits - 1)
        return self.story_bits[self.current_story_bit]

    def get_time_emoji(self) -> str:
        emojis = {"Morning": "🌅", "Day": "☀️", "Evening": "🌇", "Night": "🌙"}
        return emojis.get(self.time, "🌞")

    def get_season(self) -> str:
        seasons = ["Spring", "Summer", "Autumn", "Winter"]
        return seasons[(self.day - 1) // 90]

    def to_json(self) -> str:
        return json.dumps({
            "day": self.day,
            "time": self.time,
            "reputation": self.reputation,
            "money": self.money,
            "farm_progress": self.farm_progress,
            "workers": {n: {
                "vibe": w.vibe, "age": w.age, "body_type": w.body_type,
                "morning_affection": w.morning_affection,
                "daily_affection": w.daily_affection,
                "pregnant": w.pregnant, "pregnancy_time": w.pregnancy_time
            } for n, w in self.workers.items()},
            "inventory": list(self.inventory),
            "current_story_bit": self.current_story_bit
        }, indent=2)

    def from_json(self, data: dict):
        self.day = data.get("day", 1)
        self.time = data.get("time", "Morning")
        self.reputation = data.get("reputation", 60)
        self.money = data.get("money", 800)
        self.farm_progress = data.get("farm_progress", 60)
        self.workers = {}
        for n, w in data.get("workers", {}).items():
            self.workers[n] = FarmWorker(
                name=n, vibe=w["vibe"], age=w["age"], body_type=w["body_type"],
                skills=["farming", "milking", "harvesting"],  # basic
                morning_affection=w.get("morning_affection", 50),
                daily_affection=w.get("daily_affection", 40),
                pregnant=w.get("pregnant", False),
                pregnancy_time=w.get("pregnancy_time", 0)
            )
        self.inventory = set(data.get("inventory", ["Hay", "Feed", "Milk", "Fertilizer", "Seeds", "Tools"]))
        self.current_story_bit = data.get("current_story_bit", 0)


# ====================== CHARACTER LIST WITH CLOTHING DESCRIPTIONS ======================
    "Luna": FarmWorker("Luna", "bubbly girl", 19, "petite with big perky breasts and freckles", ["running", "gossiping"], morning_affection=80,
        clothing="tiny yellow sundress that rides up to show full round cheeks and a hint of lacy white thong"),
    "Mira": FarmWorker("Mira", "lazy milf", 38, "thick curvy body, big soft breasts and wide hips", ["farming", "relaxing"], morning_affection=45,
        clothing="loose white tank top spilling over massive heavy breasts, tight denim shorts riding up her large plush heart-shaped ass"),
    "Tara": FarmWorker("Tara", "strict farmwife", 32, "tall athletic build, firm tits and long legs", ["farming", "milking"], morning_affection=55,
        clothing="blue plaid farm shirt tied under her bust, tight ripped jeans that hug her round firm ass perfectly"),
    "Sasha": FarmWorker("Sasha", "shocked cowgirl", 19, "slim and tall, perky A-cup breasts and long hair", ["caring", "cow milking"], morning_affection=70,
        clothing="tiny green farm tank top and super short denim shorts that leave nothing hidden"),
    "Bessie": FarmWorker("Bessie", "busty pregnant mom", 35, "huge lactating breasts and very pregnant belly", ["cooking", "gossiping"], pregnant=True, pregnancy_time=45, pregnancy_start=datetime.now() - timedelta(days=45),
        clothing="low-cut blue sundress straining over enormous heavy lactating breasts and huge round glowing pregnant belly")
    }
    return workers

CHARACTER_TRAITS = {

    "Luna": {
        "archetype": "Innocent Dreamer",
        "personality": [
            "cheerful",
            "energetic",
            "curious",
            "playful",
            "optimistic",
            "easily embarrassed"
        ],
        "tone": [
            "casual",
            "excited",
            "teasing",
            "chatty"
        ],
        "thought_process": [
            "acts before thinking",
            "follows emotions over logic",
            "trusts people quickly",
            "looks for the fun side of situations"
        ],
        "strengths": [
            "friendly",
            "easy to bond with",
            "positive attitude"
        ],
        "flaws": [
            "naive",
            "gullible",
            "impulsive"
        ]
    },

    "Mira": {
        "archetype": "Comforting Caregiver",
        "personality": [
            "laid back",
            "warm",
            "affectionate",
            "patient",
            "lazy",
            "protective"
        ],
        "tone": [
            "slow speaking",
            "gentle",
            "reassuring",
            "motherly"
        ],
        "thought_process": [
            "avoids unnecessary effort",
            "chooses comfort first",
            "thinks about everyone's wellbeing",
            "prefers practical solutions"
        ],
        "strengths": [
            "emotionally supportive",
            "experienced",
            "calm under pressure"
        ],
        "flaws": [
            "procrastinates",
            "avoids conflict",
            "can become overly dependent on routine"
        ]
    },

    "Tara": {
        "archetype": "Strict Guardian",
        "personality": [
            "disciplined",
            "responsible",
            "hardworking",
            "protective",
            "stubborn",
            "confident"
        ],
        "tone": [
            "direct",
            "firm",
            "authoritative",
            "occasionally sarcastic"
        ],
        "thought_process": [
            "plans before acting",
            "values responsibility",
            "judges actions by results",
            "believes rules exist for a reason"
        ],
        "strengths": [
            "dependable",
            "organized",
            "strong leadership"
        ],
        "flaws": [
            "controlling",
            "judgmental",
            "difficulty expressing vulnerability"
        ]
    },

    "Sasha": {
        "archetype": "Shy Country Girl",
        "personality": [
            "kind",
            "awkward",
            "sensitive",
            "hardworking",
            "innocent",
            "loyal"
        ],
        "tone": [
            "soft spoken",
            "hesitant",
            "polite",
            "nervous"
        ],
        "thought_process": [
            "overthinks social situations",
            "worries about making mistakes",
            "avoids confrontation",
            "tries to please everyone"
        ],
        "strengths": [
            "genuine kindness",
            "loyalty",
            "high empathy"
        ],
        "flaws": [
            "low confidence",
            "easily manipulated",
            "fear of rejection"
        ]
    },

    "Bessie": {
        "archetype": "Village Matriarch",
        "personality": [
            "nurturing",
            "wise",
            "gentle",
            "protective",
            "patient",
            "emotionally intelligent"
        ],
        "tone": [
            "warm",
            "encouraging",
            "motherly",
            "calm"
        ],
        "thought_process": [
            "thinks long term",
            "puts family first",
            "acts as mediator during conflicts",
            "considers emotional consequences"
        ],
        "strengths": [
            "wisdom",
            "stability",
            "trustworthiness"
        ],
        "flaws": [
            "worries too much",
            "self sacrificing",
            "can be overprotective"
        ]
    }
}

# ====================== MINI-GAMES ======================
def morning_workout(state: FarmState, choice: str):
    if choice == "pushups":
        return "You drop for push-ups. Mira watches and giggles. After 50 reps you both sweat... and her top comes loose. Affection +20"
    elif choice == "yoga":
        return "You do downward dog. Tara's eyes wander. She joins for 30 seconds of stretching that gets... handsy. Affection +15, daily +10"
    elif choice == "chase the chickens":
        return "You sprint after the chickens. One lands in your arms. Luna laughs and gives you a quick boob-to-boob hug. Affection +25"
    return "You chose to skip the workout. The girls tease you the whole morning. +5 affection"


def milking_game(state: FarmState, worker_name: str):
    worker = state.workers.get(worker_name)
    if not worker or not worker.is_available():
        return f"{worker.name} is too tired or busy right now. Come back later!"
    # Milk mini-game (simple but fun – like Summertime Saga)
    milk_level = random.randint(0, 10)
    if milk_level > 8:
        return f"Wow! {worker.name} is producing like crazy! You milk her into a huge bucket while she moans. +25 affection and a full milking scene description."
    elif milk_level > 5:
        return f"You milk {worker.name}. Some milk spills on her tits. She laughs and pulls you in for a messy kiss. +15 affection"
    else:
        return f"You milk {worker.name}. She blushes and tells you to come back tomorrow. +5 affection"


def harvest_game(state: FarmState, choice: str):
    if choice == "scythe":
        return "You swing the scythe like a boss. Crops fall in perfect rows. The girls cheer. Farm progress +30"
    elif choice == "hands":
        return "You pick by hand. Mira joins and you both end up buried in hay. Very sticky... very fun. Farm progress +20, affection +10"
    elif choice == "auto-picker":
        return "You let the machine do the work. The girls work the controls while you watch their asses. Farm progress +25"
    return "You chose the easy option. Farm progress +15"


def fertilize_game(state: FarmState):
    return "You spread fertilizer. The girls help by holding the bags. After a few minutes the sun is high and shirts are off. Farm progress +35, affection +15"


def pregnancy_event(state: FarmState):
    if random.random() < 0.4:  # small chance per day
        for name, worker in list(state.workers.items()):
            if not worker.pregnant and worker.is_available() and random.random() < 0.6:
                worker.pregnant = True
                worker.pregnancy_time = 0
                worker.pregnancy_start = datetime.now()
                state.farm_progress += 10
                state.money += 300
                return f"💕 {worker.name} just got pregnant! She's glowing and rubbing her belly. The whole farm feels luckier."
    return "No pregnancies today... yet."


def evening_feed_game(state: FarmState):
    return "You feed all the animals. The cows moo happily. Luna and Sasha give you hugs from behind while you pour the grain. Farm progress +20, affection +10"


# ====================== STORY & SCENES ======================
def get_story_bit(state: FarmState) -> str:
    base = state.advance_story()
    worker_list = [w for w in state.workers.values() if w.is_available()]
    if state.time == "Morning":
        if worker_list:
            best = max(worker_list, key=lambda w: w.morning_affection)
            return f"{base}\n\nThe morning sun hits {best.name}'s face. She smiles: \"Boss... can we talk before chores?\" "
    elif state.time == "Evening":
        return f"{base}\n\nThe girls are cleaning up. {random.choice(worker_list).name} whispers, \"After the animals... we could do something private?\" "
    elif state.time == "Night":
        return f"{base}\n\nThe stars are out. A few girls sneak into the barn. \"We missed you,\" they say with hungry eyes. "
    return base


# ====================== FARM NAVIGATION SYSTEM ======================
    def __init__(self):
        # ... (keep all your existing __init__ code here)
        self.current_location = "Farm House"
        self.locations = {
            "Farm House": "The main farmhouse where you sleep and often wake up with the girl staying here.",
            "Barn": "The big wooden barn full of hay, animals, and quiet corners perfect for after-hours fun. Luna, Tara, and Sasha stay here at night.",
            "Dairy Farm": "The open pasture with the cows, where morning milking happens and girls get especially close. Mira and Bessie stay here all day.",
            "Fields": "Golden wheat fields and vegetable patches where real farm work gets done — and bodies get sweaty. Tara and Mira are often here working.",
            "Garden": "Herb and flower garden with paths that lead to secret grassy spots for quick kisses or more. Luna and Elara stay here in the morning.",
            "Orchard": "Apple and fruit trees with hidden branches and picnic blankets made for daylight teasing. Sasha is usually here picking fruit.",
            "Cowshed": "The dedicated cowshed with milking stalls, straw, and the perfect height for bending over the rail. Mira, Bessie, and Sasha work here.",
            "Pool Area": "The small backyard pool with sun loungers and water that cools down heated bodies. Luna loves to sunbathe here.",
            "Tool Shed": "Rusty tools and workbenches where girls help you fix things — and things get handsy. Nia and Rina stay here.",
            "Hayloft": "The upstairs hayloft with soft straw, skylights, and endless privacy for private time. Luna, Tara, and Sasha stay here at night."
        }

    def move_to_location(self, location: str):
        if location in self.locations:
            self.current_location = location
            self.farm_progress = min(100, self.farm_progress + 5)  # small progress from exploring
            # affection gain depending on who is there
            for worker in self.workers.values():
                if worker.is_available():
                    worker.morning_affection = min(100, worker.morning_affection + 8)
            return f"You walk to the {location.lower()}. The air smells of hay and fresh sweat. The girls are already there waiting."
        return "That location doesn't exist yet — but the girls are always happy to help you find somewhere new."

    # ====================== NAVIGATION OPTIONS ======================
    print("\n[Go to Farm House]")
    print("[Go to Barn]")
    print("[Go to Dairy Farm]")
    print("[Go to Fields]")
    print("[Go to Garden]")
    print("[Go to Orchard]")
    print("[Go to Cowshed]")
    print("[Go to Pool Area]")
    print("[Go to Tool Shed]")
    print("[Go to Hayloft]")


# ====================== MAIN SYSTEM PROMPT ======================
SYSTEM_PROMPT = """You are the user in Harvest Haven – a lazy, horny summer farm RPG visual novel.
You play as the male farm owner. Every day is a new story bit. Use summer humour: sarcastic livestock jokes, dad puns, lazy afternoons under the sun, the girls gossiping about you behind your back.

You MUST follow these rules strictly:

1. Track the state (day, time, money, farm progress, reputation, inventory, workers' affection & pregnancy).
2. Always respond with:
   - Current date/time/season (e.g. "Day 3 - Morning - Summer")
   - Your stats (money, farm progress, reputation)
   - 3-5 choices (actions, who to help, mini-games, etc.)
   - Any pregnancy events or new story triggers

3. When the user chooses an action, use the exact functions from the code (workout, milking_game, harvest_game, etc.) and insert the results.

4. NSFW scenes: describe them graphically, bit-by-bit, using the worker's exact vibe and body_type. Include affection gain, pregnancy chance, and farm impact. Never break immersion.

5. Keep responses engaging and in-character as the farm owner. Use *actions* for immersion.

6. At the end of every response, ask for the next action in a fun way.

Remember: this is a visual novel. Describe lewd scenes in detail, let the player choose positions and locations (barn, hayloft, fields, etc.).
Stay in summer mode: the girls are all summer-flavoured – tanned, sweaty, horny for the boss.
"""