Update app.py

app.py

@@ -118,3 +118,20 @@ def embeddings_ep(content: Text = None):
                            headers=headers
                            )
     return JSONResponse(content=result.json())
+
+
+@app.post("/create_image")
+def create_image_ep(content: Text = None):
+    url = 'https://api.openai.com/v1/images/generations'
+    obj = json.loads(content.content)
+    data = {
+        "prompt": obj["prompt"],
+        "n": obj["n"],
+        "size": obj["size"]
+    }
+    print("data = \n", data)
+    result = requests.post(url=url,
+                           data=json.dumps(data),
+                           headers=headers
+                           )
+    return JSONResponse(content=result.json())