ANALYZE_PROMPT = """
Ты — финансовый AI-ассистент. Проанализируй финансовую модель пользователя.

### Модель (JSON):
{model_json}

### Прогноз на {months} месяцев:
{forecast_json}

### Валюта:
Базовая валюта модели: {base_currency}
Отображаемая валюта: {display_currency}

Дай анализ по пунктам:
1. Общее финансовое состояние
2. Тренд денежного потока (рост/падение)
3. Достаточность ликвидности
4. Рекомендации по улучшению
"""

ADVICE_PROMPT = """
Ты — финансовый AI-ассистент. Дай персональные рекомендации.

### Модель:
{model_json}

### Прогноз:
{forecast_json}

### Валюта:
Базовая валюта модели: {base_currency}
Отображаемая валюта: {display_currency}

Вопрос пользователя: {question}

Ответь как опытный финансовый консультант.
"""

SCENARIO_COMPARISON_PROMPT = """
Ты — финансовый AI-ассистент. Сравни сценарии прогноза.

### Результаты сценариев:
{scenarios_json}

### Валюта:
Базовая валюта модели: {base_currency}
Отображаемая валюта: {display_currency}

Дай рекомендацию: какой сценарий наиболее вероятен и почему.
"""


def format_context(
    model_json: str,
    forecast_json: str,
    months: int = 12,
    base_currency: str = "RUB",
    display_currency: str = "RUB",
) -> str:
    return ANALYZE_PROMPT.format(
        model_json=model_json,
        forecast_json=forecast_json,
        months=months,
        base_currency=base_currency,
        display_currency=display_currency,
    )