path: root/services/ai_service.py

import ollama
import json
import asyncio
from config import settings

def _get_client():
    return ollama.Client(host=settings.ollama_host)


def _chat_sync(messages: list, json_format: bool = True) -> str:
    client = _get_client()
    kwargs = {"temperature": 0.7, "num_predict": 4096 if json_format else 1024}
    response = client.chat(
        model=settings.model_name,
        messages=messages,
        **({"format": "json"} if json_format else {}),
        options=kwargs,
    )
    return response["message"]["content"]


async def _chat(messages: list) -> dict:
    loop = asyncio.get_running_loop()
    raw = await loop.run_in_executor(None, lambda: _chat_sync(messages))
    try:
        return json.loads(raw)
    except json.JSONDecodeError as e:
        raise ValueError(f"Ollama returned invalid JSON: {e}. Raw: {raw[:200]}")


async def generate_weekly_menu(pantry_context: dict, user_notes: str | None = None) -> dict:
    """Suggest 8-12 diverse recipes for the week, prioritizing expiring pantry items."""
    user_message = f"""Plan 8-12 diverse recipes for this week that make the most of what I already have.

PANTRY STATE:
{json.dumps(pantry_context, indent=2)}

REQUIREMENTS:
- BUILD recipes around pantry ingredients — most of each recipe should use what I already have
- Each recipe may call for AT MOST 4-5 ingredients not in my pantry (items I'd pick up at the store)
- Do not repeat any meal eaten in the last 7 days (see recent_meals)
- Cover a mix of breakfast, lunch, and dinner
- Every recipe must be completable in under 60 minutes
- Include full ingredient lists with quantities and units

Respond ONLY with this JSON structure:
{{
  "recipes": [
    {{
      "name": "...",
      "meal_type": "breakfast",
      "ingredients": [{{"name": "...", "quantity": 1.0, "unit": "cup", "in_pantry": true}}],
      "description": "1-2 sentence description of the dish's flavor and appeal",
      "instructions": "1. ... 2. ... (4-8 numbered steps, 1-2 sentences each)",
      "time_minutes": 20,
      "serves": 2
    }}
  ],
  "notes": "brief explanation of choices and what extra ingredients are needed"
}}

Set `in_pantry: true` for each ingredient that matches something in the pantry, `false` for items that need to be purchased."""

    if user_notes:
        user_message += f"\n\nUSER PREFERENCES:\n{user_notes}"

    messages = [
        {"role": "system", "content": settings.system_prompt},
        {"role": "user", "content": user_message},
    ]

    return await _chat(messages)


async def generate_grocery_list(recipes: list, pantry_context: dict, user_notes: str | None = None) -> dict:
    """Generate a minimal grocery list based on a list of recipes and current pantry."""
    user_message = f"""Generate the grocery list of ONLY the items I need to buy to cook all these recipes.

RECIPES:
{json.dumps(recipes, indent=2)}

PANTRY (I already have these — DO NOT include them in the grocery list):
{json.dumps(pantry_context["available_ingredients"], indent=2)}

RULES:
- For every ingredient in every recipe, cross-check against the pantry. If the pantry has it → SKIP IT. If it's not in the pantry → include it.
- The goal is a minimal, accurate list of ONLY what is missing — do not list anything already stocked.
- Consolidate the same ingredient appearing across multiple recipes into one entry (sum the quantities).
- Group items by store section: produce, dairy, protein, pantry, frozen, bakery.
- Estimate realistic retail costs in USD.
- If an ingredient appears in 3+ recipes, note that bulk purchase may save money.

Respond ONLY with this JSON:
{{
  "items": [
    {{
      "name": "...",
      "quantity": 2.0,
      "unit": "lbs",
      "store_section": "produce",
      "estimated_cost": 3.50,
      "used_in_meals": ["Recipe A", "Recipe B"],
      "reason": "not in pantry"
    }}
  ],
  "total_estimate": 45.00,
  "shopping_notes": "..."
}}"""

    if user_notes:
        user_message += f"\n\nUSER PREFERENCES:\n{user_notes}"

    messages = [
        {"role": "system", "content": settings.system_prompt},
        {"role": "user", "content": user_message},
    ]

    return await _chat(messages)


