from google import genai
from dotenv import load_dotenv
import os
load_dotenv()

client = genai.Client(api_key=os.getenv("GOOGLE_API_KEY"))

def get_response(prompt: str) -> str:
    response = client.models.generate_content(
        model="gemini-2.5-flash-preview-05-20", contents=prompt
    )
    return(response.text)