Spaces:
Sleeping
Sleeping
import os | |
from dotenv import load_dotenv | |
import google.generativeai as genai | |
import random | |
load_dotenv() | |
gemini_key_env = os.getenv('GEMINI_API_KEY') | |
if ',' in gemini_key_env: | |
keys = [key.strip() for key in gemini_key_env.split(',') if key.strip()] | |
api_key = random.choice(keys) | |
print(f"Selected random key: {api_key[:20]}...") | |
else: | |
api_key = gemini_key_env | |
print(f"Using single key: {api_key[:20]}...") | |
genai.configure(api_key=api_key) | |
model = genai.GenerativeModel('gemini-1.5-pro') | |
response = model.generate_content('Say hello in one word') | |
print(f'Success! Response: {response.text.strip()}') |