| import os | |
| import google.generativeai as genai | |
| API_KEY = os.environ.get("GOOGLE_API_KEY", "[GOOGLE_API_KEY]") | |
| genai.configure(api_key=API_KEY) | |
| model = genai.GenerativeModel('gemini-pro') | |
| def generate_content(prompt, _place_holder = None): | |
| response = model.generate_content(prompt) | |
| return response.text | |
| if __name__ == "__main__": | |
| print(generate_content("What are you?")) | |