stability_sdk 0.0.2

An implementation of Stability AI SDK in Dart.

example/main.dart

import 'package:cached_memory_image/cached_memory_image.dart';
import 'package:flutter/material.dart';
import 'package:flutter_dotenv/flutter_dotenv.dart';
import 'package:stability_sdk/stability_sdk.dart';

Future<void> main() async {
  await dotenv.load(fileName: ".env");
  runApp(const MyApp());
}

class MyApp extends StatelessWidget {
  const MyApp({super.key});

  @override
  Widget build(BuildContext context) {
    return MaterialApp(
      title: 'Brush AI',
      theme: ThemeData(
        primarySwatch: Colors.blue,
      ),
      home: const MyHomePage(title: 'Brush AI'),
    );
  }
}

class MyHomePage extends StatefulWidget {
  const MyHomePage({super.key, required this.title});

  final String title;

  @override
  State<MyHomePage> createState() => _MyHomePageState();
}

class _MyHomePageState extends State<MyHomePage> {
  late StabilityApiClient client;
  late TextEditingController queryController;
  String? image;

  @override
  void initState() {
    client = StabilityApiClient.init(dotenv.get('STABILITY_API_KEY'));
    queryController = TextEditingController();

    super.initState();
  }

  Future<void> generateImage(String prompt) async {
    final request = StabilityRequest(prompt);
    client.generateAsync(request).listen((answer) {
      if (answer?.artifacts.isNotEmpty == true) {
        setState(() {
          image = answer!.artifacts.first.getImage();
        });
      }
    });
  }

  @override
  Widget build(BuildContext context) {
    return Scaffold(
      appBar: AppBar(
        title: Text(widget.title),
      ),
      body: Center(
        child: Column(
          mainAxisAlignment: MainAxisAlignment.center,
          children: <Widget>[
            const Text(
              'You have pushed the button this many times:',
            ),
            TextField(
              controller: queryController,
            ),
            if (image != null)
              CachedMemoryImage(
                base64: image,
                uniqueKey: image.toString(),
              )
            else
              const CircularProgressIndicator.adaptive()
          ],
        ),
      ),
      floatingActionButton: FloatingActionButton(
        onPressed: () {
          if (queryController.text.isNotEmpty) {
            generateImage(queryController.text);
          }
        },
        tooltip: 'Generate Art',
        child: const Icon(Icons.add),
      ),
    );
  }
}