Update app.py

app.py

@@ -54,18 +54,44 @@ def predict():
 
     result = {}
 
+    top1_is_fresh = 'Fresh' in top1_label
+    top2_is_fresh = 'Fresh' in top2_label
+
+    # Rule 2: top1 confidence ≥ 80%
     if top1_conf >= 80:
-        result['message'] = 'Fresh' if 'Fresh' in top1_label else 'Rotten'
-    elif ('Fresh' in top1_label and 'Rotten' in top2_label) or ('Rotten' in top1_label and 'Fresh' in top2_label):
-        result['message'] = f"{top1_label} ({top1_conf:.1f}%) vs {top2_label} ({top2_conf:.1f}%)"
-    elif 'Rotten' in top1_label:
-        result['message'] = '⚠️ Do not eat this fruit.'
-    elif 'Fresh' in top1_label and top1_conf < 80:
-        result['message'] = '🍃 Seems fresh, but be cautious.'
+        result['message'] = '✅ Looks Fresh!' if top1_is_fresh else '⚠️ Rotten Fruit Detected!'
+        result['prediction'] = 'Fresh' if top1_is_fresh else 'Rotten'
+        result['confidence'] = round(top1_conf, 2)
 
-    result['confidence'] = round(top1_conf, 2)
-    result['fruitType'] = top1_label
-    result['fresh'] = 'Fresh' in top1_label
-    return jsonify(result)
+    else:
+        # Rule 3: top1 and top2 are of different categories
+        if top1_is_fresh != top2_is_fresh:
+            result['message'] = f"{'Fresh' if top1_is_fresh else 'Rotten'} ({top1_conf:.1f}%) vs {'Fresh' if top2_is_fresh else 'Rotten'} ({top2_conf:.1f}%)"
+            result['prediction'] = f"{'Fresh' if top1_is_fresh else 'Rotten'} vs {'Fresh' if top2_is_fresh else 'Rotten'}"
+            result['confidence'] = {
+                'Fresh': round(top1_conf, 2) if top1_is_fresh else round(top2_conf, 2),
+                'Rotten': round(top1_conf, 2) if not top1_is_fresh else round(top2_conf, 2)
+            }
+            # Rule 5 message:
+            if not top1_is_fresh and top1_conf > top2_conf:
+                result['advice'] = '⚠️ Do not eat this fruit.'
+            elif top1_is_fresh and top1_conf > top2_conf:
+                result['advice'] = '🍃 Seems fresh, but be cautious.'
+            elif top2_is_fresh and top2_conf > top1_conf:
+                result['advice'] = '🍃 Seems fresh, but be cautious.'
+            else:
+                result['advice'] = '⚠️ Do not eat this fruit.'
 
-    
+        # Rule 4: both predictions from same category (Fresh & Fresh OR Rotten & Rotten)
+        else:
+            category = 'Fresh' if top1_is_fresh else 'Rotten'
+            result['message'] = f"{category} ({top1_conf:.1f}%)"
+            result['prediction'] = category
+            result['confidence'] = round(top1_conf, 2)
+            # Rule 5 message:
+            if category == 'Rotten':
+                result['advice'] = '⚠️ Do not eat this fruit.'
+            else:
+                result['advice'] = '🍃 Seems fresh, but be cautious.'
+
+    return jsonify(result)