Update app.py

app.py

@@ -21,9 +21,17 @@ def report_feedback():
     if last_prediction is not None:
         image, predicted_label = last_prediction
         correct_label = 1 if predicted_label == 0 else 0  # Invert the label
-        model.add_data(image, correct_label)  # Pass the incorrect prediction to the model
-        return "Feedback recorded. Thank you!"
-    return "No prediction available to report."
+        print(f"Reporting feedback: predicted_label={predicted_label}, correct_label={correct_label}")  # Debugging line
+        try:
+            model.add_data(image, correct_label)  # Pass the incorrect prediction to the model
+            print("Feedback recorded successfully.")  # Debugging line
+            return "Feedback recorded. Thank you!"
+        except Exception as e:
+            print(f"Error recording feedback: {e}")  # Debugging line
+            return f"Error recording feedback: {e}"
+    else:
+        print("No prediction available to report.")  # Debugging line
+        return "No prediction available to report."
 
 # Define the Gradio interface
 with gr.Blocks() as demo: