Update app.py

app.py

@@ -41,8 +41,11 @@ transform = transforms.Compose([
 # Predict and save
 def predict_retinopathy(image):
     # Validate image
-    if not looks_like_fundus(image):
-        return None, "❌ Please upload a valid retinal (fundus) scan."
+     if not looks_like_fundus(image):
+        return (
+            invalid_img,
+            "❌ Invalid image detected.\n\nPlease upload a *retinal fundus scan* for accurate Diabetic Retinopathy assessment."
+        )
 
     timestamp = datetime.datetime.now().strftime("%Y%m%d_%H%M%S")
     img = image.convert("RGB").resize((224, 224))
@@ -67,7 +70,10 @@ def predict_retinopathy(image):
     filename = f"{timestamp}_{label}_{confidence:.2f}.png"
     cam_pil.save(os.path.join(save_dir, filename))
 
-    return cam_pil, f"{label} (Confidence: {confidence:.2f})"
+   return (
+        cam_pil,
+        f"{label} (Confidence: {confidence:.2f})"
+    )
 
 # Gradio app
 gr.Interface(