1  #!/usr/bin/env python3
 2  """Direct REST API for Gemini that WORKS"""
 3  
 4  import requests
 5  import json
 6  import sys
 7  
 8  API_KEY = "AIzaSyC9g4B4sY9xeaUntjNmN2MeWFyp5gL3_EM"
 9  url = f"https://generativelanguage.googleapis.com/v1/models/gemini-pro:generateContent?key={API_KEY}"
10  
11  def query_gemini(prompt):
12      headers = {'Content-Type': 'application/json'}
13      
14      data = {
15          "contents": [{
16              "parts": [{"text": prompt}]
17          }]
18      }
19      
20      try:
21          response = requests.post(url, headers=headers, json=data, timeout=10)
22          print(f"Status Code: {response.status_code}")
23          
24          if response.status_code == 200:
25              result = response.json()
26              return result['candidates'][0]['content']['parts'][0]['text']
27          else:
28              return f"Error {response.status_code}: {response.text[:200]}"
29      except Exception as e:
30          return f"Exception: {str(e)}"
31  
32  def main():
33      prompt = sys.argv[1] if len(sys.argv) > 1 else "Explain AI in one sentence"
34      
35      print(f"🔍 Querying: {prompt}")
36      print("="*50)
37      
38      result = query_gemini(prompt)
39      
40      print("📝 Response:")
41      print(result)
42      print("="*50)
43  
44  if __name__ == "__main__":
45      main()