ai_service.py

import base64
import logging
from openai import AsyncOpenAI
from config import KIMI_CODE_API_KEY, KIMI_CODE_BASE_URL, KIMI_CODE_MODEL, DEEPSEEK_API_KEY
from prompts import build_system_prompt

client = AsyncOpenAI(
    api_key=KIMI_CODE_API_KEY,
    base_url=KIMI_CODE_BASE_URL,
    timeout=60.0,
    max_retries=1
)

ds_client = AsyncOpenAI(
    api_key=DEEPSEEK_API_KEY,
    base_url="https://api.deepseek.com",
    timeout=60.0,
    max_retries=1
)

async def generate_response(user_text: str) -> str:
    # Primary: DeepSeek Reasoner (stable, fast, thinking mode)
    try:
        response = await ds_client.chat.completions.create(
            model="deepseek-reasoner",
            messages=[
                {"role": "system", "content": build_system_prompt()},
                {"role": "user", "content": user_text}
            ]
        )
        return response.choices[0].message.content.strip()
    except Exception as e:
        logging.warning(f"DeepSeek API Failed: {e}. Falling back to Kimi K2.5...")

    # Fallback: Kimi K2.5
    try:
        response = await client.chat.completions.create(
            model=KIMI_CODE_MODEL,
            messages=[
                {"role": "system", "content": build_system_prompt()},
                {"role": "user", "content": user_text}
            ]
        )
        return response.choices[0].message.content.strip()
    except Exception as e:
        logging.error(f"Kimi Fallback also failed: {e}")
        return "Извините, у меня небольшие неполадки со связью (оба сервера перегружены). Нажмите кнопку оплаты или попробуйте позже."

async def analyze_slip(image_bytes: bytes) -> bool:
    try:
        base64_image = base64.b64encode(image_bytes).decode('utf-8')
        
        response = await client.chat.completions.create(
            model=KIMI_CODE_MODEL,
            messages=[
                {
                    "role": "user",
                    "content": [
                        {
                            "type": "text", 
                            "text": "Analyze this Thai bank slip. Does it show a successful transfer of exactly 3000 THB? Answer strictly YES or NO."
                        },
                        {
                            "type": "image_url",
                            "image_url": {
                                "url": f"data:image/jpeg;base64,{base64_image}"
                            }
                        }
                    ]
                }
            ]
        )
        answer = response.choices[0].message.content.strip().upper()
        return "YES" in answer
    except Exception as e:
        error_msg = str(e) or repr(e)
        logging.error(f"Kimi Vision API Error: {error_msg}")
        return False