import os
import urllib.request
import urllib.error
import json
import google.auth
import google.auth.transport.requests

os.environ["GOOGLE_APPLICATION_CREDENTIALS"] = "/root/gcp-key.json"

def test():
    try:
        print("Obtaining Google credentials...")
        credentials, project = google.auth.default(
            scopes=["https://www.googleapis.com/auth/cloud-platform"]
        )
        auth_req = google.auth.transport.requests.Request()
        credentials.refresh(auth_req)
        print(f"Token obtained successfully! Project: {project}")
        
        # Call Gemini API
        url = "https://generativelanguage.googleapis.com/v1beta/models/gemini-1.5-flash:generateContent"
        headers = {
            "Authorization": f"Bearer {credentials.token}",
            "Content-Type": "application/json",
            "x-goog-user-project": "gen-lang-client-0207686288"
        }
        data = {
            "contents": [{"parts": [{"text": "Hello, Gemini! Answer with a short sentence."}]}]
        }
        
        print("Calling Gemini API...")
        req = urllib.request.Request(url, data=json.dumps(data).encode('utf-8'), headers=headers, method="POST")
        with urllib.request.urlopen(req) as response:
            res = json.loads(response.read().decode('utf-8'))
            text = res["candidates"][0]["content"]["parts"][0]["text"]
            print(f"Response from Gemini:\n{text}")
    except urllib.error.HTTPError as e:
        print(f"HTTP Error calling Gemini: {e}")
        try:
            print(f"Error body: {e.read().decode('utf-8')}")
        except Exception as read_err:
            print(f"Could not read error body: {read_err}")
    except Exception as e:
        print(f"Error calling Gemini: {e}")

if __name__ == "__main__":
    test()