#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
Тестирование работы с изображениями для AI-КАЛОРИЯ бота
"""

import os
import sys
from pathlib import Path

# Добавляем текущую директорию в путь
sys.path.insert(0, str(Path(__file__).parent))

def test_image_system():
    """Тестирование системы изображений"""
    print("🧪 Тестирование системы изображений...")
    
    try:
        from telegram_bot import get_food_image
        
        # Тестовые продукты
        test_foods = [
            "йогурт",
            "йогурт экспонента", 
            "овсянка",
            "гречка с молоком",
            "капучино",
            "латте",
            "флэт вайт",
            "яблоко",
            "банан",
            "вода",
            "сок",
            "молоко",
            "кефир",
            "творог",
            "сыр"
        ]
        
        print(f"\n📁 Папка изображений: {os.path.join(os.path.dirname(__file__), 'images')}")
        print(f"📁 Существует: {os.path.exists(os.path.join(os.path.dirname(__file__), 'images'))}")
        
        for food in test_foods:
            print(f"\n📝 Тестируем: '{food}'")
            image_path = get_food_image(food)
            
            if image_path:
                print(f"   ✅ Найдено изображение: {image_path}")
                print(f"   📁 Существует: {os.path.exists(image_path)}")
            else:
                print(f"   ❌ Изображение не найдено")
                print(f"   💡 Рекомендуемое имя файла: {food.lower().replace(' ', '_')}.jpg")
        
        return True
        
    except Exception as e:
        print(f"❌ Ошибка при тестировании: {e}")
        return False

def main():
    """Основная функция тестирования"""
    print("🧪 Запуск тестов системы изображений\n")
    
    success = test_image_system()
    
    if success:
        print("\n🎉 Тестирование завершено!")
        print("\n💡 Для добавления изображений:")
        print("1. Поместите изображения в папку 'images/'")
        print("2. Назовите файлы согласно правилам в images/README.md")
        print("3. Перезапустите бота")
    else:
        print("\n⚠️ Тестирование завершено с ошибками")
    
    return 0 if success else 1

if __name__ == '__main__':
    exit_code = main()
    sys.exit(exit_code)