Update app.py

app.py

@@ -1,7 +1,9 @@
 # app.py
 import gradio as gr
 import os
-#from PIL import Image
+from PIL import Image
+import requests
+from io import BytesIO
 from openai import OpenAI 
 from dotenv import load_dotenv
 
@@ -12,6 +14,10 @@ load_dotenv()
 api_key = os.getenv("OPENAI_API_KEY")  # Ensure your OpenAI API key is stored in the .env file
 client = OpenAI(api_key=api_key)
 
+from io import BytesIO
+
+# Initialize OpenAI client
+client = OpenAI()
 
 def generate_image(input_text):
     # Call OpenAI's DALL·E 3 model for image generation
@@ -24,7 +30,12 @@ def generate_image(input_text):
     )
     # Extract the image URL from the response
     image_url = response.data[0].url
-    return image_url
+    
+    # Fetch the image from the URL
+    image_response = requests.get(image_url)
+    img = Image.open(BytesIO(image_response.content))
+    
+    return img
 
 # Create a Gradio interface
 def create_interface():
@@ -39,7 +50,7 @@ def create_interface():
             generate_btn = gr.Button("Generate Image")
         
         # Image output
-        image_output = gr.Image(label="Generated Image", type="url")
+        image_output = gr.Image(label="Generated Image", type="pil")
 
         # Define the interaction: When the button is clicked, generate the image
         generate_btn.click(generate_image, inputs=input_text, outputs=image_output)