Update app.py

app.py

@@ -2,10 +2,10 @@ import base64
 import gradio as gr
 import random
 from fastai.vision.all import *
-from openai import OpenAI
 from pathlib import Path
 from PIL import Image
 import io
+import fal
 
 # Dictionary of plant names and their Wikipedia links
 search_terms_wikipedia = {
@@ -90,6 +90,11 @@ example_images = [
     str(Path('example_images/example_5.jpg'))
 ]
 
+# Function to handle FAL queue updates
+def on_queue_update(update):
+    """Handle queue updates from FAL."""
+    print(f"Queue update: {update}")
+
 
 # Main function to process the uploaded image
 def process_image(img, generate_image=True):
@@ -109,20 +114,21 @@ def process_image(img, generate_image=True):
     endangerment_status = get_status(predicted_class)
     print(f"Status found: {endangerment_status}")
 
-    # Generate artistic interpretation using DALL-E
-    print("Sending request to DALL-E...")
+    # Generate artistic interpretation using FAL
+    print("Sending request to FAL...")
     try:
-        client = OpenAI()
-
         if generate_image:
-            result = client.images.generate(
-                model="gpt-image-1",   
-                prompt=random.choice(prompt_templates).format(flower=predicted_class),
-                size="1024x1024",
-                quality="low"
+            prompt = random.choice(prompt_templates).format(flower=predicted_class)
+            
+            # Use fal-client to generate image
+            result = fal.subscribe(
+                "fal-ai/flux/dev",
+                input={"prompt": prompt},
+                on_queue_update=on_queue_update
             )
-
-            image_base64 = result.data[0].b64_json
+            
+            # Get the image from the result
+            image_base64 = result["images"][0]
             image_bytes = base64.b64decode(image_base64)
             generated_image = Image.open(io.BytesIO(image_bytes))
         else:
@@ -142,7 +148,7 @@ def clear_outputs():
         label_output: None,
         generated_image: None,
         wiki_output: None,
-        endangerment_output: None  # ← NEW
+        endangerment_output: None
     }