1  #!/usr/bin/env python3
 2  """WORKING Google Gemini API Example"""
 3  
 4  import google.generativeai as genai
 5  import sys
 6  
 7  # Configure with your API key
 8  GEMINI_API_KEY = "AIzaSyC9g4B4sY9xeaUntjNmN2MeWFyp5gL3_EM"
 9  
10  # Configure the API
11  genai.configure(api_key=GEMINI_API_KEY)
12  
13  # Create the model
14  model = genai.GenerativeModel('gemini-pro')
15  
16  def query_gemini(prompt):
17      """Query Gemini with proper SDK"""
18      try:
19          response = model.generate_content(prompt)
20          return response.text
21      except Exception as e:
22          return f"Error: {str(e)}"
23  
24  def main():
25      prompt = sys.argv[1] if len(sys.argv) > 1 else "Explain AI in one sentence"
26      
27      print(f"🔍 Querying Gemini: {prompt}")
28      print("="*50)
29      
30      result = query_gemini(prompt)
31      
32      print("📝 Response:")
33      print(result)
34      print("="*50)
35  
36  if __name__ == "__main__":
37      main()