Update backendapi.py

backendapi.py

@@ -161,15 +161,24 @@ def predict():
 
         # Generate explanation using Gemini AI
         explanation = generate_explanation(predicted_class)
-
-        # Store the result in the in-memory dictionary
-        prediction_id = str(time.time())  # Unique ID for the request
-        prediction_results[prediction_id] = {
-            "predicted_class": predicted_class,
-            "conservation_status": conservation_status,
-            "confidence_level": confidence_level,
-            "explanation": explanation
+        prediction_results = {}
+        if confidence_level<=30:
+            prediction_id = str(time.time())  # Unique ID for the request
+            prediction_results[prediction_id] = {
+            "predicted_class": "---",
+            "conservation_status": "---",
+            "confidence_level": "---",
+            "explanation": "Sorry! This image doesn’t seem to match any animals I was trained on. I’m doing my best with the knowledge I have. Please try a different image – ideally of a single, clear animal."
         }
+        else:        
+        # Store the result in the in-memory dictionary
+            prediction_id = str(time.time())  # Unique ID for the request
+            prediction_results[prediction_id] = {
+                "predicted_class": predicted_class,
+                "conservation_status": conservation_status,
+                "confidence_level": confidence_level,
+                "explanation": explanation
+            }
         print(f"Prediction ID: {prediction_id}, Result: {prediction_results[prediction_id]}")
         return jsonify({"prediction_id": prediction_id}), 200