import sys
import os
from google import genai

def main():
    if len(sys.argv) < 2:
        print("Usage: python3 gemini_wrapper.py <prompt>")
        sys.exit(1)

    api_key = os.getenv("GOOGLE_API_KEY")
    if not api_key:
        print("Error: GOOGLE_API_KEY environment variable not set.")
        sys.exit(1)

    client = genai.Client(api_key=api_key)
    
    # Using gemini-2.0-flash for speed/cost efficiency in loops
    response = client.models.generate_content(model="gemini-2.0-flash", contents=sys.argv[1])
    print(response.text)

if __name__ == "__main__":
    main()