import ai.sunra.client.*;
import ai.sunra.client.queue.*;
import java.util.Map;
import com.google.gson.JsonObject;
import java.util.function.Consumer;
ClientConfig config = ClientConfig.builder()
.withCredentials(CredentialsResolver.fromEnv())
.build();
SunraClient client = SunraClient.withConfig(config);
Map<String, Object> input = Map.of(
"prompt", "एक कांच की चायदानी जिसमें खिले हुए फूल की चाय है, जो एक लकड़ी की मेज पर एक धूप वाली खिड़की के पास रखी है, जिसमें सुबह की कोमल रोशनी है।",
"prompt_enhancer", true,
"aspect_ratio", "16:9",
"output_format", "jpeg",
"safety_tolerance", 2
);
Consumer<QueueStatus.StatusUpdate> statusUpdateHandler = update -> {
String status = update.getStatus().toString();
String message = String.format("\nस्थिति अद्यतन: %s, अनुरोध आईडी: %s",
status, update.getRequestId());
System.out.println(message);
};
SubscribeOptions<JsonObject> options = SubscribeOptions.<JsonObject>builder()
.input(input)
.resultType(JsonObject.class)
.onQueueUpdate(statusUpdateHandler)
.logs(true)
.build();
Output<JsonObject> response = client.subscribe("black-forest-labs/flux-kontext-max/text-to-image", options);
System.out.println("पूरा हुआ!");
System.out.println(response.getData());