Update app.py

app.py

@@ -4,6 +4,8 @@ import fal_client
 from PIL import Image
 import io
 import base64
+import random
+import requests
 
 search_terms_wikipedia = {
     "blazing star": "https://en.wikipedia.org/wiki/Mentzelia",
@@ -34,6 +36,15 @@ search_terms_wikipedia = {
     "goldfields coreopsis": "https://en.wikipedia.org/wiki/Coreopsis"
 }
 
+# Update prompt templates
+prompt_templates = [
+    "A cosmic {flower} blooming in space, with petals made of swirling galaxies and nebulae, glowing softly against a backdrop of distant stars.",
+    "An enchanted garden filled with a bioluminescent {flower}, each petal radiating vibrant, otherworldly colors, illuminating the dark, mystical forest around them.",
+    "A mechanical {flower} with petals made of polished metal and intricate gears, unfolding in a steampunk-inspired futuristic landscape.",
+    "A surreal pot of a {flower} where each bloom is a miniature landscape, showing tiny mountains, rivers, and clouds nestled within the petals.",
+    "An abstract explosion of a {flower}, blending vibrant colors and fluid shapes in a chaotic, dreamlike composition, evoking movement and emotion."
+]
+
 def on_queue_update(update):
     if isinstance(update, fal_client.InProgress):
         for log in update.logs:
@@ -52,16 +63,16 @@ def process_image(img):
     result = fal_client.subscribe(
         "fal-ai/flux/schnell",
         arguments={
-            "prompt": f"A detailed, artistic interpretation of {predicted_class} flower in natural setting",
-            "image_size": "square"  # Changed from "256x256" to "square"
+            "prompt": random.choice(prompt_templates).format(flower=predicted_class),
+            "image_size": "square"
         },
         with_logs=True,
         on_queue_update=on_queue_update,
     )
     
-    # Convert the image data
-    image_data = base64.b64decode(result['image'])
-    generated_image = Image.open(io.BytesIO(image_data))
+    image_url = result['images'][0]['url']
+    response = requests.get(image_url)
+    generated_image = Image.open(io.BytesIO(response.content))
     
     return classification_results, generated_image, wiki_url
 
@@ -81,17 +92,7 @@ with gr.Blocks() as demo:
     # Example images
     examples = [
         'https://www.deserthorizonnursery.com/wp-content/uploads/2024/03/Brittlebush-Encelia-Farinosa-desert-horizon-nursery.jpg',
-        'https://cdn.mos.cms.futurecdn.net/VJE7gSuQ9KWbkqEsWgX5zS.jpg',
-        'https://silverfallsseed.com/wp-content/uploads/2016/01/tidy-tips-_-9.jpg',
-        'https://cdn11.bigcommerce.com/s-1b9100svju/images/stencil/1280x1280/products/2044/1440/DETA-635__75522.1664817787.jpg?c=1',
-        'https://shop.stoverseed.com/cdn/shop/products/DesertBluebells_CaliforniaBluebells-_Phaceliacampanularia_closeup_530x@2x.jpg?v=1596506040',
-        'https://images.squarespace-cdn.com/content/v1/57488264f85082748c7c80db/d0e8480a-4eff-4219-88f8-e5ae6b32e81e/monkeyflower.jpg',
-        'https://www.parksconservancy.org/sites/default/files/styles/basic/public/A_GEN_131213_WTE_109.jpg?itok=9SDtr4b2',
-        'https://silverfallsseed.com/wp-content/uploads/2016/01/tidy-tips-_-9.jpg',
-        'https://vnps.org/wp-content/uploads/2021/05/Narrow-leaved-Blue-eyed-Grass-4-Sisyrinchium-angustifolium-by-Betsy-Washington.jpg',
-        'https://gardenerspath.com/wp-content/uploads/2022/02/How-to-Grow-Baby-Blue-Eyes-Feature.jpg',
-        'https://images.squarespace-cdn.com/content/v1/57488264f85082748c7c80db/d0e8480a-4eff-4219-88f8-e5ae6b32e81e/monkeyflower.jpg'
-        
+        'https://cdn.mos.cms.futurecdn.net/VJE7gSuQ9KWbkqEsWgX5zS.jpg'
     ]
     gr.Examples(
         examples=examples,