Update app.py

app.py

@@ -1,3 +1,4 @@
+import base64
 import fal_client
 import gradio as gr
 import io
@@ -8,6 +9,7 @@ import requests
 from PIL import Image
 from dotenv import load_dotenv
 from fastai.vision.all import *
+from openai import OpenAI
 from pathlib import Path
 
 # Dictionary of plant names and their Wikipedia links
@@ -119,21 +121,18 @@ def process_image(img):
     # Generate artistic interpretation using DALL-E
     print("Sending request to DALL-E...")
     try:
-        response = openai.images.generate(
-            model="dall-e-3",
-            prompt=random.choice(prompt_templates).format(flower=predicted_class),
+        client = OpenAI()
+
+        result = client.images.generate(
+            model="gpt-image-1",
+            prompt="Draw a 2D pixel art style sprite sheet of a tabby gray cat",
             size="1024x1024",
-            quality="standard",
-            n=1,
+            background="transparent",
+            quality="high",
         )
 
-        # Get the image URL
-        image_url = response.data[0].url
-        print(f"Image URL: {image_url}")
-
-        # Download the generated image
-        response = requests.get(image_url)
-        generated_image = Image.open(io.BytesIO(response.content))
+        image_base64 = result.json()["data"][0]["b64_json"]
+        generated_image = base64.b64decode(image_base64)
 
     except Exception as e:
         print(f"Error generating image: {e}")