fix delete image

app.py

@@ -6,6 +6,7 @@ import io
 import random
 import requests
 
+# Dictionary of plant names and their Wikipedia links
 search_terms_wikipedia = {
     "blazing star": "https://en.wikipedia.org/wiki/Mentzelia",
     "bristlecone pine": "https://en.wikipedia.org/wiki/Pinus_longaeva",
@@ -35,7 +36,7 @@ search_terms_wikipedia = {
     "goldfields coreopsis": "https://en.wikipedia.org/wiki/Coreopsis"
 }
 
-# Update prompt templates
+# Templates for AI image generation
 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.",
@@ -44,20 +45,32 @@ prompt_templates = [
     "An abstract explosion of a {flower}, blending vibrant colors and fluid shapes in a chaotic, dreamlike composition, evoking movement and emotion."
 ]
 
+# Example images for the interface
+example_images = [
+    '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://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://valleywaternews.org/wp-content/uploads/2016/06/ceanothus-clusters.jpg?w=1440',
+    'https://cdn11.bigcommerce.com/s-1b9100svju/images/stencil/1280x1280/products/2044/1440/DETA-635__75522.1664817787.jpg?c=1'
+]
+
+# Function to handle AI generation progress updates
 def on_queue_update(update):
     if isinstance(update, fal_client.InProgress):
         for log in update.logs:
            print(log["message"])
 
+# Main function to process the uploaded image
 def process_image(img):
-    # First do the classification
-    predicted_class, idx, probs = learn.predict(img)
+    # Classify the image
+    predicted_class, _, probs = learn.predict(img)
     classification_results = dict(zip(learn.dls.vocab, map(float, probs)))
     
-    # Get Wikipedia URL for the predicted class
+    # Get Wikipedia link
     wiki_url = search_terms_wikipedia.get(predicted_class, "No Wikipedia entry found.")
     
-    # Generate FLUX image
+    # Generate artistic interpretation by calling the Flux API
     result = fal_client.subscribe(
         "fal-ai/flux/schnell",
         arguments={
@@ -68,47 +81,58 @@ def process_image(img):
         on_queue_update=on_queue_update,
     )
     
+    # Get the generated image
     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
 
-# Load the learner
+# Function to clear all outputs
+def clear_outputs():
+    return {
+        label_output: None,
+        generated_image: None,
+        wiki_output: None
+    }
+
+# Load the AI model
 learn = load_learner('export.pkl')
 
-# Create Gradio interface
+# Create the web interface
 with gr.Blocks() as demo:
+    # Input section
     with gr.Row():
         input_image = gr.Image(height=192, width=192, label="Upload Image for Classification", type="pil")
+    
+    # Output section
     with gr.Row():
         with gr.Column():
             label_output = gr.Label(label="Classification Results")
             wiki_output = gr.Textbox(label="Wikipedia Article Link", lines=1)
         generated_image = gr.Image(label="AI Generated Interpretation")
     
-    # 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://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://valleywaternews.org/wp-content/uploads/2016/06/ceanothus-clusters.jpg?w=1440',
-        'https://cdn11.bigcommerce.com/s-1b9100svju/images/stencil/1280x1280/products/2044/1440/DETA-635__75522.1664817787.jpg?c=1'
-    ]
+    # Add example images
     gr.Examples(
-        examples=examples,
+        examples=example_images,
         inputs=input_image,
         examples_per_page=6,
         fn=process_image,
         outputs=[label_output, generated_image, wiki_output]
     )
     
-    # Set up event handler
+    # Set up what happens when an image is uploaded or removed
     input_image.change(
         fn=process_image,
         inputs=input_image,
         outputs=[label_output, generated_image, wiki_output]
     )
+    
+    input_image.clear(
+        fn=clear_outputs,
+        inputs=[],
+        outputs=[label_output, generated_image, wiki_output]
+    )
 
+# Start the application
 demo.launch(inline=False)