feat: Added image create method

example/src/App.tsx

@@ -2,13 +2,16 @@ import * as React from 'react';
 
 import {
   Animated,
+  Button,
+  Image,
   Keyboard,
   Platform,
   SafeAreaView,
   ScrollView,
   StyleSheet,
   Text,
   TextInput,
+  View,
   useColorScheme,
 } from 'react-native';
 import OpenAI from 'react-native-openai';
@@ -50,7 +53,7 @@ export default function App() {
       showSubscription.remove();
       hideSubscription.remove();
     };
-  }, []);
+  }, [yPosition]);
 
   React.useEffect(() => {
     openAI.chat.addListener('onChatMessageReceived', (payload) => {
@@ -68,6 +71,10 @@ export default function App() {
     };
   }, [openAI]);
 
+  const [mode, setMode] = React.useState<'text' | 'image'>('text');
+
+  const [images, setImages] = React.useState<string[]>([]);
+
   return (
     <SafeAreaView
       style={[
@@ -99,16 +106,24 @@ export default function App() {
               return;
             }
             setResult('');
+            setImages([]);
             console.log(e.nativeEvent.text);
-            openAI.chat.stream({
-              messages: [
-                {
-                  role: 'user',
-                  content: e.nativeEvent.text,
-                },
-              ],
-              model: 'gpt-3.5-turbo',
-            });
+            if (mode === 'text') {
+              openAI.chat.stream({
+                messages: [
+                  {
+                    role: 'user',
+                    content: e.nativeEvent.text,
+                  },
+                ],
+                model: 'gpt-3.5-turbo',
+              });
+            } else {
+              const result = await openAI.image.create({
+                prompt: e.nativeEvent.text,
+              });
+              setImages(result.data.map((image) => image.url));
+            }
           }}
           style={[
             styles.input,
@@ -119,6 +134,22 @@ export default function App() {
           ]}
         />
       </Animated.View>
+      <View style={{ flexDirection: 'row' }}>
+        <Button
+          title="Text"
+          color={mode === 'text' ? 'darkblue' : undefined}
+          onPress={() => {
+            setMode('text');
+          }}
+        />
+        <Button
+          title="Image"
+          color={mode === 'image' ? 'darkblue' : undefined}
+          onPress={() => {
+            setMode('image');
+          }}
+        />
+      </View>
       <ScrollView
         style={{ width: '100%', backgroundColor: 'transparent' }}
         contentContainerStyle={{
@@ -136,6 +167,21 @@ export default function App() {
         >
           Result: {result}
         </Text>
+        {images.map((image) => (
+          <View
+            key={image}
+            style={{
+              width: '100%',
+              height: 300,
+              backgroundColor: 'transparent',
+            }}
+          >
+            <Image
+              source={{ uri: image }}
+              style={{ width: '100%', height: '100%' }}
+            />
+          </View>
+        ))}
       </ScrollView>
     </SafeAreaView>
   );

ios/OpenAIKit/Image/Image.swift

@@ -4,7 +4,7 @@ public struct Image {
     public let url: String
 }
 
-extension Image: Decodable {}
+extension Image: Codable {}
 
 extension Image {
     public enum Size: String {

ios/OpenAIKit/Image/ImageInput.swift

@@ -0,0 +1,9 @@
+import Foundation
+
+struct ImageInput: Codable {
+    let prompt: String
+    var n: Int?
+    var size: Image.Size?
+    var user: String?
+
+}

ios/OpenAIKit/Image/ImageResponse.swift

@@ -5,4 +5,6 @@ public struct ImageResponse {
     public let data: [Image]
 }
 
-extension ImageResponse: Decodable {}
+extension ImageResponse: Codable {
+
+}

ios/ReactNativeOpenai.mm

@@ -12,6 +12,10 @@ @interface RCT_EXTERN_MODULE(ReactNativeOpenai, RCTEventEmitter)
 RCT_EXTERN_METHOD(create:(NSDictionary *)input withResolver:(RCTPromiseResolveBlock)resolve
                   withRejecter:(RCTPromiseRejectBlock)reject)
 
+// Image
+RCT_EXTERN_METHOD(imageCreate:(NSDictionary *)input withResolver:(RCTPromiseResolveBlock)resolve
+                  withRejecter:(RCTPromiseRejectBlock)reject)
+
 
 + (BOOL)requiresMainQueueSetup
 {

ios/ReactNativeOpenai.swift

@@ -151,4 +151,30 @@ extension ReactNativeOpenai {
             }
         }
     }
+
+    // MARK: - Image
+    @objc(imageCreate:withResolver:withRejecter:)
+    public func imageCreate(input: NSDictionary,resolve: @escaping RCTPromiseResolveBlock, reject: @escaping RCTPromiseRejectBlock){
+        Task {
+            do {
+                let decoded = try DictionaryDecoder().decode(ImageInput.self, from: input)
+                let imageResult = try await openAIClient.images.create(
+                    prompt: decoded.prompt,
+                    n:decoded.n ?? 1,
+                    size:decoded.size ?? .fiveTwelve,
+                    user:decoded.user
+                )
+
+                if let payload = String(data: try JSONEncoder().encode(imageResult), encoding: .utf8) {
+                    resolve(payload)
+                } else {
+                    reject("error", "error", nil)
+                }
+
+            } catch {
+                reject("error", "error", error)
+            }
+        }
+    }
+
 }

src/index.tsx

@@ -17,11 +17,13 @@ class OpenAI {
   module = NativeModules.ReactNativeOpenai;
   private bridge: NativeEventEmitter;
   public chat: Chat;
+  public image: Image;
 
   public constructor(config: Config) {
     this.bridge = new NativeEventEmitter(this.module);
     this.module.initialize(config);
     this.chat = new Chat(this.module, this.bridge);
+    this.image = new Image(this.module);
   }
 }
 
@@ -112,6 +114,25 @@ namespace ChatModels {
   };
 }
 
+namespace ImageModels {
+  type ImageSize = '256x256' | '512x512' | '1024x1024';
+
+  export type ImageInput = {
+    prompt: string;
+    n?: number;
+    size?: ImageSize;
+  };
+
+  type Image = {
+    url: string;
+  };
+
+  export type ImageOutput = {
+    created: Date;
+    data: Image[];
+  };
+}
+
 class Chat {
   private bridge: NativeEventEmitter;
   private module: any;
@@ -150,4 +171,22 @@ class Chat {
   }
 }
 
+class Image {
+  private module: any;
+
+  public constructor(module: any) {
+    this.module = module;
+  }
+
+  public async create(
+    input: ImageModels.ImageInput
+  ): Promise<ImageModels.ImageOutput> {
+    const result = await this.module.imageCreate(input);
+    if (Platform.OS === 'ios') {
+      return JSON.parse(result);
+    }
+    return result;
+  }
+}
+
 export default OpenAI;