4

app.py

@@ -0,0 +1,33 @@
+import gradio as gr
+import openai
+import urllib.request
+from PIL import Image
+import os
+openai.api_key = os.environ['key1']
+
+def generate_image(prompt):
+    response = openai.Image.create(
+        prompt=prompt,
+        n=1,
+        size="1024x1024"
+    )
+
+    image_url = response['data'][0]['url']
+
+    # Open the URL image, resize it to 512x512 and return it
+    with urllib.request.urlopen(image_url) as url:
+        with open('temp.jpg', 'wb') as f:
+            f.write(url.read())
+        img = Image.open('temp.jpg')
+        img = img.resize((512, 512))
+
+    return img
+
+iface = gr.Interface(
+    fn=generate_image, 
+    inputs="text", 
+    outputs=gr.outputs.Image(type='pil'),
+    examples=["a macaw in a hat sitting on a bench in a park"]
+)
+
+iface.launch()

requirements.txt

@@ -0,0 +1,2 @@
+urllib
+pil