v 1.0

app.py

@@ -6,7 +6,7 @@ import os
 import nltk
 #nltk.download('punkt')
 
-def generate_image(api_key, prompt):
+def generate_image(api_key, prompt, resolution):
     if not api_key:
         print("Error: API Key is required.")
         return
@@ -14,12 +14,12 @@ def generate_image(api_key, prompt):
     response = openai.Image.create(
         prompt=prompt,
         n=1,
-        size="256x256"
+        size=resolution
     )
 
     image_url = response['data'][0]['url']
 
-    # Open the URL image, resize it to 512x512 and return it
+    # Open the URL image, resize it to the chosen resolution and return it
     with urllib.request.urlopen(image_url) as url:
         with open('temp.jpg', 'wb') as f:
             f.write(url.read())
@@ -31,11 +31,12 @@ iface = gr.Interface(
     fn=generate_image, 
     inputs=[
         gr.inputs.Textbox(lines=1, label="API Key", type="password"), 
-        gr.inputs.Textbox(lines=1, label="Prompt")
+        gr.inputs.Textbox(lines=1, label="Prompt"),
+        gr.inputs.Radio(choices=["256x256", "512x512", "1024x1024"], label="Resolution")
     ], 
     outputs=gr.outputs.Image(type="pil"), 
     title="DALL-E Image Generator",
-    description="Enter your API key and a prompt to generate an image from DALL-E."
+    description="Enter your API key, a prompt, and choose a resolution to generate an image from DALL-E."
 )