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()}') |