change to openai image generation

app.py

@@ -106,6 +106,7 @@ def get_status(flower_name):
 
 
 
+# Main function to process the uploaded image
 # Main function to process the uploaded image
 def process_image(img):
     print("Starting prediction...")
@@ -121,30 +122,33 @@ def process_image(img):
     endangerment_status = get_status(predicted_class)
     print(f"Status found: {endangerment_status}")
     
-    # Generate artistic interpretation
-    print("Sending request to FAL API...")
-    result = fal_client.subscribe(
-        "fal-ai/flux/schnell",
-        arguments={
-            "prompt": random.choice(prompt_templates).format(flower=predicted_class),
-            "image_size": "portrait_4_3"
-        },
-        with_logs=True,
-        on_queue_update=on_queue_update,
-    )
-    print("FAL API responded")
-
-    # Retrieve image
-    image_url = result['images'][0]['url']
-    print(f"Image URL: {image_url}")
-    response = requests.get(image_url)
-    generated_image = Image.open(io.BytesIO(response.content))
+    # 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),
+            size="1024x1024",
+            quality="standard",
+            n=1,
+        )
+        
+        # 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))
+        
+    except Exception as e:
+        print(f"Error generating image: {e}")
+        generated_image = None
     
     print("Image retrieved and ready to return")
     return classification_results, generated_image, wiki_url, endangerment_status
 
 
-
 # Function to clear all outputs
 def clear_outputs():
     return {