import os
import google.generativeai as genai
from dotenv import load_dotenv
import logging

# Налаштування логування
logging.basicConfig(level=logging.INFO)
logger = logging.getLogger(__name__)

def test_gemini():
    try:
        # Завантаження API ключа
        load_dotenv()
        api_key = os.getenv('GEMINI_API_KEY')
        if not api_key:
            raise ValueError("GEMINI_API_KEY не знайдено")
            
        # Конфігурація Gemini
        genai.configure(api_key=api_key)
        
        # Створення моделі
        model = genai.GenerativeModel('gemini-pro')
        
        # Тестовий запит
        prompt = "What is 2+2? Answer in one word."
        logger.info(f"Sending test prompt: {prompt}")
        
        # Генерація відповіді
        response = model.generate_content(prompt)
        
        if response and hasattr(response, 'text'):
            logger.info(f"Response received: {response.text}")
            return True
        else:
            logger.error("No valid response received")
            return False
            
    except Exception as e:
        logger.error(f"Error testing Gemini: {str(e)}")
        return False

if __name__ == "__main__":
    print("Testing Gemini connection...")
    if test_gemini():
        print("Test successful!")
    else:
        print("Test failed!")