1

txt2images.js

@@ -0,0 +1,30 @@
+const axios = require('axios')
+api_base = process.env.OPENAI_API_BASE;
+api_key = process.env.OPENAI_API_KEY;
+api_version = '2023-06-01-preview'
+url = api_base + "openai/images/generations:submit?api-version=" + api_version
+headers = { "api-key": api_key, "Content-Type": "application/json" }
+
+function sleep(ms) {
+    return new Promise((resolve) => setTimeout(resolve, ms));
+}
+async function txt2images(prompt,size) {
+    body = {
+        "prompt": prompt,
+        "size": size,
+        "n": 1
+    }
+    submission = await axios.post(url, body, { headers });
+    operation_location = submission.headers['operation-location'];
+    status = "";
+    while (status != "succeeded") {
+        await sleep(1000);
+        res = await axios.get(operation_location, { headers });
+        status = res.data.status;
+        console.log(status)
+        if (status == "succeeded") {
+            return res.data.result.data[0].url;
+        }
+    }
+}
+exports.txt2img = txt2images;