import json
import httpx
from google.oauth2.credentials import Credentials
from google.auth.transport.requests import Request

def test():
    oauth_path = "/opt/Gemini_Swarm_Core/.gemini/oauth_creds.json"
    try:
        with open(oauth_path) as f:
            creds_data = json.load(f)
    except Exception as e:
        print(f"Error loading oauth_creds.json: {e}")
        return

    creds = Credentials(
        token=creds_data.get("access_token"),
        refresh_token=creds_data.get("refresh_token"),
        token_uri="https://oauth2.googleapis.com/token",
        client_id="681255809395-oo8ft2oprdrnp9e3aqf6av3hmdib135j.apps.googleusercontent.com",
        client_secret="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",
        scopes=["https://www.googleapis.com/auth/cloud-platform"]
    )

    try:
        print("Refreshing credentials...")
        req = Request()
        creds.refresh(req)
        print("Refreshed successfully!")
    except Exception as e:
        print(f"Refresh failed: {e}")
        return

    # Call Gemini API via REST
    print("Calling Gemini API...")
    url = "https://generativelanguage.googleapis.com/v1beta/models/gemini-2.5-flash:generateContent"
    headers = {
        "Authorization": f"Bearer {creds.token}",
        "Content-Type": "application/json"
    }
    data = {
        "contents": [{"parts": [{"text": "Hello! Answer in one short sentence in Russian."}]}]
    }
    
    try:
        r = httpx.post(url, json=data, headers=headers, timeout=10.0)
        print("Status:", r.status_code)
        if r.status_code == 200:
            res_data = r.json()
            text = res_data["candidates"][0]["content"]["parts"][0]["text"]
            print("Response:", text)
        else:
            print("Error response:", r.text)
    except Exception as e:
        print(f"Request failed: {e}")

if __name__ == "__main__":
    test()