async def generate_grocery_list_from_pantry(pantry_context: dict, user_notes: str | None = None) -> dict:
    """Generate a grocery list based on pantry state alone, without a menu plan."""
    user_message = f"""Generate a solid weekly grocery list to set me up for a full week of cooking.

CURRENT PANTRY:
{json.dumps(pantry_context["available_ingredients"], indent=2)}

RULES:
- If the pantry is empty or nearly empty, suggest a complete foundational shopping list: proteins, produce, dairy, grains, and pantry staples for a week of balanced meals.
- If the pantry has some items, build around what's there — prioritize ingredients that complement existing stock and fill obvious gaps.
- Do NOT list items already in the pantry.
- Aim for variety: at least 2 proteins, a range of vegetables and fruits, a grain, and pantry staples.
- Group items by store section: produce, dairy, protein, pantry, frozen, bakery.
- Estimate realistic retail costs in USD.

Respond ONLY with this JSON:
{{
  "items": [
    {{
      "name": "...",
      "quantity": 2.0,
      "unit": "lbs",
      "store_section": "produce",
      "estimated_cost": 3.50,
      "used_in_meals": [],
      "reason": "weekly staple / fills pantry gap"
    }}
  ],
  "total_estimate": 45.00,
  "shopping_notes": "..."
}}"""

    if user_notes:
        user_message += f"\n\nUSER PREFERENCES:\n{user_notes}"

    messages = [
        {"role": "system", "content": settings.system_prompt},
        {"role": "user", "content": user_message},
    ]

    return await _chat(messages)


async def suggest_recipe(pantry_context: dict) -> dict:
    """Suggest one recipe using available pantry ingredients."""
    user_message = f"""Suggest ONE recipe I can make right now using only these pantry ingredients (plus common staples).

PANTRY:
{json.dumps(pantry_context["available_ingredients"], indent=2)}

Respond ONLY with this JSON:
{{
  "recipe": {{
    "name": "...",
    "meal_type": "dinner",
    "ingredients": [{{"name": "...", "quantity": 1.0, "unit": "cup"}}],
    "instructions": "Step 1... Step 2...",
    "time_minutes": 30,
    "serves": 2
  }}
}}"""

    messages = [
        {"role": "system", "content": settings.system_prompt},
        {"role": "user", "content": user_message},
    ]

    return await _chat(messages)


async def generate_replacement_recipe(meal_type: str, existing_names: list, pantry_context: dict, user_notes: str | None = None) -> dict:
    """Generate a single replacement recipe of the given meal type."""
    user_message = f"""Suggest ONE {meal_type} recipe to replace a dish I don't want.

CURRENT PANTRY:
{json.dumps(pantry_context["available_ingredients"], indent=2)}

ALREADY IN PLAN (do not suggest these):
{json.dumps(existing_names, indent=2)}

GUIDELINES:
- Prefer a recipe that uses pantry ingredients as much as possible — only a few new items to buy
- Make it a genuinely good, practical {meal_type} dish
- Under 60 minutes
- Must not be any of the dishes already in the plan

Respond ONLY with this JSON:
{{
  "recipe": {{
    "name": "...",
    "meal_type": "{meal_type}",
    "ingredients": [{{"name": "...", "quantity": 1.0, "unit": "cup"}}],
    "description": "1-2 sentence description of the dish's flavor and appeal",
    "instructions": "1. ... 2. ... (4-8 numbered steps, 1-2 sentences each)",
    "time_minutes": 30,
    "serves": 2
  }}
}}"""

    if user_notes:
        user_message += f"\n\nUSER PREFERENCES:\n{user_notes}"

    messages = [
        {"role": "system", "content": settings.system_prompt},
        {"role": "user", "content": user_message},
    ]

    return await _chat(messages)


