Google Gemini Salesforce Integration using Apex

public class GeminiCallout {
    public static void makePostCallout() {
        String endpoint = 'https://generativelanguage.googleapis.com/v1beta/models/gemini-pro:generateContent';
        String apiKey = 'API_KEY';
        endpoint+= '?key=' + apiKey;
        HttpRequest req = new HttpRequest();
        req.setEndpoint(endpoint);
        req.setMethod('POST');
        req.setTimeout(120000);
        req.setHeader('Content-Type', 'application/json');
        String requestBody = '{\"contents\":[{\"parts\":[{\"text\":\"Write a story in 3 lines about a girl.\"}]}]}';
        req.setBody(requestBody);
        Http http = new Http();
        HttpResponse res = http.send(req);
        if (res.getStatusCode() == 200) {
            String jsonResponse = res.getBody();
			
            // Parse the JSON response
            Map<String, Object> parsedResponse = 
            (Map<String, Object>)JSON.deserializeUntyped(jsonResponse);
            
            // Navigate through the nested structure to get to the "candidates" list
            List<Object> candidates = (List<Object>)parsedResponse.get('candidates');
            
            // Assuming there's at least one candidate and you want the first one
            Map<String, Object> firstCandidate = (Map<String, Object>)candidates[0];
            
            // Navigate to "content" -> "parts"
            Map<String, Object> content = (Map<String, Object>)firstCandidate.get('content');
            List<Object> parts = (List<Object>)content.get('parts');
            
            // Assuming there's at least one part and you want the text from the first part
            Map<String, Object> firstPart = (Map<String, Object>)parts[0];
            String text = (String)firstPart.get('text');
            System.debug('Callout Result - ' + text);
		} else {
            System.debug('HTTP error: ' + res.getStatusCode() + ' ' + res.getStatus());
        }
	}
}

GeminiCallout.makePostCallout();