add flag_prediction

ui/app.py

@@ -77,7 +77,29 @@ class PlantDiseaseApp:
         except Exception as e:
             return None, f"Error during prediction: {str(e)}", ""
 
+    def flag_prediction(self, image, result_info, feedback_text):
+        if image is None:
+            return "No image uploaded."
+
+        if not feedback_text.strip():
+            return "Please enter feedback before submitting."
 
+        try:
+            timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
+
+            entry = {
+                "timestamp": timestamp,
+                "feedback": feedback_text,
+                "model": self.current_modelName,
+                "result_info": result_info
+            }
+
+            self.flagged_predictions.append(entry)
+
+            return "Thanks! Your feedback has been recorded."
+
+        except Exception as e:
+            return f"Error saving feedback: {str(e)}"
 
 def create_interface():
     app = PlantDiseaseApp()