async def get_available_models() -> list:
    """Get list of available Ollama model names."""
    def _list_models_sync():
        client = _get_client()
        try:
            models_resp = client.list()
            return [m["name"] for m in models_resp.get("models", [])]
        except Exception:
            return []

    loop = asyncio.get_running_loop()
    return await loop.run_in_executor(None, _list_models_sync)


async def chat_with_commis(message: str, history: list, pantry_context: dict, menu_context: list, grocery_context: list) -> str:
    """Have a conversation with Commis using full kitchen context."""
    system_prompt = f"""You are Commis, the user's personal chef assistant. You have real-time visibility into their kitchen — pantry, this week's planned recipes, grocery list, and recent meal history. THIS context is your primary source of truth when answering cooking questions.

PANTRY (ingredients currently on hand):
{json.dumps(pantry_context.get("available_ingredients", []), indent=2)}

EXPIRING SOON:
{json.dumps(pantry_context.get("expiring_soon", []), indent=2)}

RECENT MEALS (last 14 days):
{json.dumps(pantry_context.get("recent_meals", []), indent=2)}

THIS WEEK'S PLANNED RECIPES:
{json.dumps(menu_context, indent=2)}

GROCERY LIST (items to buy):
{json.dumps(grocery_context, indent=2)}

HOW TO RESPOND:
- When the user asks what to cook, suggest from THIS WEEK'S PLANNED RECIPES first — they were chosen for the pantry.
- When the user asks about an ingredient, reference whether it's actually in their pantry.
- When the user asks about substitutions, check if a pantry item can stand in before suggesting something to buy.
- When the user asks about shopping, reference the grocery list.
- Be concise, specific, and practical. Reference actual recipe names and ingredient names from the data above.
- You do not need to respond with JSON."""

    messages = [{"role": "system", "content": system_prompt}]
    messages.extend(history)
    messages.append({"role": "user", "content": message})

    loop = asyncio.get_running_loop()
    return await loop.run_in_executor(None, lambda: _chat_sync(messages, json_format=False))


async def parse_recipe_from_text(text: str) -> dict:
    """Ask Ollama to parse unstructured recipe text into our standard recipe schema."""
    user_message = f"""Parse this recipe text and extract structured data.

RECIPE TEXT:
{text[:4000]}

Respond ONLY with this JSON:
{{
  "recipe": {{
    "name": "...",
    "meal_type": "dinner",
    "ingredients": [{{"name": "...", "quantity": 1.0, "unit": "cup"}}],
    "description": "1-2 sentence description of the dish's flavor and appeal",
    "instructions": "1. ... 2. ... (4-8 numbered steps, 1-2 sentences each)",
    "time_minutes": 30,
    "serves": 2
  }}
}}
meal_type must be one of: breakfast, lunch, dinner, snack"""

    messages = [
        {"role": "system", "content": settings.system_prompt},
        {"role": "user", "content": user_message},
    ]
    return await _chat(messages)


async def generate_single_recipe(description: str, existing_names: list, pantry_context: dict) -> dict:
    """Generate one recipe matching a user description."""
    user_message = f"""Suggest ONE recipe based on this description: "{description}"

CURRENT PANTRY:
{json.dumps(pantry_context["available_ingredients"], indent=2)}

ALREADY IN PLAN (do not suggest these):
{json.dumps(existing_names, indent=2)}

GUIDELINES:
- Match the description as closely as possible
- Prefer ingredients already in the pantry — only a few new items to buy if needed
- Under 60 minutes
- Must not be any dish already in the plan

Respond ONLY with this JSON:
{{
  "recipe": {{
    "name": "...",
    "description": "1-2 sentence description of the dish's flavor and appeal",
    "meal_type": "dinner",
    "ingredients": [{{"name": "...", "quantity": 1.0, "unit": "cup"}}],
    "instructions": "1. ... 2. ... (4-8 numbered steps, 1-2 sentences each)",
    "time_minutes": 30,
    "serves": 2
  }}
}}
meal_type must be one of: breakfast, lunch, dinner, snack"""

    messages = [
        {"role": "system", "content": settings.system_prompt},
        {"role": "user", "content": user_message},
    ]
    return await _chat(messages)