Update app.py

app.py

@@ -37,6 +37,39 @@ search_terms_wikipedia = {
     "goldfields coreopsis": "https://en.wikipedia.org/wiki/Coreopsis"
 }
 
+flowers_endangerment = {
+    "Blazing Star": "Not considered endangered.",
+    "Bristlecone Pine": "Least Concern (stable population).",
+    "California Bluebell": "Not listed as endangered or threatened.",
+    "California Buckeye": "Not endangered.",
+    "California Buckwheat": "Generally secure.",
+    "California Fuchsia": "Not endangered overall; some subspecies at risk.",
+    "California Checkerbloom": "Not generally endangered; some subspecies critically imperiled.",
+    "California Lilac": "Most species not endangered; some species are endangered.",
+    "California Poppy": "Generally secure; some subspecies face threats.",
+    "California Sagebrush": "Considered secure (G4-G5).",
+    "California Wild Grape": "Apparently secure (G4).",
+    "California Wild Rose": "Secure (G4).",
+    "Coyote Mint": "Varies by species; some federally listed as endangered.",
+    "Elegant Clarkia": "Secure (G5).",
+    "Baby Blue Eyes": "Secure.",
+    "Hummingbird Sage": "Apparently secure (G4).",
+    "Delphinium": "Varies by species; some are endangered.",
+    "Matilija Poppy": "Not currently endangered.",
+    "Blue-Eyed Grass": "Not endangered.",
+    "Penstemon Spectabilis": "Not endangered.",
+    "Seaside Daisy": "Not endangered.",
+    "Sticky Monkeyflower": "Not endangered.",
+    "Tidy Tips": "Generally not endangered; some subspecies may be at risk.",
+    "Wild Cucumber": "Generally not endangered.",
+    "Douglas Iris": "Not endangered.",
+    "Goldfields Coreopsis": "Varies by species; many not endangered."
+}
+
+def get_status(flower_name):
+    """Return the endangerment status of a given flower name."""
+    return flowers_endangerment.get(flower_name, "Flower not found in database.")
+
 # Templates for AI image generation
 prompt_templates = [
     "A dreamy watercolor scene of a {flower} on a misty morning trail, with golden sunbeams filtering through towering redwoods, and a curious hummingbird hovering nearby.",
@@ -75,8 +108,10 @@ def get_status(flower_name):
 
 # Main function to process the uploaded image
 def process_image(img):
-    # Classify the image
+    print("Starting prediction...")
     predicted_class, _, probs = learn.predict(img)
+    print(f"Prediction complete: {predicted_class}")
+    
     classification_results = dict(zip(learn.dls.vocab, map(float, probs)))
     
     # Get Wikipedia link
@@ -84,8 +119,10 @@ def process_image(img):
     
     # Get endangerment status
     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={
@@ -95,15 +132,19 @@ def process_image(img):
         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))
     
+    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 {