GeezerVison is showing github.com — chart-api/examples/java/ChartAiExample.java at main · chartquery/chart-api

Expand file tree

ChartAiExample.java

Copy path

More file actions

Latest commit

History

49 lines (44 loc) · 2.12 KB

ChartAiExample.java

Copy path

File metadata and controls

49 lines (44 loc) · 2.12 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

// AI Chart API – Java example (Java 15+)

// Usage: CHARTQUERY_API_KEY=your-key java ChartAiExample.java

import java.net.URI;

import java.net.URLEncoder;

import java.net.http.HttpClient;

import java.net.http.HttpRequest;

import java.net.http.HttpResponse;

import java.nio.charset.StandardCharsets;

import java.nio.file.Files;

import java.nio.file.Path;

import java.time.Duration;

import java.util.Set;

import java.util.regex.Matcher;

import java.util.regex.Pattern;

public class ChartAiExample {

static final HttpClient CLIENT = HttpClient.newHttpClient();

static final String KEY = System.getenv("CHARTQUERY_API_KEY");

static String enc(String v) {

return URLEncoder.encode(v, StandardCharsets.UTF_8);

}

static HttpResponse<byte[]> call(String method, String url, String body) throws Exception {

HttpRequest.Builder b = HttpRequest.newBuilder().uri(URI.create(url)).header("x-api-key", KEY)

.timeout(Duration.ofSeconds(120));

b = body == null ? b.method(method, HttpRequest.BodyPublishers.noBody())

: b.header("Content-Type", "application/json").method(method, HttpRequest.BodyPublishers.ofString(body));

HttpResponse<byte[]> r = CLIENT.send(b.build(), HttpResponse.BodyHandlers.ofByteArray());

if (r.statusCode() >= 400) {

System.err.println("HTTP " + r.statusCode() + ": " + new String(r.body(), StandardCharsets.UTF_8));

System.exit(1);

}

return r;

}

static String field(String json, String name) {

Matcher m = Pattern.compile("\"" + name + "\"\\s*:\\s*\"([^\"]*)\"").matcher(json);

return m.find() ? m.group(1) : "";

}

public static void main(String[] args) throws Exception {

String query = "prompt=" + enc("Bar chart of monthly sales from January to June, indigo bars, clean style") + "&"

+ "format=" + enc("png");

HttpResponse<byte[]> response = call("GET", "https://api.chartquery.com/v1/chart/ai" + (query.isEmpty() ? "" : "?" + query), null);

Files.write(Path.of("ai-chart.png"), response.body());

System.out.println("Saved ai-chart.png");

}