Update app.py

app.py

@@ -1,5 +1,4 @@
 import base64
-import fal_client
 import gradio as gr
 import io
 import openai
@@ -95,13 +94,6 @@ example_images = [
 ]
 
 
-# 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):
     print("Starting prediction...")
@@ -147,7 +139,7 @@ def clear_outputs():
         label_output: None,
         generated_image: None,
         wiki_output: None,
-        status_output: None  # ← NEW
+        endangerment_output: None  # ← NEW
     }
 
 
@@ -165,7 +157,7 @@ with gr.Blocks() as demo:
         with gr.Column():
             label_output = gr.Label(label="Classification Results")
             wiki_output = gr.Textbox(label="Wikipedia Article Link", lines=1)
-            status_output = gr.Textbox(label="Endangerment Status", lines=1)  # ← NEW
+            endangerment_output = gr.Textbox(label="Endangerment Status", lines=1)  # ← NEW
         generated_image = gr.Image(label="AI Generated Interpretation")
 
     # Add example images using local paths
@@ -174,19 +166,19 @@ with gr.Blocks() as demo:
         inputs=input_image,
         examples_per_page=6,
         fn=process_image,
-        outputs=[label_output, generated_image, wiki_output, status_output]  # ← UPDATED
+        outputs=[label_output, generated_image, wiki_output, endangerment_output]
     )
 
     input_image.change(
         fn=process_image,
         inputs=input_image,
-        outputs=[label_output, generated_image, wiki_output, status_output]  # ← UPDATED
+        outputs=[label_output, generated_image, wiki_output, endangerment_output]
     )
 
     input_image.clear(
         fn=clear_outputs,
         inputs=[],
-        outputs=[label_output, generated_image, wiki_output, status_output]  # ← UPDATED
+        outputs=[label_output, generated_image, wiki_output, endangerment_output]
     )
 
 # Start the application