Update app.py

app.py

@@ -141,7 +141,7 @@ def process_image(img):
     generated_image = Image.open(io.BytesIO(response.content))
     
     print("Image retrieved and ready to return")
-    return classification_results, generated_image, wiki_url, endangerment_status
+    return classification_results, #generated_image, wiki_url, endangerment_status
 
 
 
@@ -149,9 +149,9 @@ def process_image(img):
 def clear_outputs():
     return {
         label_output: None,
-        generated_image: None,
-        wiki_output: None,
-        status_output: None  # ← NEW
+       # generated_image: None,
+       # wiki_output: None,
+        #status_output: None  # ← NEW
     }
 
 
@@ -169,9 +169,9 @@ with gr.Blocks() as demo:
     with gr.Row():
         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
-        generated_image = gr.Image(label="AI Generated Interpretation")
+            #wiki_output = gr.Textbox(label="Wikipedia Article Link", lines=1)
+            #status_output = gr.Textbox(label="Endangerment Status", lines=1)  # ← NEW
+        #generated_image = gr.Image(label="AI Generated Interpretation")
 
     # Add example images using local paths
     gr.Examples(
@@ -179,19 +179,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, status_output]  # ← UPDATED
 )
 
     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, status_output]  # ← UPDATED
 )
 
     input_image.clear(
         fn=clear_outputs,
         inputs=[],
-        outputs=[label_output, generated_image, wiki_output, status_output]  # ← UPDATED
+        outputs=[label_output]#, generated_image, wiki_output, status_output]  # ← UPDATED
 )