Model prompt change

app.py

@@ -83,11 +83,22 @@ def explain_recycling(class_label):
 
 
 # ------------------ PIPELINE ------------------
-def full_pipeline(image):
+top_label = None
+
+def classify_pipeline(image):
+    global top_label
+
     predictions = classify_image(image)
     top_label = max(predictions, key=predictions.get)  # top-1
+    return predictions
+
+def analyze_pipeline():
+    global top_label
+
+    if top_label is None:
+        return "Please classify an image first."
     explanation = explain_recycling(top_label)
-    return predictions, explanation
+    return explanation
 
 
 # ------------------ GRADIO UI ------------------
@@ -181,6 +192,13 @@ with gr.Blocks() as demo:
         num_top_classes=3,
         label="🔍 Classifier Prediction (Top 3)"
     )
+    analyze_btn = gr.Button("Analyze Waste", variant="primary")
+
+    analyze_btn.click(
+        classify_pipeline,
+        inputs=img_input,
+        outputs=cls_output
+    )
 
     explain_output = gr.Textbox(
         label="🧩 Detailed Recycling & Disposal Advice",
@@ -189,12 +207,12 @@ with gr.Blocks() as demo:
     )
 
     # BUTTON
-    analyze_btn = gr.Button("Analyze Waste", variant="primary")
+    analyze_result_btn = gr.Button("Analyze Waste", variant="primary")
 
-    analyze_btn.click(
-        full_pipeline,
-        inputs=img_input,
-        outputs=[cls_output, explain_output]
+    analyze_result_btn.click(
+        analyze_pipeline,
+        inputs=None,
+        outputs=explain_output
     )