menamiai / test_gemini_only.py
dfdfdsfgs's picture
Add Hugging Face Spaces frontend with API key fallback system
4588d9f
raw
history blame contribute delete
621 Bytes
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()}')